Uncertainty Propagation Networks for Neural Ordinary Differential Equations

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Jahanshahi, Hadi, Zhu, Zheng H.
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866918352136437760
author Jahanshahi, Hadi
Zhu, Zheng H.
author_facet Jahanshahi, Hadi
Zhu, Zheng H.
contents This paper introduces Uncertainty Propagation Network (UPN), a novel family of neural differential equations that naturally incorporate uncertainty quantification into continuous-time modeling. Unlike existing neural ODEs that predict only state trajectories, UPN simultaneously model both state evolution and its associated uncertainty by parameterizing coupled differential equations for mean and covariance dynamics. The architecture efficiently propagates uncertainty through nonlinear dynamics without discretization artifacts by solving coupled ODEs for state and covariance evolution while enabling state-dependent, learnable process noise. The continuous-depth formulation adapts its evaluation strategy to each input's complexity, provides principled uncertainty quantification, and handles irregularly-sampled observations naturally. Experimental results demonstrate UPN's effectiveness across multiple domains: continuous normalizing flows (CNFs) with uncertainty quantification, time-series forecasting with well-calibrated confidence intervals, and robust trajectory prediction in both stable and chaotic dynamical systems.
format Preprint
id arxiv_https___arxiv_org_abs_2508_16815
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Uncertainty Propagation Networks for Neural Ordinary Differential Equations
Jahanshahi, Hadi
Zhu, Zheng H.
Machine Learning
I.2.6; G.3
This paper introduces Uncertainty Propagation Network (UPN), a novel family of neural differential equations that naturally incorporate uncertainty quantification into continuous-time modeling. Unlike existing neural ODEs that predict only state trajectories, UPN simultaneously model both state evolution and its associated uncertainty by parameterizing coupled differential equations for mean and covariance dynamics. The architecture efficiently propagates uncertainty through nonlinear dynamics without discretization artifacts by solving coupled ODEs for state and covariance evolution while enabling state-dependent, learnable process noise. The continuous-depth formulation adapts its evaluation strategy to each input's complexity, provides principled uncertainty quantification, and handles irregularly-sampled observations naturally. Experimental results demonstrate UPN's effectiveness across multiple domains: continuous normalizing flows (CNFs) with uncertainty quantification, time-series forecasting with well-calibrated confidence intervals, and robust trajectory prediction in both stable and chaotic dynamical systems.
title Uncertainty Propagation Networks for Neural Ordinary Differential Equations
topic Machine Learning
I.2.6; G.3
url https://arxiv.org/abs/2508.16815