论文

图上成本增强Schrödinger桥可精确求解

Cost-augmented Schrödinger bridges on graphs are exactly solvable: a Feynman-Kac tilt replaces learned control

精选理由

这篇论文提出了一种精确求解图上Schrödinger桥的新方法,相比学习控制方法更高效准确。

研究人员提出了一种在图上移动质量分布的新方法,通过Feynman-Kac倾斜替代学习控制。该方法使用两个端点重缩放的交替计算,无需时间离散化或学习。在蛋白质折叠模型中,自由能成本降低了折叠路径的预期能垒。在道路网络测试中,精确桥的 rollout 结果与目标匹配,内存随网络规模线性增长。

原文 · arXiv cs.LG

Cost-augmented Schrödinger bridges on graphs are exactly solvable: a Feynman-Kac tilt replaces learned control

The generalized Schrödinger bridge on a graph moves mass between two distributions while charging a cost for the states visited. It has been approached by learning the rates of a controlled continuous-time Markov chain, with a temporal-difference penalty that restores the cost. A state cost folds into the reference process as a Feynman-Kac tilt. The cost-augmented bridge is then a plain bridge against the tilted reference, and the penalty is unnecessary. The bridge is computed exactly by alternating two endpoint rescalings, each one sparse matrix-exponential application; nothing is discretized in time or learned. The alternation converges at a rate set by the endpoint coupling alone. For a quadratic congestion cost on time-averaged occupancies, damped best response around the exact bridge is gradient descent on a strongly convex function, and its residual bounds its error. On a protein-folding model, a free-energy cost lowers the expected barrier of the folding paths. On the learned approach's road network, roll-outs of the exact bridge match the target within sampling error, and on networks with millions of intersections its memory grows linearly.