Chexvision Scratch

--- 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-ResNet > **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-scratch/resolve/main/arch_scratch.png" width="88%" alt="SE-ResNet architecture"/> </p> ## Training Pipeline <p align="center"> <img src="https://huggingface.co/arudaev/chexvision-scratch/resolve/main/pipeline_training.png" width="42%" alt="Training pipeline"/> </p> ## Training Metrics - Best validation macro AUC-ROC: `0.8141` - Best validation binary AUC-ROC: `0.7739` - Best validation binary F1: `0.6587` - Best checkpoint epoch: `41` ## Per-Class AUC-ROC at Best Epoch | Pathology | AUC-ROC | Visual | |----------------------|----------|---------------| | Atelectasis | `0.8022` | `████████░░` | | Cardiomegaly | `0.9059` | `█████████░` | | Effusion | `0.8831` | `█████████░` | | Infiltration | `0.7060` | `███████░░░` | | Mass | `0.8596` | `█████████░` | | Nodule | `0.7525` | `████████░░` | | Pneumonia | `0.7298` | `███████░░░` | | Pneumothorax | `0.8329` | `████████░░` | | Consolidation | `0.8080` | `████████░░` | | Edema | `0.9122` | `█████████░` | | Emphysema | `0.8545` | `█████████░` | | Fibrosis | `0.7622` | `████████░░` | | Pleural_Thickening | `0.7782` | `████████░░` | | Hernia | `0.8101` | `████████░░` | ## Training Configuration - Repository: `arudaev/chexvision-scratch` - Dataset: [arudaev/chest-xray-14-320](https://huggingface.co/datasets/arudaev/chest-xray-14-320) · revision `44443e6ee968b3c6094b63f14a27698c40b50680` - Architecture: Custom residual CNN with Squeeze-Excitation channel attention (depth [3, 4, 6, 3]) trained from scratch with shared features and dual classification hea...