arXiv:2511.10824stat.MLcs.LG2025-11中稿 · TAG-DS 2025被引 4

用局部运输映射建模分布间回归,更灵活地捕捉复杂关系。

Neural Local Wasserstein Regression

  • 基于Wasserstein距离的局部加权,用神经网络拟合运输算子
  • 在高维和非线性分布关系上优于全局映射方法
  • 适合处理多变量、复杂几何的数据分布预测任务

我们研究分布到分布的回归估计问题,其中自变量和因变量均为概率测度。现有方法通常依赖全局最优传输映射或切空间线性化,限制了近似能力,并在多变量底层域中扭曲几何结构。本文提出神经局部Wasserstein回归(Neural Local Wasserstein Regression),一种灵活的非参数框架,通过Wasserstein空间中局部定义的传输映射建模回归。该方法类比经典核回归:基于2-Wasserstein距离的核权重将估计器局域化在参考测度附近,神经网络则参数化可灵活适应复杂数据几何的传输算子。这种局部视角扩展了可接受变换的类别,避免了全局映射假设和线性化结构的局限。我们设计了基于DeepSets架构和Sinkhorn近似损失的实用训练流程,并结合贪心参考选择策略以提升可扩展性。在高斯和混合模型的合成实验,以及MNIST上的分布预测任务中,结果表明本方法能有效捕捉现有方法难以建模的非线性和高维分布关系。

原文摘要 · Abstract (English)

We study the estimation problem of distribution-on-distribution regression, where both predictors and responses are probability measures. Existing approaches typically rely on a global optimal transport map or tangent-space linearization, which can be restrictive in approximation capacity and distort geometry in multivariate underlying domains. In this paper, we propose the \emph{Neural Local Wasserstein Regression}, a flexible nonparametric framework that models regression through locally defined transport maps in Wasserstein space. Our method builds on the analogy with classical kernel regression: kernel weights based on the 2-Wasserstein distance localize estimators around reference measures, while neural networks parameterize transport operators that adapt flexibly to complex data geometries. This localized perspective broadens the class of admissible transformations and avoids the limitations of global map assumptions and linearization structures. We develop a practical training procedure using DeepSets-style architectures and Sinkhorn-approximated losses, combined with a greedy reference selection strategy for scalability. Through synthetic experiments on Gaussian and mixture models, as well as distributional prediction tasks on MNIST, we demonstrate that our approach effectively captures nonlinear and high-dimensional distributional relationships that elude existing methods.

分布回归Wasserstein神经网络非参数

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。