Explainable AI for Medical Images
Explainable AI for Medical Images
This repository shows an example of how to use MATLAB to produce post-hoc explanations (using Grad-CAM and image LIME) for a medical image classification task.
Both methods (gradCAM
and imageLIME
) are available as part of the MATLAB Deep Learning toolbox and require only a single line of code to be applied to results of predictions made by a deep neural network (plus a few lines of code to display the results as a colormap overlaid on the actual images).
Example of gradCAM results. |
Example of imageLIME results. |
Experiment objective
Given a chest x-ray (CXR), our solution should classify it into Posteroanterior (PA) or Lateral (L) view.
Dataset
A small subset of the PadChest dataset1.
Requirements
- MATLAB 2020a or later
- Deep Learning Toolbox
- Deep Learning Toolbox™ Model for SqueezeNet Network support package
- Parallel Computing Toolbox (only required for training using a GPU)
Suggested steps
- Download or clone the repository.
- Open MATLAB.
- Edit the contents of the
dataFolder
variable in thexai_medical.mlx
file to reflect the path to your selected dataset. - Run the
xai_medical.mlx
script and inspect results.
Additional remarks
- You are encouraged to expand and adapt the example to your needs.
- The choice of pretrained network and hyperparameters (learning rate, mini-batch size, number of epochs, etc.) is merely illustrative.
- You are encouraged to (use Experiment Manager app to) tweak those choices and find a better solution.
Notes
[1] This example uses a small subset of images to make it easier to get started without having to worry about large downloads and long training times.
인용 양식
Oge Marques (2024). Explainable AI for Medical Images (https://github.com/ogemarques/xai-matlab/releases/tag/1.0), GitHub. 검색 날짜: .
MATLAB 릴리스 호환 정보
플랫폼 호환성
Windows macOS Linux태그
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!sample_rgb
버전 | 게시됨 | 릴리스 정보 | |
---|---|---|---|
1.0 |