AMAES: 公共脑 MRI 数据上的增强掩码自编码器预训练用于 3D 本地分割
💡
原文中文,约500字,阅读约需2分钟。
📝
内容提要
本研究通过自监督预训练在特定领域数据集上探究其对3D语义分割模型的影响。使用AMAES在提出的数据集上进行预训练显著提高了分割性能。可在GitHub获取复现结果的代码、模型检查点和BRAINS-45K数据集。
🎯
关键要点
- 本研究探讨了自监督预训练对3D语义分割模型的影响。
- 引入了BRAINS-45K数据集,包含44,756个脑部MRI体积,是最大的公共数据集。
- 对现代分割架构的预训练设计进行了简化和优化。
- 结合了一种新颖的增强策略,通过遮罩图像建模和基于强度的增强翻转实现。
- 使用U-Net和MedNeXt架构评估预训练对下游任务的影响。
- 结果显示,使用AMAES进行预训练显著提高了分割性能。
- 可在GitHub获取复现结果的代码、模型检查点和BRAINS-45K数据集。
➡️