arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.00640eess.IVcs.AIcs.CV
收藏

AMAES:基于公开脑部MRI数据的增强掩码自编码器预训练用于3D原生分割

AMAES: Augmented Masked Autoencoder Pretraining on Public Brain MRI Data for 3D-Native Segmentation

University of Copenhagen(哥本哈根大学);Pioneer Centre for AI(先锋人工智能中心)

Asbjørn Munk, Jakob Ambsdorf, Sebastian Llambias, Mads Nielsen

AI总结:

该研究构建了最大规模公开脑MRI数据集BRAINS-45K,提出结合掩码图像建模与强度增强反转的AMAES预训练框架,在多项下游分割任务中显著提升性能。

摘要:

本研究探究了3D语义分割模型在大规模领域特定数据集上进行自监督预训练的效果。我们推出了BRAINS-45K数据集,它包含来自公开来源的44756个脑部MRI体数据,是目前可用的最大公开数据集;同时我们重新审视了现代分割架构预训练的多项设计选择,简化并优化了现有最优方法,还将其与一种新型增强策略相结合。由此得到的AMAES框架基于掩码图像建模和基于强度的增强反转,平衡了内存占用、运行时间和微调性能。我们以常用的U-Net和最新的MedNeXt架构为主干网络,在三项具有挑战性的下游任务上评估了预训练的效果,任务覆盖单序列、低资源场景以及域外泛化。结果表明,使用AMAES在本文提出的数据集上预训练,在大多数评估场景中显著提升了分割性能;并且尽管是在大规模数据集上预训练,使用增强方式预训练模型仍具有增益。用于复现结果的代码和模型检查点,以及BRAINS-45K数据集均可在https://github.com/asbjrnmunk/amaes获取。

英文摘要:

This study investigates the impact of self-supervised pretraining of 3D semantic segmentation models on a large-scale, domain-specific dataset. We introduce BRAINS-45K, a dataset of 44,756 brain MRI volumes from public sources, the largest public dataset available, and revisit a number of design choices for pretraining modern segmentation architectures by simplifying and optimizing state-of-the-art methods, and combining them with a novel augmentation strategy. The resulting AMAES framework is based on masked-image-modeling and intensity-based augmentation reversal and balances memory usage, runtime, and finetuning performance. Using the popular U-Net and the recent MedNeXt architecture as backbones, we evaluate the effect of pretraining on three challenging downstream tasks, covering single-sequence, low-resource settings, and out-of-domain generalization. The results highlight that pretraining on the proposed dataset with AMAES significantly improves segmentation performance in the majority of evaluated cases, and that it is beneficial to pretrain the model with augmentations, despite pretraing on a large-scale dataset. Code and model checkpoints for reproducing results, as well as the BRAINS-45K dataset are available at \url{https://github.com/asbjrnmunk/amaes}.

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

↑