在流形上将最优传输耦合转为确定性映射,解决曲率带来的挑战。
Barycentric Projections of Optimal Transport Plans on Riemannian Manifolds
- 提出流形上的内蕴巴氏投影,用条件弗雷切特均值实现映射转换
- 证明内蕴投影在测地线平方损失下最优,其误差定义为条件方差蒙日缺陷
- 提出切向对数-指数投影,适合作局部位移近似,适用于实际学习任务
最优传输耦合是概率性对象,但许多学习流程需要确定性映射。在欧氏空间中,巴氏投影通过条件期望将耦合转化为映射,但在黎曼流形上,曲率和割点使该操作变得非平凡。本文提出黎曼流形上运输耦合的巴氏投影框架。内蕴投影将每个源点映射到其目标分布的条件弗雷切特均值,在测地线平方损失下是最优的确定性代表。对应的最小值为集成条件弗雷切特方差,仅当耦合由映射生成时为零,从而定义了条件方差蒙日缺陷。同时研究了切向对数-指数投影,证明其在欧氏情形下精确,并与蒙日情形下的 Brenier-McCann 映射兼容,可解释为内在目标的一阶黎曼梯度更新。对于离散耦合,两种构造均可按行分解为加权弗雷切特均值与对数-指数问题。在球面数据、合成 SPD 数据及真实脑电协方差矩阵上的实验验证了角色分工:内蕴投影为变分代表,切向投影为有用的局部位移代理。
原文摘要 · Abstract (English)
Optimal transport couplings are probabilistic objects, while many learning pipelines require deterministic maps. In Euclidean space, barycentric projection converts a coupling into a map by taking conditional expectations, but on a Riemannian manifold curvature and cut loci make this operation nontrivial. We develop a framework for barycentric projections of transport couplings on Riemannian manifolds. The intrinsic projection maps each source point to the conditional Fréchet mean of its destination law and is shown to be the best deterministic representative under squared geodesic loss. The corresponding minimum value is an integrated conditional Fréchet variance, which vanishes exactly for map-induced couplings and therefore defines a conditional-variance Monge defect. We also study a tangential log-exp projection, prove its Euclidean exactness, its compatibility with Brenier-McCann maps in the Monge case, and its interpretation as the first unit Riemannian gradient update for the intrinsic objective. For discrete couplings, both constructions decompose row-wise into weighted Fréchet mean and log-exp problems. Experiments on spherical data, synthetic SPD data, and real EEG covariance matrices support the proposed division of roles: the intrinsic projection is the variational representative, while the tangential projection is a useful local displacement surrogate.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。