A prior guide feature enhancement network for classifying ischaemic stroke onset time based on DWI and FLAIR imaging

Junjie Ying,Yu Xin,Chaochao Wang
DOI: https://doi.org/10.1016/j.bspc.2024.106897
IF: 5.1
2024-10-02
Biomedical Signal Processing and Control
Abstract:Thrombolytic therapy for patients with acute ischaemic stroke (AIS) is only effective within 4.5 h after stroke onset. Therefore, accurately predicting the time since stroke (TSS) is crucial in making appropriate treatment plans for AIS patients. With the rapid advancements in computer technology, deep learning techniques have made it possible to automate medical image analysis and diagnosis, improving medical diagnosis efficiency and accuracy. This makes deep learning a promising trend in medical image analysis development. The TSS classification task involves stroke lesion localization and region of interest (ROI) feature extraction. When using deep learning techniques to solve a TSS classification task, the following three issues need to be considered. (1) When using a deep learning network to locate the stroke lesion, domain shift problems exist between datasets due to differences in imaging conditions, making it difficult to obtain reliable results on other datasets using a network trained on a specific dataset. (2) Due to the small size of some stroke lesions, the corresponding stroke lesion ROI area is also small. In the CNN downsampling process, ROI features are seriously lost. (3) In MRI images, the unclear boundary and morphology of stroke lesions and other tissues make it difficult to distinguish stroke lesion features from other tissue features in feature extraction. To address the aforementioned issues, we propose a deep learning-based TSS classification network called PGFENet. In PGFENet, a multilevel adversarial learning-based domain adaptation strategy is used to address domain shift problems and reduce distribution differences between different datasets. Additionally, we design a key slice selection module to select key slices that provide rich features for classification, addressing the feature loss issue. Furthermore, a prior guide feature enhancement module is proposed to enhance stroke lesion feature discriminability, which constructs a mismatch segmentation loss based on prior knowledge of DWI-FLAIR mismatch. Finally, an attention-augmented joint prediction module is proposed to comprehensively consider the features of multiple slices to achieve reliable TSS classification predictions. The experiments demonstrate that PGFENet performs better than other mainstream classification networks. Specifically, the best TSS classification accuracy, precision, F1-score and AUC are obtained by the proposed network.
engineering, biomedical
What problem does this paper attempt to address?