Chexvision Densenet

--- license: mit language: - en library_name: pytorch pipeline_tag: image-classification tags: - chexvision - medical-imaging - chest-xray - radiology - pytorch - multi-label-classification datasets: - arudaev/chest-xray-14-320 --- # CheXVision-DenseNet > **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). ## 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) ## Architecture <p align="center"> <img src="https://huggingface.co/arudaev/chexvision-densenet/resolve/main/arch_densenet.png" width="88%" alt="DenseNet-121 architecture"/> </p> ## Fine-Tuning Strategy <p align="center"> <img src="https://huggingface.co/arudaev/chexvision-densenet/resolve/main/finetuning_densenet.png" width="62%" alt="Two-phase fine-tuning schedule"/> </p> ## Training Pipeline <p align="center"> <img src="https://huggingface.co/arudaev/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: `arudaev/chexvision-densenet` - Dataset: [arudaev/chest-xray-14-320](https://huggingface.co/datasets/arudaev/chest-xray-14-320) · revision `44443e6ee968b3c6094b63f14a27698c40b50680` -...