English

Adapting Vision Transformers to Ultra-High Resolution Semantic Segmentation with Relay Tokens

Computer Vision and Pattern Recognition 2026-01-12 v1

Abstract

Current approaches for segmenting ultra high resolution images either slide a window, thereby discarding global context, or downsample and lose fine detail. We propose a simple yet effective method that brings explicit multi scale reasoning to vision transformers, simultaneously preserving local details and global awareness. Concretely, we process each image in parallel at a local scale (high resolution, small crops) and a global scale (low resolution, large crops), and aggregate and propagate features between the two branches with a small set of learnable relay tokens. The design plugs directly into standard transformer backbones (eg ViT and Swin) and adds fewer than 2 % parameters. Extensive experiments on three ultra high resolution segmentation benchmarks, Archaeoscape, URUR, and Gleason, and on the conventional Cityscapes dataset show consistent gains, with up to 15 % relative mIoU improvement. Code and pretrained models are available at https://archaeoscape.ai/work/relay-tokens/ .

Keywords

Cite

@article{arxiv.2601.05927,
  title  = {Adapting Vision Transformers to Ultra-High Resolution Semantic Segmentation with Relay Tokens},
  author = {Yohann Perron and Vladyslav Sydorov and Christophe Pottier and Loic Landrieu},
  journal= {arXiv preprint arXiv:2601.05927},
  year   = {2026}
}

Comments

13 pages +3 pages of suppmat