提升多发性硬化病灶分割模型跨数据集泛化能力,更贴近临床实际。
Toward Generalizable Multiple Sclerosis Lesion Segmentation Models
- 基于UNet++架构,在多个公开数据集上系统训练以增强泛化性。
- 合并数据集后模型性能超越MICCAI-2016竞赛冠军,跨数据集表现稳定。
- 采用分位数归一化与大规模数据融合,适合真实医疗场景应用。
自动化多发性硬化(MS)病灶分割在早期诊断和疾病进展监测中具有重要意义。尽管基于深度学习的分割模型在诸多领域表现优异,但当前MS病灶分割的最先进水平仍不理想。不同于以往仅聚焦单一评估数据集的挑战,本研究旨在开发能在多样数据集间保持一致性能的通用模型,模拟真实临床中不同扫描仪、设置和患者群体的复杂情况。为此,我们在所有高质量公开的MS病灶分割数据集上系统训练了最先进的UNet++架构。结果表明,模型在剩余测试数据集上表现一致且具备良好的泛化能力,数据集越大越异质,模型性能越好。据我们所知,这是迄今使用公开数据集进行的最全面的跨数据集评估。此外,通过合并数据集显式扩充规模可进一步提升性能:在MSSEG2016-train、ISBI2015和3D-MR-MS数据集合并训练的模型,超越了MICCAI-2016竞赛优胜者。同时,我们证实模型泛化性还依赖于首次在该任务中使用的分位数归一化处理MRI强度。
原文摘要 · Abstract (English)
Automating Multiple Sclerosis (MS) lesion segmentation would be of great benefit in initial diagnosis as well as monitoring disease progression. Deep learning based segmentation models perform well in many domains, but the state-of-the-art in MS lesion segmentation is still suboptimal. Complementary to previous MS lesion segmentation challenges which focused on optimizing the performance on a single evaluation dataset, this study aims to develop models that generalize across diverse evaluation datasets, mirroring real-world clinical scenarios that involve varied scanners, settings, and patient cohorts. To this end, we used all high-quality publicly-available MS lesion segmentation datasets on which we systematically trained a state-of-the-art UNet++ architecture. The resulting models demonstrate consistent performance across the remaining test datasets (are generalizable), with larger and more heterogeneous datasets leading to better models. To the best of our knowledge, this represents the most comprehensive cross-dataset evaluation of MS lesion segmentation models to date using publicly available datasets. Additionally, explicitly enhancing dataset size by merging datasets improved model performance. Specifically, a model trained on the combined MSSEG2016-train, ISBI2015, and 3D-MR-MS datasets surpasses the winner of the MICCAI-2016 competition. Moreover, we demonstrate that the generalizability of our models also relies on our original use of quantile normalization on MRI intensities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。