arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2103.06104eess.IVcs.CV

U-Net Transformer: Self and Cross Attention for Medical Image Segmentation

CEDRIC - Conservatoire National des Arts et Metiers(国立工艺学院CEDRIC实验室);Visible Patient SAS(Visible Patient SAS公司)

Olivier Petit, Nicolas Thome, Clément Rambour, Luc Soler

英文摘要:

Medical image segmentation remains particularly challenging for complex and low-contrast anatomical structures. In this paper, we introduce the U-Transformer network, which combines a U-shaped architecture for image segmentation with self- and cross-attention from Transformers. U-Transformer overcomes the inability of U-Nets to model long-range contextual interactions and spatial dependencies, which are arguably crucial for accurate segmentation in challenging contexts. To this end, attention mechanisms are incorporated at two main levels: a self-attention module leverages global interactions between encoder features, while cross-attention in the skip connections allows a fine spatial recovery in the U-Net decoder by filtering out non-semantic features. Experiments on two abdominal CT-image datasets show the large performance gain brought out by U-Transformer compared to U-Net and local Attention U-Nets. We also highlight the importance of using both self- and cross-attention, and the nice interpretability features brought out by U-Transformer.

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

↑