CSAM: A 2.5D Cross-Slice Attention Module for Anisotropic Volumetric Medical Image Segmentation
Not on the map. Why?
Abstract (arXiv)
A large portion of volumetric medical data, especially magnetic resonance imaging (MRI) data, is anisotropic, as the through-plane resolution is typically much lower than the in-plane resolution. Both 3D and purely 2D deep learning-based segmentation methods are deficient in dealing with such volumetric data since the performance of 3D methods suffers when confronting anisotropic data, and 2D methods disregard crucial volumetric information. Insufficient work has been done on 2.5D methods, in which 2D convolution is mainly used in concert with volumetric information. These models focus on learning the relationship across slices, but typically have many parameters to train. We offer a Cross-Slice Attention Module (CSAM) with minimal trainable parameters, which captures information across all the slices in the volume by applying semantic, positional, and slice attention on deep feature maps at different scales. Our extensive experiments using different network architectures and tasks demonstrate the usefulness and generalizability of CSAM. Associated code is available at https://github.com/aL3x-O-o-Hung/CSAM.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Builds on 30 of its 51 references that are in the map. First posted on arXiv 8 Nov 2023.
Builds on
30 of 51 references.
References · 30
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2017Squeeze-and-Excitation Networkscited by 4,789
- 2018CBAM: Convolutional Block Attention Modulecited by 2,477
- 2016V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentationcited by 2,292
- 2018UNet++: A Nested U-Net Architecture for Medical Image Segmentationcited by 1,525
- 20163D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotationcited by 1,476
- 2018Attention U-Net: Learning Where to Look for the Pancreascited by 1,435
18 more
- 2021Swin-Unet: Unet-like Pure Transformer for Medical Image Segmentationcited by 1,222
- 2016Instance Normalization: The Missing Ingredient for Fast Stylizationcited by 1,198
- 2017A Survey on Deep Learning in Medical Image Analysiscited by 1,186
- 2021UNETR: Transformers for 3D Medical Image Segmentationcited by 959
- 2021Medical Transformer: Gated Axial-Attention for Medical Image Segmentationcited by 320
- 2018nnU-Net: Self-adapting Framework for U-Net-Based Medical Image Segmentationcited by 291
- 2018BAM: Bottleneck Attention Modulecited by 172
- 2021LeViT-UNet: Make Faster Encoders with Transformer for Medical Image Segmentationcited by 85
- 2021U-Net Transformer: Self and Cross Attention for Medical Image Segmentationcited by 81
- 2021AFTer-UNet: Axial Fusion Transformer UNet for Medical Image Segmentationcited by 48
- 2021A Robust Volumetric Transformer for Accurate 3D Tumor Segmentationcited by 41
- 2020Volumetric Attention for 3D Medical Image Segmentation and Detectioncited by 29
- 2019USE-Net: incorporating Squeeze-and-Excitation blocks into U-Net for prostate zonal segmentation of multi-institutional MRI datasetscited by 27
- 2022SATr: Slice Attention with Transformer for Universal Lesion Detectioncited by 17
- 2022CAT-Net: A Cross-Slice Attention Transformer Model for Prostate Zonal Segmentation in MRIcited by 15
- 2018A two-stage 3D Unet framework for multi-class segmentation on full resolution imagecited by 9
- 2020RSANet: Recurrent Slice-wise Attention Network for Multiple Sclerosis Lesion Segmentationcited by 9
- 2019Thickened 2D Networks for Efficient 3D Medical Image Segmentationcited by 6
Built on by
Citing papers per year: 2024: 2; 2025: 3; 2026: 2.
7 papers.
- 2024Sli2Vol+: Segmenting 3D Medical Images Based on an Object Estimation Guided Correspondence Flow Networkcited by 6
- 2024Cross-Slice Attention and Evidential Critical Loss for Uncertainty-Aware Prostate Cancer Detectioncited by 2
- 2025MOSAIC: A Multi-View 2.5D Organ Slice Selector with Cross-Attentional Reasoning for Anatomically-Aware CT Localization in Medical Organ Segmentationcited by 1
- 2025XAG-Net: A Cross-Slice Attention and Skip Gating Network for 2.5D Femur MRI Segmentationcited by 0
- 2025KLO-Net: A Dynamic K-NN Attention U-Net with CSP Encoder for Efficient Prostate Gland Segmentation from MRIcited by 0
- 2026LiFT: Lifted Inter-slice Feature Trajectories for 3D Image Generation from 2D Generatorscited by 0
- 2026G^2RA-NET: Graph-based Cross-Slice Relation Modeling with Attention Gating for Medical Image Segmentationcited by 0
- 2026G^2RA-NET: Graph-based Cross-Slice Relation Modeling with Attention Gating for Medical Image Segmentation17 Sep 2026
- 2026LiFT: Lifted Inter-slice Feature Trajectories for 3D Image Generation from 2D Generators18 May 2026
- 2025KLO-Net: A Dynamic K-NN Attention U-Net with CSP Encoder for Efficient Prostate Gland Segmentation from MRI15 Dec 2025
- 2025XAG-Net: A Cross-Slice Attention and Skip Gating Network for 2.5D Femur MRI Segmentation8 Aug 2025
- 2025MOSAIC: A Multi-View 2.5D Organ Slice Selector with Cross-Attentional Reasoning for Anatomically-Aware CT Localization in Medical Organ Segmentation15 May 2025
- 2024Sli2Vol+: Segmenting 3D Medical Images Based on an Object Estimation Guided Correspondence Flow Network21 Nov 2024
- 2024Cross-Slice Attention and Evidential Critical Loss for Uncertainty-Aware Prostate Cancer Detection1 Jul 2024