Hybrid Offline and Self-Knowledge Distillation for Acute Ischemic Stroke Lesion Segmentation from Non-Contrast CT Scans

Ziying Wang,Hongqing Zhu,Jiahao Liu,Ning Chen,Bingcang Huang,Weiping Lu,Ying Wang
DOI: https://doi.org/10.1016/j.compbiomed.2024.109312
IF: 7.7
2024-01-01
Computers in Biology and Medicine
Abstract:Diagnosing and treating Acute Ischemic Stroke (AIS) within 0-24 h of onset is critical for patient recovery. While Diffusion-Weighted Imaging (DWI) and Computed Tomography Perfusion (CTP) are effective for early infarction identification, Non-Contrast CT (NCCT) remains the first-line imaging modality in emergency settings due to its efficiency and cost-effectiveness. In this work, to enhance lesion segmentation in NCCT using multi-modal information, we propose OS-AISeg, which integrates Offline knowledge distillation with Self-knowledge distillation to realize AIS segmentation. Initially, we trained a multi-modality teacher network by introducing uncertainty through Subjective Logic (SL) theory to reduce prediction errors and stabilize the training process. Subsequently, during student network training, we integrate confidence region knowledge guided by uncertainty weights and feature structure information guided by brain asymmetry. The former facilitates the acquisition of effective contextual information from paired predictions, while the latter leverages asymmetric activation maps to extract high-level structural content from multi-modality images. In self-knowledge distillation, we enhance the student network's learning of consistent global feature distributions by introducing mirrored NCCT images, thereby aiding the network in extracting knowledge directly from the modality. OS-AISeg was evaluated through five-fold cross-validation on two publicly available datasets, achieving a Dice value of 0.6196 on AISD and 0.4841 on ISLES2018. Additionally, experiments were also conducted on an external dataset, BraTS2019, as well as on a private stroke dataset named GLis. Strong correlations were observed between segmented Early Infarct (EI) and ground truth in volume analysis, validating the effectiveness of the proposed method in AIS diagnosis. The code for this project is available at https://github.com/Uni-Summer/OS-AISeg.
What problem does this paper attempt to address?