Back

CerebAI: Explainable Three-Class Stroke CT Classification via ConvNeXt and Integrated Gradients

Shenoy, A. R.; Mendez, T.

2026-07-06 radiology and imaging
10.64898/2026.07.03.26357233 medRxiv
Show abstract

Stroke is a leading cause of death and long-term disability worldwide, affecting approximately 15 million individuals annually. Prompt and accurate subtype differentiation between ischemic and hemorrhagic stroke is clinically critical, as the two conditions demand diametrically opposite interventions - thrombolytic therapy versus surgical decompression. Yet the majority of existing deep learning approaches reduce this problem to binary detection, and virtually none address the opacity of their decision-making in a clinically actionable manner. We present CerebAI, an explainable, deployment-oriented three-class CT stroke classification system built on a fine-tuned ConvNeXt-Base backbone with Integrated Gradients (IG) attribution. Trained on 6,774 non-contrast CT scans stratified across No Stroke, Ischemic Stroke, and Hemorrhagic Stroke, CerebAI achieves a weighted F1-score of 0.9746 (95% CI: [0.9625, 0.9851]), accuracy of 97.47%, macro-averaged AUC of 0.9921, mean Intersection-over-Union (mIoU) of 0.9276, Expected Calibration Error (ECE) of 0.0115, mean Brier Score of 0.0150, and Cohen's {kappa} of 0.9483 - surpassing ResNet-50, EfficientNet-B4, and Vision Transformer (ViT-B/16) baselines across all reported metrics. Integrated Gradients produce pixel-precise saliency maps that localize pathological regions with greater anatomical fidelity than Gradient-weighted Class Activation Mapping (Grad-CAM), a finding we support with side-by-side qualitative comparison. CerebAI additionally incorporates a native DICOM processing pipeline to facilitate future clinical translation. Code and model weights are publicly available to support reproducibility and further research.

Matching journals

The top 7 journals account for 50% of the predicted probability mass.

1
Nature Machine Intelligence
70 papers in training set
Top 0.1%
12.4%
2
Nature Communications
5641 papers in training set
Top 16%
11.7%
3
Medical Image Analysis
35 papers in training set
Top 0.1%
6.6%
4
European Radiology
15 papers in training set
Top 0.1%
5.4%
5
npj Digital Medicine
118 papers in training set
Top 1%
5.4%
6
Imaging Neuroscience
282 papers in training set
Top 1%
4.7%
7
Nature Medicine
125 papers in training set
Top 0.4%
4.7%
50% of probability mass above
8
Scientific Reports
3612 papers in training set
Top 21%
4.7%
9
Nature
645 papers in training set
Top 5%
2.6%
10
Human Brain Mapping
329 papers in training set
Top 2%
2.6%
11
Science Advances
1243 papers in training set
Top 17%
2.1%
12
Neuro-Oncology Advances
25 papers in training set
Top 0.3%
1.7%
13
PLOS ONE
5266 papers in training set
Top 49%
1.7%
14
NeuroImage
903 papers in training set
Top 5%
1.5%
15
NeuroImage: Clinical
144 papers in training set
Top 2%
1.4%
16
Brain Communications
166 papers in training set
Top 2%
1.3%
17
European Heart Journal - Digital Health
18 papers in training set
Top 0.8%
1.1%
18
Science Translational Medicine
127 papers in training set
Top 3%
1.1%
19
Scientific Data
209 papers in training set
Top 2%
1.1%
20
Magnetic Resonance in Medicine
85 papers in training set
Top 0.5%
1.1%
21
Nature Computational Science
55 papers in training set
Top 1%
1.1%
22
Frontiers in Computational Neuroscience
60 papers in training set
Top 1%
1.0%
23
Stroke
40 papers in training set
Top 0.8%
1.0%
24
Communications Medicine
113 papers in training set
Top 4%
1.0%
25
Aperture Neuro
20 papers in training set
Top 0.5%
0.8%
26
Frontiers in Artificial Intelligence
20 papers in training set
Top 0.8%
0.8%
27
Communications Biology
993 papers in training set
Top 32%
0.8%
28
Medical Physics
14 papers in training set
Top 0.6%
0.8%
29
IEEE Transactions on Medical Imaging
21 papers in training set
Top 0.5%
0.8%
30
Computers in Biology and Medicine
128 papers in training set
Top 5%
0.8%