MOSformer: Momentum encoder-based inter-slice fusion transformer for medical image segmentation
Abstract
Medical image segmentation takes an important position in various clinical applications. 2.5D-based segmentation models bridge the computational efficiency of 2D-based models with the spatial perception capabilities of 3D-based models. However, existing 2.5D-based models primarily adopt a single encoder to extract features of target and neighborhood slices, failing to effectively fuse inter-slice information, resulting in suboptimal segmentation performance. In this study, a novel momentum encoder-based inter-slice fusion transformer (MOSformer) is proposed to overcome this issue by leveraging inter-slice information from multi-scale feature maps extracted by different encoders. Specifically, dual encoders are employed to enhance feature distinguishability among different slices. One of the encoders is moving-averaged to maintain consistent slice representations. Moreover, an inter-slice fusion transformer (IF-Trans) module is developed to fuse inter-slice multi-scale features. MOSformer is evaluated on three benchmark datasets (Synapse, ACDC, and AMOS), achieving a new state-of-the-art with 85.63%, 92.19%, and 85.43% DSC, respectively. These results demonstrate MOSformer's competitiveness in medical image segmentation.
Cite
@article{arxiv.2401.11856,
title = {MOSformer: Momentum encoder-based inter-slice fusion transformer for medical image segmentation},
author = {De-Xing Huang and Xiao-Hu Zhou and Mei-Jiang Gui and Xiao-Liang Xie and Shi-Qi Liu and Shuang-Yi Wang and Zhen-Qiu Feng and Zhi-Chao Lai and Zeng-Guang Hou},
journal= {arXiv preprint arXiv:2401.11856},
year = {2025}
}
Comments
Accepted by Biomimetic Intelligence and Robotics. 13 pages, 9 figures, 8 tables