A medical image segmentation algorithm based on spatial and channel attention mechanisms and data augmentation

Yu Li; Chao Fan; Pengjiang Qian

doi:10.1117/12.2671339

28 April 2023 A medical image segmentation algorithm based on spatial and channel attention mechanisms and data augmentation

Yu Li, Chao Fan, Pengjiang Qian

Author Affiliations +

Proceedings Volume 12610, Third International Conference on Artificial Intelligence and Computer Engineering (ICAICE 2022); 126101L (2023) https://doi.org/10.1117/12.2671339
Event: Third International Conference on Artificial Intelligence and Computer Engineering (ICAICE 2022), 2022, Wuhan, China

Abstract

With the development and progress of medical imaging technology, the resolution of medical images has been increasing, and a variety of high-definition imaging modalities such as CT, PET-CT and MRI have emerged. U-Net network has the advantages of simple network topology and small training set data requirement; thus, the field of medical image segmentation uses it extensively. However, U-Net also has some problems, such as edge loss of segmentation results, long training time and single application scenario. For the medical image segmentation problem, this paper proposes a method that combines channel attention and spatial attention and uses an improved join strategy to join the network structure. To address the problem of insufficient data volume of medical images, this paper performs a data augmentation operation on the dataset with elastic deformation. In addition, we use a local-global training strategy to further improve the performance of training on medical images. When compared to the original U-Net, the Dice coefficient and IOU metrics are significantly better when utilizing the method suggested in this work. After extensive experiments, the strategy proposed in this study can achieve good outcomes when facing medical image segmentation problems and has great potential.

Citation Download Citation

Yu Li, Chao Fan, and Pengjiang Qian "A medical image segmentation algorithm based on spatial and channel attention mechanisms and data augmentation", Proc. SPIE 12610, Third International Conference on Artificial Intelligence and Computer Engineering (ICAICE 2022), 126101L (28 April 2023); https://doi.org/10.1117/12.2671339

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available