Deep multiscale convolutional feature learning for intracranial hemorrhage classification and weakly supervised localization

Heliyon. 2024 Apr 26;10(9):e30270. doi: 10.1016/j.heliyon.2024.e30270. eCollection 2024 May 15.

Abstract

Objective: This study evaluated the performance of attentional fusion model-based multiscale features in classifying intracerebral hemorrhage and the localization of bleeding focus based on weakly supervised target localization.

Methods: A publicly available dataset provided by the American College of Neuroradiology (ASNR) was used, consisting of 750,000 computed tomography (CT) scans of the brain, manually marked by radiologists for intracranial hemorrhage and five hemorrhage subtypes. A multiscale feature classification and weakly supervised localization framework based on an attentional fusion mechanism were applied, which could be annotated at the slice level and provided intracranial hemorrhage classification and hemorrhage focus localization.

Results: The designed framework achieved excellent performance for classification and localization. The area under the curve (AUC) for predicting bleeding was 0.973. High AUC values were observed for the five hemorrhage subtypes (epidural AUC = 0.891, subdural AUC = 0.991, subarachnoid AUC = 0.983, intraventricular AUC = 0.995, intraparenchymal AUC = 0.990). This model outperformed the average entry-level radiology trainee compared to previously reported data.

Conclusion: The designed method quickly and accurately detected intracerebral hemorrhage, classifying hemorrhage subtypes and locating bleeding points with image-level annotation alone. The results indicate that this framework can significantly reduce diagnostic time while improving the detection of intracerebral hemorrhage in emergencies. It can thus be integrated into the diagnostic radiology workflow in the future.

Keywords: Deep learning; Intracranial hemorrhage; Intraparenchymal; Intraventricular; Subarachnoid.