arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2501.09096eess.IVcs.CV

用于可变对比度三维医学影像的自适应掩码自编码器自预训练

Self Pre-training with Adaptive Mask Autoencoders for Variable-Contrast 3D Medical Imaging

Siemens Healthineers AG(西门子医疗股份公司);Siemens Medical Solutions USA, Inc.(西门子医疗解决方案美国公司);FAU Erlangen-Nuremberg(埃尔朗根-纽伦堡大学)

Badhan Kumar Das, Gengyan Zhao, Han Liu, Thomas J. Re, Dorin Comaniciu, Eli Gibson, Andreas Maier

AI总结:

本文提出三维自适应掩码自编码器(AMAE),解决真实MR研究中输入对比度数量可变的问题,通过在45,364名受试者MRI数据上自预训练,使ViT分割模型的梗死分割性能提升2.8%–3.7%。

摘要:

掩码自编码器(Masked Autoencoder,MAE)近期已在预训练用于分析自然图像的视觉Transformer(Vision Transformers,ViT)方面展现出有效性。通过从部分掩码的输入中重建完整图像,ViT编码器聚合上下文信息以预测缺失区域。这种聚合上下文的能力在医学成像中尤为重要,因为解剖结构在功能和机制上与周围区域相互关联。然而,当前方法未考虑输入图像数量的变化,而这在现实世界的磁共振(Magnetic Resonance,MR)研究中通常是常见情况。为解决这一局限,我们提出了一种三维自适应掩码自编码器(3D Adaptive Masked Autoencoders,AMAE)架构,可适应每个受试者数量可变的三维输入对比度。使用包含45,364名受试者的磁共振成像(MRI)数据集进行预训练,并使用其中1648名训练、193名验证和215名测试受试者的子集进行微调。性能结果表明,这种自适应掩码自编码器的自预训练可将基于ViT的分割模型的梗死分割性能提升2.8%–3.7%。

英文摘要:

The Masked Autoencoder (MAE) has recently demonstrated effectiveness in pre-training Vision Transformers (ViT) for analyzing natural images. By reconstructing complete images from partially masked inputs, the ViT encoder gathers contextual information to predict the missing regions. This capability to aggregate context is especially important in medical imaging, where anatomical structures are functionally and mechanically linked to surrounding regions. However, current methods do not consider variations in the number of input images, which is typically the case in real-world Magnetic Resonance (MR) studies. To address this limitation, we propose a 3D Adaptive Masked Autoencoders (AMAE) architecture that accommodates a variable number of 3D input contrasts per subject. A magnetic resonance imaging (MRI) dataset of 45,364 subjects was used for pretraining and a subset of 1648 training, 193 validation and 215 test subjects were used for finetuning. The performance demonstrates that self pre-training of this adaptive masked autoencoders can enhance the infarct segmentation performance by 2.8%-3.7% for ViT-based segmentation models.

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

↑