A Real-World Demonstration of Machine Learning Generalizability: Intracranial Hemorrhage Detection on Head CT
Not on the map. Why?
Abstract (arXiv)
Machine learning (ML) holds great promise in transforming healthcare. While published studies have shown the utility of ML models in interpreting medical imaging examinations, these are often evaluated under laboratory settings. The importance of real world evaluation is best illustrated by case studies that have documented successes and failures in the translation of these models into clinical environments. A key prerequisite for the clinical adoption of these technologies is demonstrating generalizable ML model performance under real world circumstances. The purpose of this study was to demonstrate that ML model generalizability is achievable in medical imaging with the detection of intracranial hemorrhage (ICH) on non-contrast computed tomography (CT) scans serving as the use case. An ML model was trained using 21,784 scans from the RSNA Intracranial Hemorrhage CT dataset while generalizability was evaluated using an external validation dataset obtained from our busy trauma and neurosurgical center. This real world external validation dataset consisted of every unenhanced head CT scan (n = 5,965) performed in our emergency department in 2019 without exclusion. The model demonstrated an AUC of 98.4%, sensitivity of 98.8%, and specificity of 98.0%, on the test dataset. On external validation, the model demonstrated an AUC of 95.4%, sensitivity of 91.3%, and specificity of 94.1%. Evaluating the ML model using a real world external validation dataset that is temporally and geographically distinct from the training dataset indicates that ML generalizability is achievable in medical imaging applications.
Key facts. Cited by 4 papers in the map (data through 7 Oct 2026). Builds on 2 of its 40 references that are in the map. First posted on arXiv 9 Feb 2021.
Builds on
2 of 40 references.
References · 2
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2016Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localizationcited by 6,690
Built on by
Citing papers per year: 2021: 3; 2022: 0; 2023: 1.
4 papers.
- 2023Weakly Supervised Intracranial Hemorrhage Segmentation using Head-Wise Gradient-Infused Self-Attention Maps from a Swin Transformer in Categorical Learningcited by 6
- 2021Semi-supervised learning for generalizable intracranial hemorrhage detection and segmentationcited by 3
- 2021Generation of microbial colonies dataset with deep learning style transfercited by 2
- 2021Non-Convex Optimization with Spectral Radius Regularizationcited by 1
- 2023Weakly Supervised Intracranial Hemorrhage Segmentation using Head-Wise Gradient-Infused Self-Attention Maps from a Swin Transformer in Categorical Learning11 Apr 2023
- 2021Generation of microbial colonies dataset with deep learning style transfer6 Nov 2021
- 2021Semi-supervised learning for generalizable intracranial hemorrhage detection and segmentation3 May 2021
- 2021Non-Convex Optimization with Spectral Radius Regularization22 Feb 2021