arXiv:2607.21644cs.LGcs.SY2026-07

用隐空间物理可观测量实现无需目标的偏微分方程控制

Toward Goal-Agnostic Joint-Embedding Predictive Control of Partial Differential Equations

论文配图:Toward Goal-Agnostic Joint-Embedding Predictive Control of Partial Differential Equations
图 1 · 摘自论文原文
  • 构建端到端联合嵌入架构,离线训练轻量视觉变压器与动作条件隐动态
  • 通过动能探测器优化控制,使轨迹复现精度达R²=0.989,RMSE降低53%
  • 适合需灵活、无目标控制的复杂系统建模与稳定化场景

我们提出一种面向偏微分方程(PDEs)的目标无关控制框架,基于端到端联合嵌入预测架构(JEPA)。一个轻量级2D视觉变压器(ViT)与动作条件隐动态在离线阶段训练,无需奖励或下游目标,随后冻结并由模型预测路径积分(MPPI)控制器复用。在隐空间最小化控制目标,初始采用L²距离,进一步展示在可得物理可观测量时重构目标的优势。通过最小化冻结隐状态轨迹上学习到的线性动能(KE)探测器的跟踪误差,我们证明可复现未见轨迹,达到R²=0.989,且无需修改底层世界模型。在2D Navier-Stokes基准任务中,使用KE探测器的MPPI规划将平均原生奖励从-12.08±0.86提升至-10.90±0.91(95%置信区间),同时降低最后四分之一速度场均方根误差(RMSE)从0.0765降至0.0692。在三个故意隐藏、差异大且非周期的目标上,KE规划使后期场RMSE相比隐空间L²规划降低53%(0.0220对0.0469),在30次成对比较中全胜。同一冻结模型还可通过直接调节动能实现稳态配置的稳定化,平均相对误差为2.7%。尽管隐空间探测器对测量噪声和缺失像素较敏感,但结果支持隐动态可保持灵活性与目标无关性,尤其当校准可观测量能保证唯一延拓时,适合作为状态控制目标。

原文摘要 · Abstract (English)

We present a goal-agnostic control framework for partial differential equations (PDEs) built around an end-to-end joint-embedding predictive architecture (JEPA). A lightweight 2D vision-transformer (ViT) and action-conditioned latent dynamics are trained offline without a reward or downstream goal, before being frozen and reused by a model-predictive path integral (MPPI) controller. We minimize a control objective in the latent space, initially expressed via the $L^2$ distance and additionally illustrate the benefit of recasting the control objective in terms of an explicit physical observable when available. By instead minimizing the tracking error for a learned linear kinetic-energy (KE) probe on the frozen latent-state rollouts, we demonstrate the ability to reproduce the control of held-out trajectories with $R^2=0.989$, while requiring no change to the underlying world model. For a controlled 2D Navier--Stokes benchmark, using a KE-probe within MPPI planning improves the mean native reward from $-12.08\pm0.86$ for latent-$L^2$ tracking to $-10.90\pm0.91$ (95\% CI), all while lowering last-quarter velocity-field RMSE from $0.0765$ to $0.0692$. Across three intentionally withheld, dissimilar, aperiodic targets, KE planning lowers late field RMSE by $53\%$ relative to latent-$L^2$ planning ($0.0220$ versus $0.0469$), winning across 30 paired comparisons. The same frozen model also supports stabilization around a steady-state configuration via direct regulation of KE, achieving $2.7\%$ mean relative error. While the latent probe proves brittle to measurement noise and missing pixels, our findings support the claim that latent dynamics can remain flexible and goal-agnostic, particularly when calibrated observables (granted they guarantee unique continuation) are a suitable objective for state control.

PDE控制隐空间控制动能探测模型预测

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