| --- |
| 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} |
| } |
| |
