基于MedNeXt的模型实现跨数据集的脑血管周围间隙自动分割
MedNet-PVS: A MedNeXt-Based Deep Learning Model for Automated Segmentation of Perivascular Spaces
- 采用Transformer启发的3D卷积网络,适配不同MRI模态
- 在T2w数据上达Dice=0.88,超越现有文献最佳表现
- 无需注意力机制也能高精度分割,适合多中心研究使用
扩大的血管周围间隙(PVS)被越来越多视为脑小血管病、阿尔茨海默病、中风及与年龄相关的神经退行性病变的生物标志物。然而,人工分割耗时且评分者间一致性中等;现有自动化深度学习模型性能有限,且难以在不同临床和研究MRI数据集间泛化。本研究采用MedNeXt-L-k5——一种受Transformer启发的3D编码器-解码器卷积网络——进行自动化PVS分割。训练了两个模型:一个基于来自人类连接组计划-老化(HCP-Aging)数据集的200例T2加权(T2w)MRI图像,另一个基于来自七个研究、六台扫描仪的40例异质性T1加权(T1w)MRI体积。模型性能通过内部5折交叉验证(5FCV)和留一中心交叉验证(LOSOCV)评估。在HCP-Aging T2w数据上训练的MedNeXt-L-k5模型,在白质(WM)区域达到0.88±0.06的体素级Dice分数,与该数据集报告的评分者间一致性相当,并为文献中最高水平。同模型在T1w数据上仅达0.58±0.09(WM)。在LOSOCV下,体素级Dice分数分别为0.38±0.16(WM)和0.35±0.12(脑灰质,BG),簇级分数为0.61±0.19(WM)和0.62±0.21(BG)。结果表明,MedNeXt-L-k5能高效实现跨多种T1w和T2w MRI数据集的自动化PVS分割。值得注意的是,其未优于nnU-Net,说明在PVS分割任务中,Transformer中的注意力机制提供全局上下文并非必要。
原文摘要 · Abstract (English)
Enlarged perivascular spaces (PVS) are increasingly recognized as biomarkers of cerebral small vessel disease, Alzheimer's disease, stroke, and aging-related neurodegeneration. However, manual segmentation of PVS is time-consuming and subject to moderate inter-rater reliability, while existing automated deep learning models have moderate performance and typically fail to generalize across diverse clinical and research MRI datasets. We adapted MedNeXt-L-k5, a Transformer-inspired 3D encoder-decoder convolutional network, for automated PVS segmentation. Two models were trained: one using a homogeneous dataset of 200 T2-weighted (T2w) MRI scans from the Human Connectome Project-Aging (HCP-Aging) dataset and another using 40 heterogeneous T1-weighted (T1w) MRI volumes from seven studies across six scanners. Model performance was evaluated using internal 5-fold cross validation (5FCV) and leave-one-site-out cross validation (LOSOCV). MedNeXt-L-k5 models trained on the T2w images of the HCP-Aging dataset achieved voxel-level Dice scores of 0.88+/-0.06 (white matter, WM), comparable to the reported inter-rater reliability of that dataset, and the highest yet reported in the literature. The same models trained on the T1w images of the HCP-Aging dataset achieved a substantially lower Dice score of 0.58+/-0.09 (WM). Under LOSOCV, the model had voxel-level Dice scores of 0.38+/-0.16 (WM) and 0.35+/-0.12 (BG), and cluster-level Dice scores of 0.61+/-0.19 (WM) and 0.62+/-0.21 (BG). MedNeXt-L-k5 provides an efficient solution for automated PVS segmentation across diverse T1w and T2w MRI datasets. MedNeXt-L-k5 did not outperform the nnU-Net, indicating that the attention-based mechanisms present in transformer-inspired models to provide global context are not required for high accuracy in PVS segmentation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。