改进Mask2Former的注意力机制,提升小器官分割精度
Query Nearby: Offset-Adjusted Mask2Former enhances small-organ segmentation
- 引入可变形注意力与偏移调整,聚焦器官内部采样点
- 在HaNSeg和SegRap2023上小器官DSC超80%,达最新水平
- 适合医疗图像分割、尤其是小器官定位任务的研究者
医学图像分割在放疗和手术导航中至关重要,但小器官分割仍面临挑战。尽管基于Transformer的模型能更好捕捉全局信息,但训练资源消耗大,且对紧凑区域分割效果不佳。例如,直接使用ViT分割医学图像的DSC低于50%,远低于临床要求的80%。本文在Mask2Former基础上引入可变形注意力,提出偏移调整策略,引导注意力集中在同一器官内;同时利用第4层特征图提供器官粗略位置,并设计基于FCN的辅助头,通过Dice损失加速训练。实验表明,该方法在HaNSeg和SegRap2023数据集上达到当前最优性能,尤其在中等及小器官上表现突出。
原文摘要 · Abstract (English)
Medical segmentation plays an important role in clinical applications like radiation therapy and surgical guidance, but acquiring clinically acceptable results is difficult. In recent years, progress has been witnessed with the success of utilizing transformer-like models, such as combining the attention mechanism with CNN. In particular, transformer-based segmentation models can extract global information more effectively, compensating for the drawbacks of CNN modules that focus on local features. However, utilizing transformer architecture is not easy, because training transformer-based models can be resource-demanding. Moreover, due to the distinct characteristics in the medical field, especially when encountering mid-sized and small organs with compact regions, their results often seem unsatisfactory. For example, using ViT to segment medical images directly only gives a DSC of less than 50\%, which is far lower than the clinically acceptable score of 80\%. In this paper, we used Mask2Former with deformable attention to reduce computation and proposed offset adjustment strategies to encourage sampling points within the same organs during attention weights computation, thereby integrating compact foreground information better. Additionally, we utilized the 4th feature map in Mask2Former to provide a coarse location of organs, and employed an FCN-based auxiliary head to help train Mask2Former more quickly using Dice loss. We show that our model achieves SOTA (State-of-the-Art) performance on the HaNSeg and SegRap2023 datasets, especially on mid-sized and small organs.Our code is available at link https://github.com/earis/Offsetadjustment\_Background-location\_Decoder\_Mask2former.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。