arXiv:2509.10620cs.CVcs.LG2025-09ICCV被引 17

构建高分辨率3D脑部MRI通用自监督模型,提升多种神经疾病诊断能力。

Building a General SimCLR Self-Supervised Foundation Model Across Neurological Diseases to Advance 3D Brain MRI Diagnoses

  • 基于SimCLR框架,利用18,759名患者数据预训练3D脑MRI模型。
  • 在仅20%标注数据下仍优于其他模型,跨任务泛化能力强。
  • 开源代码与模型,适合临床研究与多病种影像分析使用。

3D结构磁共振成像(MRI)常用于监测多种神经系统疾病,如神经退行性疾病和中风。尽管深度学习在多项脑影像任务中表现良好,但多数模型针对特定任务且依赖少量标注数据,泛化能力有限。自监督学习(SSL)为构建大型医学基础模型提供了可能,可利用从健康到患病的多样化未标注数据,在2D医学影像中取得显著成果。然而,现有的少数3D脑部MRI基础模型在分辨率、覆盖范围或可及性方面仍受限。本文提出一个通用的高分辨率、基于SimCLR的3D脑部结构MRI自监督基础模型,使用来自11个公开数据集的18,759名患者(共44,958次扫描)进行预训练,涵盖多种神经系统疾病。我们在四个下游预测任务中,将该模型与掩码自编码器(MAE)及两个有监督基线模型对比,结果表明其在分布内与分布外设置下均全面领先。尤其值得注意的是,仅用20%标注样本微调时,模型在阿尔茨海默病预测任务中仍表现优异。我们提供公开代码与数据,并在https://github.com/emilykaczmarek/3D-Neuro-SimCLR发布训练好的模型,为临床脑部MRI分析贡献一个广泛适用且可访问的基础模型。

原文摘要 · Abstract (English)

3D structural Magnetic Resonance Imaging (MRI) brain scans are commonly acquired in clinical settings to monitor a wide range of neurological conditions, including neurodegenerative disorders and stroke. While deep learning models have shown promising results analyzing 3D MRI across a number of brain imaging tasks, most are highly tailored for specific tasks with limited labeled data, and are not able to generalize across tasks and/or populations. The development of self-supervised learning (SSL) has enabled the creation of large medical foundation models that leverage diverse, unlabeled datasets ranging from healthy to diseased data, showing significant success in 2D medical imaging applications. However, even the very few foundation models for 3D brain MRI that have been developed remain limited in resolution, scope, or accessibility. In this work, we present a general, high-resolution SimCLR-based SSL foundation model for 3D brain structural MRI, pre-trained on 18,759 patients (44,958 scans) from 11 publicly available datasets spanning diverse neurological diseases. We compare our model to Masked Autoencoders (MAE), as well as two supervised baselines, on four diverse downstream prediction tasks in both in-distribution and out-of-distribution settings. Our fine-tuned SimCLR model outperforms all other models across all tasks. Notably, our model still achieves superior performance when fine-tuned using only 20% of labeled training samples for predicting Alzheimer's disease. We use publicly available code and data, and release our trained model at https://github.com/emilykaczmarek/3D-Neuro-SimCLR, contributing a broadly applicable and accessible foundation model for clinical brain MRI analysis.

3D MRI自监督学习脑部疾病基础模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。