Inter-Slice Dual Cross Attention and Class-Level Alignment for 2.5D Medical Image Segmentation
Capturing contextual information across adjacent slices is critical for accurate medical image segmentation. While 2D methods are computationally efficient, they often fail to model inter-slice continuity. In contrast, 3D methods capture volumetric context but require substantial computational resources. To overcome these limitations, we propose a 2.5D segmentation framework that incorporates inter-slice context while maintaining relatively high efficiency. We introduce a Dual Cross Attention (DCA) module that captures both global spatial and channel dependencies across adjacent slices. Furthermore, to enhance inter-slice consistency, we employ class-wise prototype learning, which aligns pixel embeddings of the same class across adjacent slices. Additionally, we introduce a semantic correlation loss to align DCA-refined features with decoder predictions via cosine similarity, guiding each channel to correspond more semantically with its associated class. Experiments conducted on the FLARE22 and MM-WHS datasets demonstrate that our method outperforms both 2D and 2.5D baselines, validating the effectiveness of modeling inter-slice dependencies and promoting class-level representation alignment in medical image segmentation.
Authors
- Won Young Tak (ORCID: https://orcid.org/0000-0002-1914-5141)
- Yu Rim Lee (ORCID: https://orcid.org/0000-0003-1916-1448)
- Soon Ki Jung (ORCID: https://orcid.org/0000-0003-0239-6785)
- Hyunji Lee (ORCID: https://orcid.org/0000-0002-9698-4539)
- Soo Young Park (ORCID: https://orcid.org/0000-0002-4944-4396)
Institutions
- Kyungpook National University (KR)
Publication Details
- Published
- 2025-08-11
- DOI
- https://doi.org/10.1109/avss65446.2025.11149935
- Citations
- 1
- Primary Topic
- Advanced Neural Network Applications
- Type
- article
- Field-Weighted Citation Impact
- 1.28