PMFSNet: Polarized multi-scale feature self-attention network for lightweight medical image segmentation.

Journal: Computer methods and programs in biomedicine
Published Date:

Abstract

BACKGROUND AND OBJECTIVES: Current state-of-the-art medical image segmentation methods prioritize precision but often at the expense of increased computational demands and larger model sizes. Applying these large-scale models to the relatively limited scale of medical image datasets tends to induce redundant computation, complicating the process without the necessary benefits. These approaches increase complexity and pose challenges for integrating and deploying lightweight models on edge devices. For instance, recent transformer-based models have excelled in 2D and 3D medical image segmentation due to their extensive receptive fields and high parameter count. However, their effectiveness comes with the risk of overfitting when applied to small datasets. It often neglects the vital inductive biases of Convolutional Neural Networks (CNNs), essential for local feature representation.

Authors

  • Jiahui Zhong
    Guangzhou University of Chinese Medicine, Guangzhou 510000, China.
  • Wenhong Tian
    School of Information and Software Engineering, University of Electronic Science and Technology of China, Chengdu, 610054, PR China. Electronic address: tian_wenhong@uestc.edu.cn.
  • Yuanlun Xie
    School of Information and Software Engineering, University of Electronic Science and Technology of China, China.
  • Zhijia Liu
    School of Information and Software Engineering, University of Electronic Science and Technology of China, Chengdu, 610054, PR China. Electronic address: zhijialiu82@gmail.com.
  • Jie Ou
    School of Information and Software Engineering, University of Electronic Science and Technology of China, Chengdu, 610054, PR China. Electronic address: oujieww6@gmail.com.
  • Taoran Tian
    State Key Laboratory of Oral Diseases, National Clinical Research Center for Oral Diseases, West China Hospital of Stomatology, Sichuan University, Chengdu, China.
  • Lei Zhang
    Division of Gastroenterology, Union Hospital, Tongji Medical College Medical College, Huazhong University of Science and Technology, Wuhan, China.