|
| 1 | +# Week 2 Summary — PneumoDetect Lab Series |
| 2 | + |
| 3 | +**Theme:** Model Refinement, Explainability, and Performance Review |
| 4 | +**Duration:** Week 2 (Days 1–5) |
| 5 | +**Project:** [AI-Assisted-Pneumonia-Detection-Project](https://github.com/AAdewunmi/AI-Assisted-Pneumonia-Detection-Project) |
| 6 | + |
| 7 | +--- |
| 8 | + |
| 9 | +## Highlights |
| 10 | + |
| 11 | +| Day | Focus | Core Deliverables | |
| 12 | +|-----|--------|------------------| |
| 13 | +| W2-D1 | Model Training (Baseline) | `resnet50_baseline.pt`, initial metrics | |
| 14 | +| W2-D2 | Data Balancing & Weighted Sampling | Smoothed sample weights + balanced loader | |
| 15 | +| W2-D3 | Grad-CAM Integration | Implemented explainability via `src/gradcam.py` | |
| 16 | +| W2-D4 | Grad-CAM Refinement & Testing | Added `generate_cam()` + passing unit/integration tests | |
| 17 | +| W2-D5 | Performance Review & Clinical Report | `performance_report_v1.md`, Grad-CAM montages, ROC/PR curves | |
| 18 | + |
| 19 | +--- |
| 20 | + |
| 21 | +## Model Outcomes |
| 22 | + |
| 23 | +| Variant | Accuracy | Loss | Notes | |
| 24 | +|----------|-----------|------|-------| |
| 25 | +| Baseline | 0.7518 | 0.4999 | Solid starting point | |
| 26 | +| Balanced | 0.8115 | 0.4128 | Improved recall on minority class | |
| 27 | +| Fine-Tuned | 0.9985 | 0.0049 | Excellent generalization and focus | |
| 28 | + |
| 29 | +--- |
| 30 | + |
| 31 | +## Explainability Insights |
| 32 | +- Grad-CAM heatmaps now align with clinical regions of interest. |
| 33 | +- Overlays exported to `/reports/week2_gradcam_refinement/` and `/reports/week2_performance_review/`. |
| 34 | +- Fine-tuned ResNet-50 produced sharper, diagnostically meaningful activations. |
| 35 | + |
| 36 | +--- |
| 37 | + |
| 38 | +## Key Learning |
| 39 | +- Balancing datasets significantly improves model fairness. |
| 40 | +- Explainability pipelines can be systematically tested and versioned. |
| 41 | +- Structured documentation (via performance reports) bridges ML outputs and clinical communication. |
| 42 | + |
| 43 | +--- |
| 44 | + |
| 45 | +## Next Steps (Week 3) |
| 46 | +- Integrate Grad-CAM visualization and metrics into the **Flask-based clinician dashboard**. |
| 47 | +- Develop an interactive report viewer for ROC, PR, and Grad-CAM images. |
| 48 | +- Begin drafting **Clinical Report v2** and web integration prototype. |
| 49 | + |
| 50 | +--- |
| 51 | + |
| 52 | +**Maintained by:** Adrian Adewunmi |
| 53 | +**Repository:** [AI-Assisted-Pneumonia-Detection-Project](https://github.com/AAdewunmi/AI-Assisted-Pneumonia-Detection-Project) |
0 commit comments