arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.13616cs.AIcs.LG
收藏

DiffusionRollout:长时间尺度PDE求解中的不确定性感知 rollout 计划

DiffusionRollout: Uncertainty-Aware Rollout Planning in Long-Horizon PDE Solving

Seungwoo Yoo, Juil Koo, Daehyeon Choi, Minhyuk Sung

AI总结:

DiffusionRollout通过自适应选择步长策略,提升长时间尺度PDE求解的预测可靠性与准确性。

摘要:

我们提出了DiffusionRollout,一种针对自回归扩散模型的新型选择性rollout规划策略,旨在减轻物理系统由偏微分方程(PDEs) govern 的长时间预测中的误差累积。基于最近已验证的PDE求解概率方法,我们进一步探索其量化预测不确定性的能力,并展示了预测误差与多个样本计算出的标准差之间的强相关性,支持其作为模型预测信心的代理。基于这一观察,我们引入了一种机制,在自回归rollout过程中自适应地选择步长,通过减少对不准确先验输出的依赖,提高长期预测的可靠性。在长时间轨迹PDE预测基准上的广泛评估验证了所提出不确定性度量和自适应规划策略的有效性,这体现在更低的预测误差和更长的预测轨迹,这些轨迹与真实值保持高相关性。

英文摘要:

We propose DiffusionRollout, a novel selective rollout planning strategy for autoregressive diffusion models, aimed at mitigating error accumulation in long-horizon predictions of physical systems governed by partial differential equations (PDEs). Building on the recently validated probabilistic approach to PDE solving, we further explore its ability to quantify predictive uncertainty and demonstrate a strong correlation between prediction errors and standard deviations computed over multiple samples-supporting their use as a proxy for the model's predictive confidence. Based on this observation, we introduce a mechanism that adaptively selects step sizes during autoregressive rollouts, improving long-term prediction reliability by reducing the compounding effect of conditioning on inaccurate prior outputs. Extensive evaluation on long-trajectory PDE prediction benchmarks validates the effectiveness of the proposed uncertainty measure and adaptive planning strategy, as evidenced by lower prediction errors and longer predicted trajectories that retain a high correlation with their ground truths.

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

↑