A pre-trained densenet121 model for disease pattern detection in chest X-rays.
Note: The 4.1 version of this model is only compatible with the 4.1 version of the Clara Train SDK container
The model is trained using a densenet121 model [1] for disease pattern detection in chest X-rays [2,3].
The overall pipeline of the trained model is shown below:
The training was performed with the following:
train.sh
This model is trained using PLCO training data and evaluated on the PLCO validation data.
You can apply for access to the dataset at: https://biometry.nci.nih.gov/cdas/learn/plco/images/
The data must be converted to 16-bit png images before training.
This model achieves an Averaged AUC over all disease categories on the testing set of 0.832888923438338 (reproduce the results by changing VAL_DATALIST_KEY in environment.json to 'testing').
A Graph showing the training accuracy over 40 Epochs.
A graph showing the validation accuracy over 40 Epochs.
The model was validated with NVIDIA hardware and software. For hardware, the model can run on any NVIDIA GPU with memory greater than 16 GB. For software, this model is usable only as part of Transfer Learning & Annotation Tools in Clara Train SDK container. Find out more about Clara Train at the Clara Train Collections on NGC.
Full instructions for the training and validation workflow can be found in our documentation.
Input: 16-bit CXR png with a shape of 256 x 256
Preprocessing:
Output: 15 binary labels
Each bit is corresponding to the prediction of: 'Nodule', 'Mass', 'Distortion of Pulmonary Architecture', 'Pleural Based Mass', 'Granuloma', 'Fluid in Pleural Space', 'Right Hilar Abnormality', 'Left Hilar Abnormality', 'Major Atelectasis', 'Infiltrate', 'Scarring', 'Pleural Fibrosis', 'Bone/Soft Tissue Lesion', 'Cardiac Abnormality', and 'COPD'.
This training and inference pipeline was developed by NVIDIA. It is based on a classification model developed by NVIDIA researchers. This research use only software has not been cleared or approved by FDA or any regulatory agency. Clara pre-trained models are for developmental purposes only and cannot be used directly for clinical procedures.
[1] Huang, Gao, et al. "Densely connected convolutional networks." Proceedings of the IEEE conference on computer vision and pattern recognition. 2017. https://arxiv.org/abs/1608.06993.
[2] Wang, Xiaosong, et al. "Chestx-ray8: Hospital-scale chest x-ray database and benchmarks on weakly-supervised classification and localization of common thorax diseases." Proceedings of the IEEE conference on computer vision and pattern recognition. 2017. https://arxiv.org/abs/1705.02315.
[3] Chen, Haomin, et al. "Deep hierarchical multi-label classification of chest X-ray images." In International Conference on Medical Imaging with Deep Learning, pp. 109-120. PMLR, 2019. https://proceedings.mlr.press/v102/chen19a.html
End User License Agreement is included with the product. Licenses are also available along with the model application zip file. By pulling and using the Clara Train SDK container and downloading models, you accept the terms and conditions of these licenses.