Work

Semantically Aware Optimal Transport for Dense Label Transfer

Year: 2026
Type: conference-paper
Abstract: Vision foundation models produce features that generalize across visual domains without finetuning, yet naively transferring labels through these feature spaces fails under large distribution shifts.We propose SAOT (Semantically Aware Optimal Transport), which learns a transport cost within a fused unbalanced optimal transport formulation for dense label transfer from frozen vision transformer features to new domains.SAOT combines a learnable appearance metric with semantic class-prototype priors, unbalanced transport for partial matching under distribution shift, and a block-sparse solver for tractable inference.We pair this with a two-stage decoder: an MLP trained on SAOT pseudo-labels, then refined via EMAteacher self-training with class-balanced sampling.On GTA5→Cityscapes with frozen DINOv2 ViT-L/14 features, SAOT+Decoder reaches 25.7% mIoU, a 3.8× improvement over nearest-neighbor transfer (6.7%), without any backbone adaptation.Per-class results show large gains on spatially coherent classes (road 90.3%, car 76.2%, building 71.5%), suggesting that learned semantic transport costs capture domain-invariant structure even under severe synthetic-to-real shifts.On VOC train→val with frozen ViT-B/16 features, the full pipeline reaches 47.5% mIoU, indicating that the approach extends beyond synthetic-to-real adaptation.
Language: English
Open Access status: hybrid
OpenAlex ID: W7166812361
FWCI:
Cites:
Cited by:
Related to: 10