Update README.md
Browse files
README.md
CHANGED
|
@@ -61,6 +61,14 @@ The model was trained on the **IDUN HPC Cluster**.
|
|
| 61 |
- **Carbon Tracking:** Training included environmental footprint monitoring (~0.25 kW/GPU on a low-carbon Norwegian grid).
|
| 62 |
- **Stability:** Evaluation shows consistent convergence across 5 folds, validated by a rigorous audit of the patient split.
|
| 63 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 64 |
## How to Load the Model
|
| 65 |
This model requires the `MultiStreamResNet` class from the `utils.py` file.
|
| 66 |
|
|
@@ -76,10 +84,6 @@ model.load_state_dict(torch.load("best_resnet_odelia_fold0.pth", map_location="c
|
|
| 76 |
model.eval()
|
| 77 |
```
|
| 78 |
|
| 79 |
-
## Explainability (Grad-CAM)
|
| 80 |
-
The model includes a wrapper for Explainable AI. It targets the layer4 of the T2 encoder to generate 3D heatmaps,
|
| 81 |
-
helping to identify the specific anatomical regions influencing the "Malignant" prediction.
|
| 82 |
-
|
| 83 |
## Limitations & Ethical Considerations
|
| 84 |
This model is for research purposes only. It was trained on the ODELIA proprietary dataset.
|
| 85 |
Predictions should not be used for clinical diagnosis without professional medical supervision.
|
|
|
|
| 61 |
- **Carbon Tracking:** Training included environmental footprint monitoring (~0.25 kW/GPU on a low-carbon Norwegian grid).
|
| 62 |
- **Stability:** Evaluation shows consistent convergence across 5 folds, validated by a rigorous audit of the patient split.
|
| 63 |
|
| 64 |
+
## Explainability (Grad-CAM)
|
| 65 |
+
The model includes a wrapper for Explainable AI. It targets the layer4 of the T2 encoder to generate 3D heatmaps,
|
| 66 |
+
helping to identify the specific anatomical regions influencing the "Malignant" prediction.
|
| 67 |
+
|
| 68 |
+
## Evaluation Results
|
| 69 |
+
The primary metric used is ROC AUC (Area Under the Receiver Operating Analytic Curve).
|
| 70 |
+
Detailed performance graphs and confusion matrices for the ensemble can be found in the associated [GitHub Repository](https://github.com/THOUAN-Simon/ODELIA_Challenge_CV_DL)
|
| 71 |
+
|
| 72 |
## How to Load the Model
|
| 73 |
This model requires the `MultiStreamResNet` class from the `utils.py` file.
|
| 74 |
|
|
|
|
| 84 |
model.eval()
|
| 85 |
```
|
| 86 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 87 |
## Limitations & Ethical Considerations
|
| 88 |
This model is for research purposes only. It was trained on the ODELIA proprietary dataset.
|
| 89 |
Predictions should not be used for clinical diagnosis without professional medical supervision.
|