提出跨尺度注意力机制,精准分割脑卒中影像中小病灶。
Stroke Lesion Segmentation using Multi-Stage Cross-Scale Attention
- 设计多阶段跨尺度注意力模块,增强不同大小病灶的特征提取。
- 在小病灶子集上Dice和F1得分超越所有基线方法。
- 适用于临床研究中对病灶边界精确分割的需求。
从MRI数据中精确刻画脑卒中病灶对预后及认知功能评估具有重要意义。手动分割耗时且依赖专业医生,通常仅用边界框粗略标注病灶位置与范围。为深入理解病灶对卒中后血管性认知障碍与痴呆(VCID)的影响机制,需实现病灶边界的精准分割。本文提出多阶段跨尺度注意力(MSCSA)机制,并应用于U-Net系列模型,以提升大脑结构特征与不同尺寸病灶间的映射能力。基于ATLAS v2.0数据集,MSCSA在小病灶子集上的Dice和F1分数均优于所有基线方法,同时在全数据集上保持优异性能。特别地,集成策略结合MSCSA在全数据集和小病灶子集上均取得最高Dice与F1得分。结果表明MSCSA在小病灶分割中有效,且对大病灶训练方案具有鲁棒性。代码已开源:https://github.com/nadluru/StrokeLesSeg。
原文摘要 · Abstract (English)
Precise characterization of stroke lesions from MRI data has immense value in prognosticating clinical and cognitive outcomes following a stroke. Manual stroke lesion segmentation is time-consuming and requires the expertise of neurologists and neuroradiologists. Often, lesions are grossly characterized for their location and overall extent using bounding boxes without specific delineation of their boundaries. While such characterization provides some clinical value, to develop a precise mechanistic understanding of the impact of lesions on post-stroke vascular contributions to cognitive impairments and dementia (VCID), the stroke lesions need to be fully segmented with accurate boundaries. This work introduces the Multi-Stage Cross-Scale Attention (MSCSA) mechanism, applied to the U-Net family, to improve the mapping between brain structural features and lesions of varying sizes. Using the Anatomical Tracings of Lesions After Stroke (ATLAS) v2.0 dataset, MSCSA outperforms all baseline methods in both Dice and F1 scores on a subset focusing on small lesions, while maintaining competitive performance across the entire dataset. Notably, the ensemble strategy incorporating MSCSA achieves the highest scores for Dice and F1 on both the full dataset and the small lesion subset. These results demonstrate the effectiveness of MSCSA in segmenting small lesions and highlight its robustness across different training schemes for large stroke lesions. Our code is available at: https://github.com/nadluru/StrokeLesSeg.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。