English

HER-Seg: Holistically Efficient Segmentation for High-Resolution Medical Images

Image and Video Processing 2025-07-22 v2 Computer Vision and Pattern Recognition

Abstract

High-resolution segmentation is critical for precise disease diagnosis by extracting fine-grained morphological details. Existing hierarchical encoder-decoder frameworks have demonstrated remarkable adaptability across diverse medical segmentation tasks. While beneficial, they usually require the huge computation and memory cost when handling large-size segmentation, which limits their applications in foundation model building and real-world clinical scenarios. To address this limitation, we propose a holistically efficient framework for high-resolution medical image segmentation, called HER-Seg. Specifically, we first devise a computation-efficient image encoder (CE-Encoder) to model long-range dependencies with linear complexity while maintaining sufficient representations. In particular, we introduce the dual-gated linear attention (DLA) mechanism to perform cascaded token filtering, selectively retaining important tokens while ignoring irrelevant ones to enhance attention computation efficiency. Then, we introduce a memory-efficient mask decoder (ME-Decoder) to eliminate the demand for the hierarchical structure by leveraging cross-scale segmentation decoding. Extensive experiments reveal that HER-Seg outperforms state-of-the-arts in high-resolution medical 2D, 3D and video segmentation tasks. In particular, our HER-Seg requires only 0.59GB training GPU memory and 9.39G inference FLOPs per 1024×\times1024 image, demonstrating superior memory and computation efficiency. The code is available at https://github.com/xq141839/HER-Seg.

Keywords

Cite

@article{arxiv.2504.06205,
  title  = {HER-Seg: Holistically Efficient Segmentation for High-Resolution Medical Images},
  author = {Qing Xu and Zhenye Lou and Chenxin Li and Yue Li and Xiangjian He and Tesema Fiseha Berhanu and Rong Qu and Wenting Duan and Zhen Chen},
  journal= {arXiv preprint arXiv:2504.06205},
  year   = {2025}
}

Comments

Under Review

R2 v1 2026-06-28T22:51:07.306Z