跳到正文
OpenReview · State space models· Romit Maulik; Vishwas Rao; Jiali Wang; Gianmarco Mengaldo; Emil Constantinescu; Bethany Lusch; Prasanna Balaprakash; Ian Foster; Rao Kotamarthi·· 2022-05-02精选AI 评分75

利用机器学习降阶模型实现高效高维变分数据同化

Efficient high-dimensional variational data assimilation with machine-learned reduced-order models

AI 导读

基于摘要分析。本文研究地球物理数值预报中高维变分数据同化的计算成本问题,以低维、数据驱动且可微的代理模型替代昂贵的数值前向模型,并通过自动微分加速初始条件优化;作者将该框架称为 AIEADA(Artificial Intelligence Emulator-Assisted Data Assimilation)。 机制上,变分数据同化在贝叶斯框架下结合模型预报先验与观测似然,以模型初始条件为决策变量求解优化问题。AIEADA 保留这一同化思路,但用学习得到的降阶代理模型执行前向预测,使目标函数对决策变量的梯度能够经自动微分获得。摘要未说明降阶表示、代理模型结构、训练数据规模及具体目标函数,尚不能将其归为某种特定状态空间网络。 在位势高度的代理模型辅助同化预报中,作者报告,相比传统基于方程的数据同化预报,该方法获得四个数量级的加速,使相关计算可在工作站而非专用高性能计算机上完成;作者还报告,相比不使用同化、仅由代理模型进行预报的方案,加入同化具有精度收益。摘要未给出具体精度指标、计算资源配置及加速是否计入代理模型训练成本,实验协议、消融及统计信息尚未验证。 平台推测(待验证):可迁移的假设是,以可微低维动力学代理模型结合观测更新,缓解 EEG 潜在状态估计中反复调用复杂前向模型的成本。可复用的是代理前向计算与自动微分优化流程;需改造的是 EEG 观测模型、潜在状态表示和噪声假设,并需具备足以学习动力学的数据及观测约束。低信噪比、跨被试差异、通道变化和小数据可能使代理误差在同化中累积。一个可证伪的小实验是在固定被试内划分下,比较同一代理模型有无同化时的留出时段预测误差与推理耗时;该假设不构成已验证的 BCI 效果,已有 EEG 相关工作尚未检索确认。

阅读价值

值得关注其以可微低维代理模型降低变分数据同化求解成本的机制;作者报告的加速幅度仍需结合计算口径、精度与训练成本核对,EEG/BCI 适用性尚未验证。

来源:OpenReview · State space models · openreview.net