File size: 2,771 Bytes
fc063d8 f0fde20 fc063d8 f0fde20 863a11d f0fde20 863a11d f0fde20 b83c1a8 f0fde20 863a11d f0fde20 863a11d f0fde20 f6f9c40 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 | ---
language:
- en
license: cc-by-4.0
tags:
- medical
- histopathology
- cancer-classification
- oncotree
- resnet50
- pytorch
- onnx
- int8
metrics:
- accuracy
- f1
library_name: timm
pipeline_tag: image-classification
model-index:
- name: Aegis Stage 2 OncoTree Subtype Classifier
results:
- task:
type: image-classification
name: Histopathology Cancer Subtype Classification
metrics:
- name: Test Accuracy
type: accuracy
value: 0.9977
- name: Macro F1-Score
type: f1
value: 0.9967
---
# Aegis Stage 2 — OncoTree Subtype Classifier (ResNet-50)
**Aegis Stage 2** is a high-performance deep learning model trained on **10,986 real histopathology tiles** across 8 mapped data sources to perform fine-grained cancer subtype categorization according to the **MSK OncoTree Taxonomy**.
- **Architecture**: ResNet-50 (`timm`)
- **Total Parameters**: 23,526,473 (~23.53 Million)
- **INT8 ONNX Model Size**: **22.66 MB**
- **Test Accuracy**: **99.77%** (425 / 426 test tiles correct)
- **Macro-F1 Score**: **99.67%**
- **License**: [Creative Commons Attribution 4.0 (CC-BY-4.0)](https://creativecommons.org/licenses/by/4.0/)
---
## Performance Benchmarks (Unseen Test Set)
Evaluated across 426 unseen histopathology test tiles:
| Subtype | Class Name | Support | Precision | Recall | **F1-Score** | Status |
|:---|:---|:---:|:---:|:---:|:---:|:---:|
| **BRCA** | Invasive Breast Carcinoma | 171 | 1.0000 | 1.0000 | **1.0000** |
| **GB** | Glioblastoma | 68 | 1.0000 | 1.0000 | **1.0000** |
| **CRC** | Colorectal Carcinoma | 65 | 1.0000 | 1.0000 | **1.0000** |
| **LUAD** | Lung Adenocarcinoma | 60 | 1.0000 | 0.9833 | **0.9916** |
| **LUSC** | Lung Squamous Cell | 62 | 0.9841 | 1.0000 | **0.9920** |
---
## Dataset Attribution & Citation
This model was trained on public histopathology research datasets:
- **LC25000** ([CC-BY-4.0](https://huggingface.co/datasets/1aurent/LC25000))
- **Dorsar Lung Cancer** ([MIT License](https://huggingface.co/datasets/dorsar/lung-cancer))
- **Breast Histopathology IDC** ([CC-BY-4.0](https://huggingface.co/datasets/dbzadnen/breast-histopathology-images))
- **Brain Tumor Pathology** ([CC-BY-4.0](https://huggingface.co/datasets/Hemg/Brain-Tumor-MRI-Dataset))
---
## Medical Research Disclaimer
**Aegis is an open-source AI research prototype intended for educational, scientific evaluation, and research purposes only.** It is not a certified medical device and must not be used for primary clinical diagnosis or treatment planning.
## Citation
```bibtex
@software{aegis_stage1,
title = {Aegis Stage 2: OncoTree Subtype Classifier ResNet-50},
author = {Ranveer Soni},
year = {2026},
url = {https://huggingface.co/AegisOSS/stage-2},
license = {CC-BY-4.0}
}
|