SCOT: Soft Correspondence via Optimal Transport for Multi-Source Cross-City Transfer

원문 초록

Cross-city transfer uses labeled data from well-instrumented cities to improve prediction in label-scarce ones. The task is difficult when cities have incompatible region partitions and no ground-truth correspondence. Existing methods either rely on heuristic anchors, which can create hubness, or align global distributions, which can over-mix heterogeneous regions. We propose SCOT (Soft Correspondence via Optimal Transport), an optimal transport-based framework for learning explicit, mass-controlled soft correspondences between unequal region sets. SCOT combines an optimal transport-weighted contrastive loss, a one-sided cycle regularizer, and a shared prototype hub for multi-source transfer. Across real-world cities and tasks, SCOT consistently outperforms strong baselines, reducing MAE and MAPE by 5-50%. Additional analyses show that the learned correspondences are selective, region-specific, and robust to heterogeneous settings.