让扩散模型生成更安全的微分方程控制方案,兼顾安全与性能。
From Uncertain to Safe: Conformal Adaptation of Diffusion Models for Safe PDE Control
- 用置信区间量化模型不确定性,指导训练和推理过程。
- 在三个真实物理控制任务中,唯一满足全部安全约束的方法。
- 适合需要高可靠性控制的科学计算与工程应用。
深度学习在偏微分方程(PDE)约束控制中的应用日益受到关注,但现有方法普遍忽视实际应用中至关重要的安全要求。为此,我们提出安全扩散模型用于PDE控制(SafeDiffCon),通过置信区间量化模型不确定性,在训练后阶段和推理阶段实现安全约束下的最优控制。首先,该方法对预训练扩散模型进行后训练,利用基于合规范式预测估计的不确定性置信区间,重构扩散损失函数以生成更符合安全约束的控制序列,同时提升控制目标表现;其次,在推理阶段,扩散模型通过迭代引导与参数微调,动态调整生成过程与内部参数,条件于控制目标并融合估计的不确定性置信区间。我们在一维伯格斯方程、二维不可压缩流体及可控核聚变问题三个控制任务上评估了SafeDiffCon。结果表明,SafeDiffCon是唯一满足所有安全约束的方法,而其他经典与深度学习基线均失败。同时,在保障安全的前提下,其控制性能最优。代码见:https://github.com/AI4Science-WestlakeU/safediffcon。
原文摘要 · Abstract (English)
The application of deep learning for partial differential equation (PDE)-constrained control is gaining increasing attention. However, existing methods rarely consider safety requirements crucial in real-world applications. To address this limitation, we propose Safe Diffusion Models for PDE Control (SafeDiffCon), which introduce the uncertainty quantile as model uncertainty quantification to achieve optimal control under safety constraints through both post-training and inference phases. Firstly, our approach post-trains a pre-trained diffusion model to generate control sequences that better satisfy safety constraints while achieving improved control objectives via a reweighted diffusion loss, which incorporates the uncertainty quantile estimated using conformal prediction. Secondly, during inference, the diffusion model dynamically adjusts both its generation process and parameters through iterative guidance and fine-tuning, conditioned on control targets while simultaneously integrating the estimated uncertainty quantile. We evaluate SafeDiffCon on three control tasks: 1D Burgers' equation, 2D incompressible fluid, and controlled nuclear fusion problem. Results demonstrate that SafeDiffCon is the only method that satisfies all safety constraints, whereas other classical and deep learning baselines fail. Furthermore, while adhering to safety constraints, SafeDiffCon achieves the best control performance. The code can be found at https://github.com/AI4Science-WestlakeU/safediffcon.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。