arrow
Return

Efficient multiscale spatial attention 3D abdominal multiorgan segmentation model

delete2024-05-16
delete0
PRE
AI
H
Huimin Hou
T
Tongtong Shen
H
Huafei Xu
C
Chen Zhai
郑文 cover
郑文 (Wen Zheng) *
DOI:10.1002/ima.23096delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Accurate and efficient 3D medical image segmentation models are critical for clinical applications. However, many models which performed well in recent years have consumed significant computational cost, especially those based on self-attention. For more efficient and accurate segmentation of medical 3D images, we develop an efficient multiscale spatial attention model for 3D abdominal multiorgan segmentation. We use 1D and 2D kernels in the encoder and decoder modules to simulate 3D kernels to reduce the number of parameters and computational cost. Furthermore, we use the lightweight spatial attention and mixed pooling module to capture cross-dimensional information, multiscale features, and long-range contextual information. We achieved a Dice Similarity Coefficient (DSC) of 89.86 and a Normalized Surface Dice (NSD) of 78.2 on the FLARE dataset with only 1183 M GPU memory usage, 9 M parameters, and 193G floating point operations(FLOPs).
Keywords:
3D segmentation
abdominal multiorgan segmentation
efficient segmentation
multiscale

Journal

International Journal of Imaging Systems and Technology cover
International Journal of Imaging Systems and Technology
IF:
2.5
Papers:
2.1K
Citations:
2.3K

Organization

B
Beijing Hospital
Scholars:
3.0K
Papers: 1.9K
Citations: 3
T
Taiyuan University of Technology
Scholars:
2.2W
Papers: 1.4W
Citations: 1.8W