ECG-UNet——基于U型结构的轻量化医学图像分割算法

ECG-UNet:a Lightweight Medical Image Segmentation Algorithm Based on U-Shaped Structures

近年来,Transformer模型改善了深度神经网络在传统医学图像分割领域性能欠佳的问题,但因庞大的计算参数量而难以应用于移动端,为此提出一种轻量化网络ECG-UNet。首先,在保持模型性能的前提下,在瓶颈处采用线性映射与注意力相结合的策略替代普通卷积,以减少网络的参数量;同时,在网络中引入轻量化多层感知机模块,从而在图像中学习到关于分割目标更多的位置信息;其次,使用空洞卷积来获取更大的感受野;最后,在跳跃连接上加入门控注意力机制,增强网络中的特征传播,以相对较小的计算代价换取模型性能的进一步提升。在BUSI和ISIC2018两个数据集上对该模型进行验证,结果表明:本文提出的网络结构相较于当前主流算法,在分割性能更佳的情况下大大降低了计算资源的消耗。

In recent years,Transformer models have addressed the limitations of deep neural networks in traditional medical image segmentation.However,they still underperform in segmentation at the edges of medical images and suffer from large number of parameters and computational complexity,making them unsuitable for mobile applications.In this paper,we propose a lightweight network called ECG-UNet to mitigate these issues.Firstly,the model uses a strategy combining linear mapping and attention instead of conventional convolution at the bottleneck to reduce the number of network parameters while maintaining performance.Meanwhile,we introduce a lightweight multilayer perceptron module to learn more location information of the image.Secondly,dilated convolutions are applied to expand the respective field.Finally,in exchange for further improvement of the model performance at a relatively small computational cost,a gate attention mechanism is added in the skip connections to enhance the feature propagation in the network. The model isvalidated on the BUSI and ISIC2018 datasets. The results show that the proposed networkstructure greatly reduces the computational costs while achieving superior segmentationperformance compared to current mainstream algorithms.