Examination-Level Supervision for Deep Learning-based Intracranial Hemorrhage Detection on Head CT Scans.
Journal:
Radiology. Artificial intelligence
Published Date:
Jan 1, 2024
Abstract
Purpose To compare the effectiveness of weak supervision (ie, with examination-level labels only) and strong supervision (ie, with image-level labels) in training deep learning models for detection of intracranial hemorrhage (ICH) on head CT scans. Materials and Methods In this retrospective study, an attention-based convolutional neural network was trained with either local (ie, image level) or global (ie, examination level) binary labels on the Radiological Society of North America (RSNA) 2019 Brain CT Hemorrhage Challenge dataset of 21 736 examinations (8876 [40.8%] ICH) and 752 422 images (107 784 [14.3%] ICH). The CQ500 (436 examinations; 212 [48.6%] ICH) and CT-ICH (75 examinations; 36 [48.0%] ICH) datasets were employed for external testing. Performance in detecting ICH was compared between weak (examination-level labels) and strong (image-level labels) learners as a function of the number of labels available during training. Results On examination-level binary classification, strong and weak learners did not have different area under the receiver operating characteristic curve values on the internal validation split (0.96 vs 0.96; = .64) and the CQ500 dataset (0.90 vs 0.92; = .15). Weak learners outperformed strong ones on the CT-ICH dataset (0.95 vs 0.92; = .03). Weak learners had better section-level ICH detection performance when more than 10 000 labels were available for training (average = 0.73 vs 0.65; < .001). Weakly supervised models trained on the entire RSNA dataset required 35 times fewer labels than equivalent strong learners. Conclusion Strongly supervised models did not achieve better performance than weakly supervised ones, which could reduce radiologist labor requirements for prospective dataset curation. CT, Head/Neck, Brain/Brain Stem, Hemorrhage © RSNA, 2023 See also commentary by Wahid and Fuentes in this issue.