arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2511.06239stat.MLcs.LG
收藏

功能伴随采样器:无穷维空间上的可扩展采样方法

Functional Adjoint Sampler: Scalable Sampling on Infinite Dimensional Spaces

FAIR at Meta(Meta 的 FAIR)

Byoungwoo Park, Juho Lee, Guan-Horng Liu

AI总结:

针对无穷维函数空间吉布斯分布采样的理论缺口,提出基于随机最优控制的功能伴随采样器FAS,在合成势与真实分子系统上实现了更优的转移路径采样性能。

摘要:

基于学习的有限维空间吉布斯分布采样方法已快速发展,但无穷维函数空间的理论与算法设计仍有限,尽管这类空间在采样条件扩散过程路径方面潜力巨大,可实现对满足稀有事件或边界约束的扩散过程轨迹的高效模拟。本研究提出针对无穷维函数空间的伴随采样器,这是一种基于随机最优控制的扩散采样器,在函数空间中运行,目标是无穷维希尔伯特空间上的吉布斯型分布。我们的功能伴随采样器(Functional Adjoint Sampler,FAS)基于名为随机最大值原理(stochastic maximum principle)的随机最优控制理论,将伴随采样(Adjoint Sampling,Havens等人,2025)推广到希尔伯特空间,得到了一个用于函数表示的简单且可扩展的匹配型目标函数。我们表明,FAS在合成势和真实分子系统(包括丙氨酸二肽(Alanine Dipeptide)和Chignolin)上实现了更优的转移路径采样性能。

英文摘要:

Learning-based methods for sampling from the Gibbs distribution in finite-dimensional spaces have progressed quickly, yet theory and algorithmic design for infinite-dimensional function spaces remain limited. This gap persists despite their strong potential for sampling the paths of conditional diffusion processes, enabling efficient simulation of trajectories of diffusion processes that respect rare events or boundary constraints. In this work, we present the adjoint sampler for infinite-dimensional function spaces, a stochastic optimal control-based diffusion sampler that operates in function space and targets Gibbs-type distributions on infinite-dimensional Hilbert spaces. Our Functional Adjoint Sampler (FAS) generalizes Adjoint Sampling (Havens et al., 2025) to Hilbert spaces based on a SOC theory called stochastic maximum principle, yielding a simple and scalable matching-type objective for a functional representation. We show that FAS achieves superior transition path sampling performance across synthetic potential and real molecular systems, including Alanine Dipeptide and Chignolin.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑