---
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}}
}
```