Expert-Level Detection of Acute Intracranial Hemorrhage on Head Computed Tomography using Deep Learning
Wei-Cheng Kuo and Christian Haene and Pratik Mukherjee and Esther Yuh and Jitendra Malik
EECS Department, University of California, Berkeley
Technical Report No. UCB/EECS-2021-13
May 1, 2021
http://www2.eecs.berkeley.edu/Pubs/TechRpts/2021/EECS-2021-13.pdf
Computed tomography (CT) of the head is used worldwide to diagnose neurologic emergencies. However, expertise is required to interpret these scans, and even highly trained experts may miss subtle life-threatening findings. For head CT, a unique challenge is to identify, with perfect or near-perfect sensitivity and very high specificity, often small subtle abnormalities on a multislice cross-sectional (3D) imaging modality that is characterized by poor soft tissue contrast, low signal-to-noise using current low-radiation-dose protocols, and a high incidence of artifacts.
We view the task as a semantic segmentation problem and tackle it with a patch-based fully convolutional network (PatchFCN). To develop the model, we collected a dataset of 4396 head CT scans performed at University of California at San Francisco and affiliated hospitals, and compared the algorithm’s performance to that of 4 American Board of Radiology (ABR) certified radiologists on an independent test set of 200 randomly selected head CT scans. Our algorithm demonstrates the highest accuracy to date for this clinical application, with a receiver operating characteristic (ROC) area under the curve (AUC) of 0.991 ± 0.006 for identification of exams positive for acute intracranial hemorrhage, and also exceeded the performance of 2 of 4 radiologists. We demonstrate an end-to-end network that performs joint classification and segmentation with exam-level classification comparable to experts, in addition to robust localization of abnormalities including some that are missed by radiologists, both of which are critically important elements for this application. Furthermore, we demonstrate promising multiclass segmentation and detection results competitive with the state-of-the-art in an exploratory study.
Finally, we study how to scale up the data without naive labeling by building a cost-sensitive active learning system. Our method compares favorably with the state-of-the-art, while running faster and using less memory. The approach is inspired by observing that the labeling time could vary greatly across examples, we model the labeling time and optimize the return on investment. We validate this idea by core-set selection and by collecting new data from the wild. Our method shows good estimation of human annotation time and clear performance gain under fixed annotation budget.
Advisors: Jitendra Malik
BibTeX citation:
@phdthesis{Kuo:EECS-2021-13, Author= {Kuo, Wei-Cheng and Haene, Christian and Mukherjee, Pratik and Yuh, Esther and Malik, Jitendra}, Title= {Expert-Level Detection of Acute Intracranial Hemorrhage on Head Computed Tomography using Deep Learning}, School= {EECS Department, University of California, Berkeley}, Year= {2021}, Month= {May}, Url= {http://www2.eecs.berkeley.edu/Pubs/TechRpts/2021/EECS-2021-13.html}, Number= {UCB/EECS-2021-13}, Abstract= {Computed tomography (CT) of the head is used worldwide to diagnose neurologic emergencies. However, expertise is required to interpret these scans, and even highly trained experts may miss subtle life-threatening findings. For head CT, a unique challenge is to identify, with perfect or near-perfect sensitivity and very high specificity, often small subtle abnormalities on a multislice cross-sectional (3D) imaging modality that is characterized by poor soft tissue contrast, low signal-to-noise using current low-radiation-dose protocols, and a high incidence of artifacts. We view the task as a semantic segmentation problem and tackle it with a patch-based fully convolutional network (PatchFCN). To develop the model, we collected a dataset of 4396 head CT scans performed at University of California at San Francisco and affiliated hospitals, and compared the algorithm’s performance to that of 4 American Board of Radiology (ABR) certified radiologists on an independent test set of 200 randomly selected head CT scans. Our algorithm demonstrates the highest accuracy to date for this clinical application, with a receiver operating characteristic (ROC) area under the curve (AUC) of 0.991 ± 0.006 for identification of exams positive for acute intracranial hemorrhage, and also exceeded the performance of 2 of 4 radiologists. We demonstrate an end-to-end network that performs joint classification and segmentation with exam-level classification comparable to experts, in addition to robust localization of abnormalities including some that are missed by radiologists, both of which are critically important elements for this application. Furthermore, we demonstrate promising multiclass segmentation and detection results competitive with the state-of-the-art in an exploratory study. Finally, we study how to scale up the data without naive labeling by building a cost-sensitive active learning system. Our method compares favorably with the state-of-the-art, while running faster and using less memory. The approach is inspired by observing that the labeling time could vary greatly across examples, we model the labeling time and optimize the return on investment. We validate this idea by core-set selection and by collecting new data from the wild. Our method shows good estimation of human annotation time and clear performance gain under fixed annotation budget.}, }
EndNote citation:
%0 Thesis %A Kuo, Wei-Cheng %A Haene, Christian %A Mukherjee, Pratik %A Yuh, Esther %A Malik, Jitendra %T Expert-Level Detection of Acute Intracranial Hemorrhage on Head Computed Tomography using Deep Learning %I EECS Department, University of California, Berkeley %D 2021 %8 May 1 %@ UCB/EECS-2021-13 %U http://www2.eecs.berkeley.edu/Pubs/TechRpts/2021/EECS-2021-13.html %F Kuo:EECS-2021-13