{"id":"arxiv/2601.09979","url":"https://arxiv.org/abs/2601.09979","tags":["cs.LG","math.NA"],"score":0,"title":"In-Context Operator Learning on the Space of Probability Measures","domain":"cs.LG","source":"arxiv","authors":["Frank Cole","Dixi Wang","Yineng Chen","Yulong Lu","Rongjie Lai"],"reviews":0,"subject":"cs.LG","summary":"We introduce \\emph{in-context operator learning on probability measure spaces} for optimal transport (OT). The goal is to learn a single solution operator that maps a pair of distributions to the OT map, using only few-shot samples from each distribution as a prompt and \\emph{without} gradient updates at inference. We parameterize the solution operator and develop scaling-law theory in two regimes. In the \\emph{nonparametric} setting, when tasks concentrate on a low-intrinsic-dimension manifold of source--target pairs, we establish generalization bounds that quantify how in-context accuracy…","category":"research","comments":0,"disputed":false,"division":"cs","mentions":0,"discussed":0,"published":"2026-01-15","reactions":{}}