MDA-Net:基于多维注意力的三维图像分割神经网络
计算机视觉与模式识别
2021-05-11 v1 图像与视频处理
摘要
分割整幅三维图像通常计算复杂度高且内存消耗大;相比之下,以逐切片方式执行体分割虽高效,却未充分利用三维数据。为应对此挑战,我们提出多维注意力网络(MDA-Net),将切片维、空间维与通道维注意力高效整合进基于 U-Net 的网络中,从而以低计算代价获得高分割精度。我们在 MICCAI iSeg 与 IBSR 数据集上评估模型,实验结果表明相较现有方法有一致提升。
引用
@article{arxiv.2105.04508,
title = {MDA-Net: Multi-Dimensional Attention-Based Neural Network for 3D Image Segmentation},
author = {Rutu Gandhi and Yi Hong},
journal= {arXiv preprint arXiv:2105.04508},
year = {2021}
}