FocalUNETR: A Focal Transformer for Boundary-Aware Prostate Segmentation Using CT Images

FocalUNETR: A Focal Transformer for Boundary-Aware Prostate Segmentation Using CT Images

摘要

该研究致力于解决CT图像中前列腺精确定位的挑战。首先，指出了CT图像的两大问题：前列腺边界不清晰和模型捕获全局上下文的局限性。为了解决这些问题，提出了一种基于焦点变换器的图像分割架构，能够同时提取局部特征和全局上下文。其次，通过设计辅助的边界诱导标签回归任务，以解决CT图像中不清晰的边界问题。
代码地址

方法

Fig. 1展示了FocalUNETR的架构，主要包括两个部分：（A）用于前列腺分割的主任务和（B）边界感知回归辅助任务。

图2显示了（A）Focal SA机制和（B）使用Focal SA进行CT基于前列腺分割任务的边界匹配的示例（下面部分），其中焦点SA在细粒度和