用双编码器Transformer提升脑梗死病灶分割精度
Dual-Encoder Transformer-Based Multimodal Learning for Ischemic Stroke Lesion Segmentation Using Diffusion MRI
- 双编码器TransUNet分别学习DWI和ADC图像特征
- 在ISLES 2022数据集上达到85.4%的Dice分数
- 适合医学影像分析与自动化诊断研究者
从扩散磁共振成像(MRI)中准确分割缺血性卒中病灶对临床决策和预后评估至关重要。弥散加权成像(DWI)和表观弥散系数(ADC)扫描提供了急性与亚急性缺血变化的互补信息,但病灶表现差异大,自动分割仍具挑战。本文基于ISLES 2022数据集,研究多模态扩散MRI的卒中病灶分割。对比了多种先进卷积与Transformer架构,包括U-Net变体、Swin-UNet和TransUNet。根据性能表现,提出一种双编码器TransUNet架构,以从DWI和ADC输入中学习模态特异性表示。通过三切片输入配置整合相邻切片的空间上下文信息。所有模型在统一框架下训练并使用骰子相似系数(DSC)评估。结果表明,Transformer模型优于卷积基线,所提双编码器TransUNet在测试集上取得85.4%的最高骰子分数,为扩散MRI中的自动化缺血性卒中病灶分割提供稳健解决方案。
原文摘要 · Abstract (English)
Accurate segmentation of ischemic stroke lesions from diffusion magnetic resonance imaging (MRI) is essential for clinical decision-making and outcome assessment. Diffusion-Weighted Imaging (DWI) and Apparent Diffusion Coefficient (ADC) scans provide complementary information on acute and sub-acute ischemic changes; however, automated lesion delineation remains challenging due to variability in lesion appearance. In this work, we study ischemic stroke lesion segmentation using multimodal diffusion MRI from the ISLES 2022 dataset. Several state-of-the-art convolutional and transformer-based architectures, including U-Net variants, Swin-UNet, and TransUNet, are benchmarked. Based on performance, a dual-encoder TransUNet architecture is proposed to learn modality-specific representations from DWI and ADC inputs. To incorporate spatial context, adjacent slice information is integrated using a three-slice input configuration. All models are trained under a unified framework and evaluated using the Dice Similarity Coefficient (DSC). Results show that transformer-based models outperform convolutional baselines, and the proposed dual-encoder TransUNet achieves the best performance, reaching a Dice score of 85.4% on the test set. The proposed framework offers a robust solution for automated ischemic stroke lesion segmentation from diffusion MRI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。