| ---
|
| license: mit
|
| language:
|
| - en
|
| library_name: pytorch
|
| pipeline_tag: image-classification
|
| tags:
|
| - chexvision
|
| - medical-imaging
|
| - chest-xray
|
| - radiology
|
| - pytorch
|
| - multi-label-classification
|
| datasets:
|
| - HlexNC/chest-xray-14-320
|
| ---
|
|
|
| # CheXVision-DenseNet
|
|
|
| ## Project Resources
|
|
|
| - [GitHub repository](https://github.com/arudaev/chexvision)
|
| - [Presentation deck](https://arudaev.github.io/chexvision/)
|
| - [Live demo](https://huggingface.co/spaces/arudaev/chexvision-demo)
|
| - [Dataset](https://huggingface.co/datasets/arudaev/chest-xray-14-320)
|
|
|
| > **CheXVision** — Deep Learning & Big Data university project.
|
| > 14-class chest X-ray pathology detection + binary normal/abnormal classification
|
| > on the NIH Chest X-ray14 dataset (112,120 images).
|
|
|
| ## Architecture
|
|
|
| <p align="center">
|
| <img src="https://huggingface.co/HlexNC/chexvision-densenet/resolve/main/arch_densenet.png" width="88%" alt="DenseNet Architecture"/>
|
| </p>
|
|
|
| ## Fine-Tuning Strategy
|
|
|
| <p align="center">
|
| <img src="https://huggingface.co/HlexNC/chexvision-densenet/resolve/main/finetuning_densenet.png" width="62%" alt="Fine-Tuning Strategy"/>
|
| </p>
|
|
|
| ## Training Pipeline
|
|
|
| <p align="center">
|
| <img src="https://huggingface.co/HlexNC/chexvision-densenet/resolve/main/pipeline_training.png" width="42%" alt="Training Pipeline"/>
|
| </p>
|
|
|
| ## Training Metrics
|
|
|
| - Best validation macro AUC-ROC: `0.8459`
|
| - Best validation binary AUC-ROC: `0.7867`
|
| - Best validation binary F1: `0.6736`
|
| - Best checkpoint epoch: `18`
|
|
|
|
|
| ## Per-Class AUC-ROC at Best Epoch
|
|
|
| | Pathology | AUC-ROC | Visual |
|
| |----------------------|----------|---------------|
|
| | Atelectasis | `0.8334` | `████████░░` |
|
| | Cardiomegaly | `0.9010` | `█████████░` |
|
| | Effusion | `0.8873` | `█████████░` |
|
| | Infiltration | `0.7133` | `███████░░░` |
|
| | Mass | `0.8756` | `█████████░` |
|
| | Nodule | `0.8084` | `████████░░` |
|
| | Pneumonia | `0.7397` | `███████░░░` |
|
| | Pneumothorax | `0.8705` | `█████████░` |
|
| | Consolidation | `0.8063` | `████████░░` |
|
| | Edema | `0.9255` | `█████████░` |
|
| | Emphysema | `0.9107` | `█████████░` |
|
| | Fibrosis | `0.8085` | `████████░░` |
|
| | Pleural_Thickening | `0.8377` | `████████░░` |
|
| | Hernia | `0.9242` | `█████████░` |
|
|
|
| ## Training Configuration
|
|
|
| - Repository: `HlexNC/chexvision-densenet`
|
| - Dataset: [HlexNC/chest-xray-14-320](https://huggingface.co/datasets/HlexNC/chest-xray-14-320) · revision `44443e6ee968b3c6094b63f14a27698c40b50680`
|
| - Architecture: DenseNet-121 transfer learning with a shared feature layer and dual classification heads.
|
| - Platform: Kaggle GPU kernel (NVIDIA T4 / P100)
|
| - Batch size: `24` × grad_accum `4` = **effective batch `96`**
|
| - AMP (fp16): `enabled`
|
| - Optimizer: AdamW · Scheduler: CosineAnnealingLR
|
| - Epochs configured: `60` · Early stop patience: `15`
|
|
|
| ## Intended Use
|
|
|
| This model is intended for research and educational work on automated chest X-ray pathology detection.
|
| It outputs two predictions per image:
|
| 1. **Multi-label scores** — independent sigmoid probability for each of 14 NIH pathologies
|
| 2. **Binary score** — sigmoid probability of any abnormality (Normal vs. Abnormal)
|
|
|
| ## Limitations
|
|
|
| - Not validated for clinical use. Predictions must not substitute professional medical judgment.
|
| - Trained on NIH Chest X-ray14, which contains noisy radiologist annotations (patient-level labels, not lesion-level).
|
| - Performance degrades on images from equipment, patient populations, or preprocessing pipelines
|
| that differ from the NIH training distribution.
|
| - Reported AUC metrics are on the validation split, not the held-out test set.
|
|
|
| ## CheXNet Benchmark Context
|
|
|
| CheXNet (Rajpurkar et al., 2017) — the seminal paper establishing DenseNet-121 for chest X-ray
|
| classification — reported **0.841 macro AUC-ROC** on a comparable split of this dataset.
|
| CheXVision-DenseNet matches this benchmark. See the
|
| [CheXVision demo](https://huggingface.co/spaces/HlexNC/chexvision-demo) for live inference.
|
|
|
| ## Citation
|
|
|
| ```bibtex
|
| @misc{chexvision2026,
|
| title={CheXVision: Dual-Task Chest X-ray Classification with Custom CNN and DenseNet-121},
|
| author={BIG D(ATA) Team},
|
| year={2026},
|
| howpublished={\url{https://huggingface.co/HlexNC/chexvision-densenet}}
|
| }
|
| ```
|
|
|
