Code and trained models for an external validation of saliency maps as error flags for chest radiograph classifiers
Authors/Creators
Description
Code and trained models accompanying the paper "Can saliency maps flag chest radiograph classifier errors at a new hospital? An external validation study" by Aashish Chapagai, Angad Singh Gurtatta and Erik Cambria.
The study tests whether saliency maps from a DenseNet-121 and a ViT-B/16 chest radiograph classifier, trained on NIH ChestX-ray14, can detect the classifiers' errors at the training hospital and at two external hospitals (CheXpert and VinDr-CXR), beyond the models' own confidence.
This record contains:
- CXR-Saliency-External-Validation-main.zip: a snapshot of the GitHub repository, with the seven Jupyter notebooks, paper_figures.py, the environment files and the aggregate results reported in the paper
- cnn_base.keras and vit_base.keras: the trained DenseNet-121 and ViT-B/16 models (Keras 3)
- thresholds_macro.npy, thresholds_micro.npy, thresholds_hamming.npy and their vit_ equivalents: decision thresholds tuned on NIH validation data
- disease_classes.npy: the order of the 14 NIH findings
- test_probabilities.npy and vit_test_probabilities.npy: predicted probabilities on the NIH test set
No radiographs and no per-radiograph outputs from CheXpert or VinDr-CXR are included. All datasets are available from their providers under their own terms, as described in the README. The models were trained only on NIH ChestX-ray14, provided by the NIH Clinical Center.
Files
CXR-Saliency-External-Validation-main.zip
Files
(377.5 MB)
| Name | Size | |
|---|---|---|
|
md5:0f1eaa435dcd0ca38ed315c03f93aa2e
|
29.7 MB | Download |
|
md5:42db7cab9a6662f3e97e0c6538e52d92
|
966.4 kB | Preview Download |
|
md5:6b878d515bc887d0327d5746530f0f71
|
1.1 kB | Download |
|
md5:6feb1afb160ee6855b16489973006f73
|
1.2 MB | Download |
|
md5:c035fd4982f15a281b14ac13ffc335f0
|
240 Bytes | Download |
|
md5:2bda274eefca11298bc4e563258d3448
|
240 Bytes | Download |
|
md5:c849b28e5b7a64b09f20b4eb720ffdb2
|
240 Bytes | Download |
|
md5:3120dcea45eafd3f74b4d7a9d2a87011
|
344.3 MB | Download |
|
md5:e0e537f5619822eb9863dd63e6ffa46f
|
1.2 MB | Download |
|
md5:46c9c09b5732e0f48a28cd24df401f4c
|
240 Bytes | Download |
|
md5:11c11ac10939ec7316f6d4b1fb699f06
|
240 Bytes | Download |
|
md5:551c5e2bbd3dcc87b6962f28fa7a2656
|
240 Bytes | Download |
Additional details
Related works
- Is supplement to
- Preprint: 10.48550/arXiv.2611.05011 (DOI)
Software
- Repository URL
- https://github.com/aashish-chapagai/CXR-Saliency-External-Validation
- Programming language
- Python
- Development Status
- Active