Update README.md
Browse files
README.md
CHANGED
|
@@ -1,27 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
|
| 12 |
-
|
| 13 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 14 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
|
| 16 |
-
|
| 17 |
-
|
| 18 |
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
* **Optimal Weights:** Captured at **Epoch 65** (see `results.png` for convergence history).
|
| 22 |
-
* **Target Class:** `tumor` (single-class detection).
|
| 23 |
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
* **Current Status:** This model (ACE-V1) represents peak performance on the original dataset release.
|
| 27 |
-
* **Future Updates:** I am monitoring the official BRISC 2025 repository. Once a "Clean v2.0" dataset is released, I will re-train and publish **ACE-V2** to ensure complete scientific integrity.
|
|
|
|
| 1 |
+
# ACE-V1.1: Brain Tumor Detection
|
| 2 |
+
|
| 3 |
+
**ACE-V1.1** is a specialized computer vision model fine-tuned for MRI brain tumor detection. This version is a critical update that eliminates "hallucinations" (False Positives) in healthy brain tissue.
|
| 4 |
+
|
| 5 |
+
---
|
| 6 |
+
### Hardware & Environment
|
| 7 |
+
* **Training Platform:** MacBook Pro (M1 Pro Chip)
|
| 8 |
+
* **Acceleration:** Apple Silicon Metal Performance Shaders (MPS)
|
| 9 |
+
* **Framework:** Ultralytics YOLOv11
|
| 10 |
+
* **Total Epochs:** ACE-V1 (90) + Finetuning ACE-V1.1 (30) = **120 Total Epochs**
|
| 11 |
+
---
|
| 12 |
+
|
| 13 |
+
### Key Improvements in V1.1
|
| 14 |
+
* **False-Positive Rate:** Achieved **1.00 Specificity** on healthy brain scans.
|
| 15 |
+
* **Accuracy:** Verified **0.925 mAP@0.5** on the independent test set.
|
| 16 |
+
* **Performance:** Optimized for a high F1-score to ensure reliable clinical support.
|
| 17 |
+
|
| 18 |
---
|
| 19 |
+
|
| 20 |
+
### Performance & Validation
|
| 21 |
+
| Metric | Value |
|
| 22 |
+
| :--- | :--- |
|
| 23 |
+
| **mAP50** | **0.925** |
|
| 24 |
+
| **Precision** | **91.1%** |
|
| 25 |
+
| **Recall** | **89.7%** |
|
| 26 |
+
| **Background Specificity** | **1.00 (Perfect)** |
|
| 27 |
+
|
| 28 |
+
#### **Validation Proof**
|
| 29 |
+

|
| 30 |
+
*Figure 1: Normalized Confusion Matrix showing perfect separation of healthy tissue (Background).*
|
| 31 |
+
|
| 32 |
+

|
| 33 |
+
*Figure 2: Precision-Recall curve confirming the 0.925 mAP score.*
|
| 34 |
+
|
| 35 |
+
> **Note on Training Logs:** The `results.png` file reflects a high-intensity training run conducted without a validation split (`val=False`) to maximize the training data pool. Final metrics were verified using a separate hold-out test set as shown in the PR and F1 curves.
|
| 36 |
+
|
| 37 |
---
|
| 38 |
+
### Operational Guide
|
| 39 |
+
For the most reliable results, I recommend the following inference settings based on the F1-Confidence analysis:
|
| 40 |
+
|
| 41 |
+
* **Recommended Confidence:** `0.466`
|
| 42 |
+
* **Image Size:** `640x640`
|
| 43 |
|
| 44 |
+
```python
|
| 45 |
+
from ultralytics import YOLO
|
| 46 |
|
| 47 |
+
# Load the ACE-V1.1 weights
|
| 48 |
+
model = YOLO('ACE-V1.1.pt')
|
|
|
|
|
|
|
| 49 |
|
| 50 |
+
# Run inference with the optimal threshold
|
| 51 |
+
results = model.predict(source='mri_scan.jpg', conf=0.466, save=True)
|
|
|
|
|
|