Style diagram images with centred HTML + width constraints

README.md

@@ -1,167 +1,111 @@
----
-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
-
-> **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
-
-```mermaid
-graph LR
-    IN["Input
-    3 × 224 × 224"] --> BB["DenseNet-121 Backbone
-    ImageNet pretrained
-    Dense connectivity
-    7.9M parameters"]
-    BB --> GAP2["Adaptive Avg Pool
-    1024-dim features"]
-    GAP2 --> FL["Feature Layer
-    Linear 1024→512
-    ReLU · Dropout(0.3)"]
-    FL --> MLH["Multilabel Head
-    Linear 512→14
-    sigmoid · 14 pathologies"]
-    FL --> BH["Binary Head
-    Linear 512→1
-    sigmoid · Normal/Abnormal"]
-    style MLH fill:#2e7d32,color:#fff
-    style BH fill:#1565c0,color:#fff
-    style IN fill:#37474f,color:#fff
-    style BB fill:#6a1b9a,color:#fff
-```
-
-## Fine-Tuning Strategy
-
-```mermaid
-graph LR
-    P1["🔒 Phase 1
-    Epochs 1–5
-    Backbone frozen
-    Train heads only
-    lr = 0.001"] -->|"Epoch 6
-    unfreeze_backbone()"| P2["🔓 Phase 2
-    Epochs 6–60
-    End-to-end fine-tuning
-    All layers trainable
-    lr = 0.0001"]
-    style P1 fill:#e65100,color:#fff
-    style P2 fill:#6a1b9a,color:#fff
-```
-
-## Training Pipeline
-
-```mermaid
-flowchart TD
-    DS[("🗄️ HlexNC/chest-xray-14
-112,120 images · 36 shards · ~4.7 GB")]
-    DS -->|snapshot_download| PREP["📂 data/images · data/labels.csv
-train 78,468 · val 11,210 · test 22,442"]
-    PREP --> AUG["Augmentation Pipeline
-HFlip · Rotate±15° · RandomAffine
-ColorJitter · GaussianBlur · RandomErasing"]
-    AUG --> FWD["⚡ Model Forward Pass
-torch.cuda.amp.autocast · fp16"]
-    FWD --> ML["multilabel_logits B×14
-WeightedBCE + pos_weight · 14 classes"]
-    FWD --> BIN["binary_logits B×1
-BCE · Normal vs. Abnormal"]
-    ML --> LOSS["Combined Loss
-1.0 × multilabel + 0.5 × binary"]
-    BIN --> LOSS
-    LOSS --> BACK["Backward · Grad Clip 1.0
-Gradient Accumulation ×4 · eff. batch 128"]
-    BACK --> OPT["AdamW · CosineAnnealingLR
-early stop patience = 15"]
-    OPT -->|"↑ best val AUC-ROC"| BEST["💾 Best Checkpoint
-model_state · best_val_metrics · config"]
-    BEST -->|upload_model_artifacts| HUB["🤗 HF Hub
-checkpoint · history.json · model card"]
-```
-
-## 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}}
-}
-```
+---
+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
+
+> **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}}
+}
+```