arXiv:2502.12902cs.LG2025-02被引 27

让神经算子学会预测不确定性,提升复杂系统预测可靠性

Probabilistic neural operators for functional uncertainty quantification

  • 用严格适当的评分规则构建生成模型,直接在训练中融合不确定性
  • 在不同任务和基线中均实现更优的不确定性量化表现
  • 兼容现有架构,适合物理建模与长期动态预测场景

神经算子可通过数据直接逼近微分方程系统的解算子,在多个领域表现出色。然而,模型与数据中的固有不确定性迄今很少被考虑,这对天气预报等混沌系统尤为关键。本文提出概率神经算子(PNO),一种在神经算子输出函数空间上学习概率分布的框架。PNO基于严格适当的评分规则引入生成建模,将不确定性信息直接整合进训练过程。我们提供了理论依据,并在多个领域验证其在不确定性量化上的性能提升。PNO仅需极少架构调整,多数概率预测任务表现更优,且对长时动态轨迹仍能生成校准良好、合理的不确定性表示。将其应用于大规模物理模型,可显著改善不确定性量化与极端事件识别,深化对代理模型预测能力的理解。

原文摘要 · Abstract (English)

Neural operators aim to approximate the solution operator of a system of differential equations purely from data. They have shown immense success in modeling complex dynamical systems across various domains. However, the occurrence of uncertainties inherent in both model and data has so far rarely been taken into account\textemdash{}a critical limitation in complex, chaotic systems such as weather forecasting. In this paper, we introduce the probabilistic neural operator (PNO), a framework for learning probability distributions over the output function space of neural operators. PNO extends neural operators with generative modeling based on strictly proper scoring rules, integrating uncertainty information directly into the training process. We provide a theoretical justification for the approach and demonstrate improved performance in quantifying uncertainty across different domains and with respect to different baselines. Furthermore, PNO requires minimal adjustment to existing architectures, shows improved performance for most probabilistic prediction tasks, and leads to well-calibrated predictive distributions and adequate uncertainty representations even for long dynamical trajectories. Implementing our approach into large-scale models for physical applications can lead to improvements in corresponding uncertainty quantification and extreme event identification, ultimately leading to a deeper understanding of the prediction of such surrogate models.

神经算子不确定性量化生成建模物理建模

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