simontho commited on
Commit
e8b2ff6
·
verified ·
1 Parent(s): ff13a78

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -4
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.