English

RARE-UNet: Resolution-Aligned Routing Entry for Adaptive Medical Image Segmentation

Image and Video Processing 2025-07-22 v1 Artificial Intelligence Computer Vision and Pattern Recognition

Abstract

Accurate segmentation is crucial for clinical applications, but existing models often assume fixed, high-resolution inputs and degrade significantly when faced with lower-resolution data in real-world scenarios. To address this limitation, we propose RARE-UNet, a resolution-aware multi-scale segmentation architecture that dynamically adapts its inference path to the spatial resolution of the input. Central to our design are multi-scale blocks integrated at multiple encoder depths, a resolution-aware routing mechanism, and consistency-driven training that aligns multi-resolution features with full-resolution representations. We evaluate RARE-UNet on two benchmark brain imaging tasks for hippocampus and tumor segmentation. Compared to standard UNet, its multi-resolution augmented variant, and nnUNet, our model achieves the highest average Dice scores of 0.84 and 0.65 across resolution, while maintaining consistent performance and significantly reduced inference time at lower resolutions. These results highlight the effectiveness and scalability of our architecture in achieving resolution-robust segmentation. The codes are available at: https://github.com/simonsejse/RARE-UNet.

Keywords

Cite

@article{arxiv.2507.15524,
  title  = {RARE-UNet: Resolution-Aligned Routing Entry for Adaptive Medical Image Segmentation},
  author = {Simon Winther Albertsen and Hjalte Svaneborg Bjørnstrup and Mostafa Mehdipour Ghazi},
  journal= {arXiv preprint arXiv:2507.15524},
  year   = {2025}
}

Comments

EMA4MICCAI 2025

R2 v1 2026-07-01T04:11:09.128Z