May, 2023

使用基于 3D 注意力机制的 U-Net 进行多类 MRI 脑肿瘤分割

TL;DR本文提出了一种基于 3D 注意力的 U-Net 架构,用于使用由三个非本地 MRI 卷积组合而成的单一堆叠的多模态体积进行大脑肿瘤的多区域分割。该方法通过在 U-Net 的解码器端添加注意机制来提高分割精度,从而减少了对健康组织的侧重和突出了恶性组织,提高了泛化能力并减少了计算资源。该方法在 BraTS 2021 任务 1 数据集上进行训练和评估,证明其在精度上优于其他方法。实验结果表明,该方法潜在地提高了使用多模态 MRI 数据进行大脑肿瘤分割的能力,有望为更好地理解和诊断脑部疾病做出贡献。这项工作强调了在大脑肿瘤分割中结合多种成像模式和引入注意机制以提高精度的重要性。