Efficient multiscale spatial attention 3D abdominal multiorgan segmentation model

Abstract

AbstractAccurate 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).