English

Enhancing Medical Image Segmentation via Heat Conduction Equation

Computer Vision and Pattern Recognition 2026-05-18 v2

Abstract

Medical image segmentation models struggle to achieve efficient global context modeling and long-range dependency reasoning under practical computational budgets. In this work, we propose a hybrid architecture utilizing U-Mamba with Heat Conduction Equation, which combines state-space modules for efficient long-range reasoning with Heat Conduction Operators (HCOs) in the bottleneck layers, simulating frequency-domain thermal diffusion for enhanced semantic abstraction. Experimental results show that our model attains the highest DSC (0.8719) on the Abdomen CT dataset. It suggests that blending state-space dynamics with heat-based global diffusion offers a scalable solution for medical segmentation tasks.

Keywords

Cite

@article{arxiv.2511.03260,
  title  = {Enhancing Medical Image Segmentation via Heat Conduction Equation},
  author = {Rong Wu and Yim-Sang Yu},
  journal= {arXiv preprint arXiv:2511.03260},
  year   = {2026}
}
R2 v1 2026-07-01T07:22:30.825Z