Update README.md
Browse files
README.md
CHANGED
|
@@ -20,14 +20,14 @@ pipeline_tag: image-classification
|
|
| 20 |
|
| 21 |
This repository contains the "Better Model" (Phase 3) for breast cancer classification on multi-parametric MRI. This version represents the most stable and optimized iteration, featuring a hybrid architecture designed to handle high-resolution 3D sequences with extreme memory efficiency.
|
| 22 |
|
| 23 |
-
##
|
| 24 |
- **Instance Normalization Integration:** Replaced Batch Normalization with **InstanceNorm 3D**. This critical fix allows the model to maintain stable gradients even with a Batch Size of 1, preventing the collapse observed in earlier versions.
|
| 25 |
- **Dynamic Temporal Fusion:** Uses a Multi-Stream approach where the Kinetic stream processes a variable number of post-contrast phases.
|
| 26 |
-
- **Improved Temporal Aggregation:** Transitioned to **Temporal Mean Pooling** for feature fusion, providing a more robust
|
| 27 |
- **Explainable AI (XAI):** Full support for **3D Grad-CAM** visualization on the T2 structural stream.
|
| 28 |
|
| 29 |
## Model Description
|
| 30 |
-
- **Developed by:**
|
| 31 |
- **Architecture:** Multi-Stream 3D ResNet-18 (Custom InstanceNorm version).
|
| 32 |
- **Task:** 3 Multi-class classification (Normal, Benign, Malignant).
|
| 33 |
- **Input Modalities:** 1. **T2 Stream:** Structural 3D MRI volume.
|
|
|
|
| 20 |
|
| 21 |
This repository contains the "Better Model" (Phase 3) for breast cancer classification on multi-parametric MRI. This version represents the most stable and optimized iteration, featuring a hybrid architecture designed to handle high-resolution 3D sequences with extreme memory efficiency.
|
| 22 |
|
| 23 |
+
## Key Engineering Breakthroughs (Phase 3)
|
| 24 |
- **Instance Normalization Integration:** Replaced Batch Normalization with **InstanceNorm 3D**. This critical fix allows the model to maintain stable gradients even with a Batch Size of 1, preventing the collapse observed in earlier versions.
|
| 25 |
- **Dynamic Temporal Fusion:** Uses a Multi-Stream approach where the Kinetic stream processes a variable number of post-contrast phases.
|
| 26 |
+
- **Improved Temporal Aggregation:** Transitioned to **Temporal Mean Pooling** for feature fusion, providing a more robust global temporal summary compared to complex recurrent layers (LSTM/RNN) on this specific dataset volume.
|
| 27 |
- **Explainable AI (XAI):** Full support for **3D Grad-CAM** visualization on the T2 structural stream.
|
| 28 |
|
| 29 |
## Model Description
|
| 30 |
+
- **Developed by:** THOUAN Simon
|
| 31 |
- **Architecture:** Multi-Stream 3D ResNet-18 (Custom InstanceNorm version).
|
| 32 |
- **Task:** 3 Multi-class classification (Normal, Benign, Malignant).
|
| 33 |
- **Input Modalities:** 1. **T2 Stream:** Structural 3D MRI volume.
|