Focal-UNet

目录

摘要

方法

编码器

Bottleneck

解码器

总结


摘要

由于在Transformer中的Patch划分操作,在分割预测结果中仍然存在着严重的块状和裁剪边缘问题。在这项工作中,我们提出了一种新的u型结构的医学图像分割借助于新引入的焦点调制机制。所提出的架构具有不对称的编码器和解码器深度。由于焦点模块能够聚合局部和全局特征,我们的模型可以同时受益于Transformer的全局感受野和cnn的局部细节特征。这有助于所提出的方法平衡局部和全局特征的使用。

本文的主要贡献为:1)使用焦点模块作为u型架构的基础块2)加大了编码器层的深度,增强U型架构的问题建模能力。

方法

本文提出的网络结构Focal-UNet如下图所示:它由一个编码器,Bottleneck,解码器以及一个跳跃连接组成

编码器

 编码器是由三个focal layers组成的,每个focal layer层包括4个focal module(FM)块和一个下采样卷积块,对网络的接受域更大,从输入特征向量中提取出更多可泛化特征的能力更高。更高的接受域有助于网络在训练过程中考虑更多的可泛化特征,这似乎对分割问题有效。此外,这个块的计算效率更高,综上所述,这个FM块似乎很适合用作构建u型架构的基础块。在每一层中增加一个卷积块的目的是减小图像大小并增加通道数量,这在设计u型架构时是很常见的。

Bottleneck

Bottleneck是基于四个FM块构建的,从输入中提取更深层次的特征,并增加网络的容量,以建模更复杂的问题空间。

解码器

解码器的构造类似于编码器部分,主要的区别在于每一层中使用了一个FM块和一个转置卷积进行上采样,并将每一层的通道数量减少为原来的2倍

跳跃连接

U型网络的关键在于跳跃连接,它使得网络同时使用更高和更低级别的特征来构建出所需的输出,我们还在提出的体系结构中使用了类似于U型网络中跳跃连接以促进输入到输出的特征传输,跳跃连接

总结

本文提出了一种新的基于焦点的u型结构的医学图像分割方法。因此,我们的focus - unet模型在Synapse数据集上取得了比最先进的方法更好的结果。此外,我们的实验表明,对于小型训练集,与Swin-UNet相比,所提出的FocalUNet具有出色的性能和泛化能力。

猜你喜欢

转载自blog.csdn.net/weixin_43921949/article/details/128433445