arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2505.00568cs.CVcs.AI

面向存在模态缺失的3D MRI脑肿瘤分析的多模态掩码自编码器预训练

Multimodal Masked Autoencoder Pre-training for 3D MRI-Based Brain Tumor Analysis with Missing Modalities

Oncopole(奥恩波勒);IRT Saint Exupéry(伊尔特圣埃克苏佩里研究所);INSERM Cancer Research Center of Toulouse(图卢兹癌症研究中心)

Lucas Robinet, Ahmad Berjaoui, Elizabeth Cohen-Jonathan Moyal

AI总结:

针对多模态MRI脑肿瘤分析中模态缺失的问题,提出BM-MAE预训练策略,同一模型可适配任意模态组合,无需改架构即可微调,在下游任务上优于或媲美需单独预训练的基线,还能高效重建缺失模态。

摘要:

多模态磁共振成像(MRI)是临床医生诊断脑肿瘤的一线检查手段,为手术规划、治疗监测和生物标志物识别提供关键见解。在大型数据集上预训练已被证明能帮助模型学习可迁移表示,并在标记数据极少的情况下实现适配。这种特性在医学成像领域尤为重要,因为该领域的标注通常十分稀缺。然而,将这种范式应用于多模态医学数据时会面临一个挑战:大多数现有方法假设在预训练和微调阶段所有成像模态都可用。但在实际中,由于采集问题、专家不可用或小型内部数据集的特定实验设计,模态缺失现象经常发生。因此,一种常见的方法是为每种所需的模态组合训练一个单独的模型,这使得该过程既耗费资源,又难以在临床中实际应用。为此,我们提出了BM-MAE,一种专为多模态MRI数据设计的掩码图像建模预训练策略。同一个预训练模型能无缝适配任何可用模态组合,提取丰富的表示,同时捕捉模态内和模态间的信息。这使得模型无需改变架构即可在任何模态子集上进行微调,同时仍能受益于在完整模态集上预训练的模型。大量实验表明,所提出的预训练策略优于或能与需要为每个模态子集单独预训练的基线模型相媲美,同时在多个下游任务上显著超越从头训练的模型。此外,它还能快速高效地重建缺失的模态,凸显其实际价值。代码和训练好的模型可在以下网址获取:https://github.com/Lucas-rbnt/BM-MAE

英文摘要:

Multimodal magnetic resonance imaging (MRI) constitutes the first line of investigation for clinicians in the care of brain tumors, providing crucial insights for surgery planning, treatment monitoring, and biomarker identification. Pre-training on large datasets have been shown to help models learn transferable representations and adapt with minimal labeled data. This behavior is especially valuable in medical imaging, where annotations are often scarce. However, applying this paradigm to multimodal medical data introduces a challenge: most existing approaches assume that all imaging modalities are available during both pre-training and fine-tuning. In practice, missing modalities often occur due to acquisition issues, specialist unavailability, or specific experimental designs on small in-house datasets. Consequently, a common approach involves training a separate model for each desired modality combination, making the process both resource-intensive and impractical for clinical use. Therefore, we introduce BM-MAE, a masked image modeling pre-training strategy tailored for multimodal MRI data. The same pre-trained model seamlessly adapts to any combination of available modalities, extracting rich representations that capture both intra- and inter-modal information. This allows fine-tuning on any subset of modalities without requiring architectural changes, while still benefiting from a model pre-trained on the full set of modalities. Extensive experiments show that the proposed pre-training strategy outperforms or remains competitive with baselines that require separate pre-training for each modality subset, while substantially surpassing training from scratch on several downstream tasks. Additionally, it can quickly and efficiently reconstruct missing modalities, highlighting its practical value. Code and trained models are available at: https://github.com/Lucas-rbnt/BM-MAE

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑