arXiv:2605.21388cs.LGcs.AI2026-05

揭示PDE生成分布的映射规律,为单步生成模型提供理论支撑

On the Regularity and Generalization of One-Step Wasserstein-guided Generative Models for PDE-Induced Measures

论文配图:On the Regularity and Generalization of One-Step Wasserstein-guided Generative Models for PDE-Induced Measures
图 1 · 摘自论文原文
  • 基于最优传输理论,证明PDE诱导测度的映射满足Hölder连续性
  • 推导出DeepParticle模型的泛化误差上界,支持其在真实数据上的表现
  • 适用于科学计算中基于PDE生成分布的模型设计与分析

尽管生成模型在科学计算中表现出色,但其统计精度的理论研究仍显不足。本文建立了一个理论框架,用于理解线性椭圆与抛物方程在有界域上、以及环面上扩散和Fokker-Planck方程所诱导概率测度的传输映射正则性与泛化性质。在标准结构假设下,我们证明这些目标测度满足加倍条件。结合加倍测度间最优传输的正则性理论,得出从均匀源测度到目标测度的最优传输映射是Hölder连续的。这一正则性为通过单次前向映射学习PDE诱导分布的一步生成模型提供了逼近论依据。以DeepParticle为例,我们推导出学习映射与最优映射之间过剩风险的界,并在目标分布偏移下建立鲁棒性估计。实验结果验证了理论预测的收敛速率。

原文摘要 · Abstract (English)

Despite the remarkable empirical success of generative models, the available theory on their statistical accuracy in scientific computing remains largely pessimistic. This paper develops a theoretical framework for understanding the regularity of transport maps and the generalization properties of one-step Wasserstein-guided generative models for PDE-induced probability measures. We consider normalized target densities associated with linear elliptic and parabolic equations on bounded domains, as well as diffusion and Fokker--Planck equations on the torus. Under standard structural assumptions, we prove that these target measures satisfy doubling conditions. By combining this fact with regularity theory for optimal transport between doubling measures, we show that the optimal transport map from a uniform source measure to the target measure is Hölder continuous. This regularity yields an approximation-theoretic justification for one-step generative models that learn PDE-induced distributions via a single pushforward map. As a representative instance, we study DeepParticle and derive excess-risk bounds characterizing the discrepancy between the learned map and the population-optimal map. We also establish a robustness estimate under target shift and illustrate the theory with experiments which support the derived rates.

生成模型PDE最优传输泛化能力

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