Affiliation:
1. School of Physics and Electronic Information Engineering, Henan Polytechnic University, Jiaozuo 454000, China
Abstract
In recent years, various deep-learning methodologies have been developed for processing medical images, with Unet and its derivatives proving particularly effective in medical image segmentation. Our primary objective is to enhance the accuracy of these networks while also reducing the number of parameters and computational demands to facilitate deployment on mobile medical devices. To this end, we introduce a novel medical image segmentation network, MSLUnet, which aims to minimize parameter count and computational load without compromising segmentation effectiveness. The network features a U-shaped architecture. In the encoder module, we utilize multiple small convolutional kernels for successive convolutions rather than large ones, allowing for capturing multi-scale feature information at granular levels through varied receptive field scales. In the decoder module, an inverse bottleneck structure with depth-separable convolution employing large kernels is incorporated. This design effectively extracts spatial dimensional information and ensures a comprehensive integration of both shallow and deep features. Additionally, a lightweight three-branch attention mechanism within the skip connections enhances information transfer by capturing global contextual data across spatial and channel dimensions. Experimental evaluations conducted on several publicly available medical image datasets indicate that MSLUnet is more competitive than existing models in terms of efficiency and effectiveness.
Funder
National Natural Science Foundation of China
Reference46 articles.
1. Review of Medical Image Segmentation Based on UNet;Xu;J. Front. Comput. Sci. Technol.,2023
2. He, K., Zhang, X., Ren, S., and Sun, J. (2016, January 27–30). Deep Residual Learning for Image Recognition. Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA.
3. Vaswani, A., Shazeer, N.M., Parmar, N., Uszkoreit, J., Jones, L., Gomez, A.N., Kaiser, L., and Polosukhin, I. (2017, January 4–9). Attention is All you Need. Proceedings of the 31st Conference on Neural Information Processing Systems (NIPS 2017), Long Beach, CA, USA.
4. A Reinforcement Learning Level-Based Particle Swarm Optimization Algorithm for Large-Scale Optimization;Wang;Inf. Sci.,2022
5. SCCWalk: An Efficient Local Search Algorithm and Its Improvements for Maximum Weight Clique Problem;Wang;Artif. Intell.,2020