ICODE: Modeling Dynamical Systems with Extrinsic Input Information
Zhaoyi Li, Wenjie Mei, Ke Yu, Yang Bai, Shihua Li
TL;DR
The paper introduces Input Concomitant Neural ODEs (ICODEs), a contraction-guaranteed neural ODE framework that directly incorporates real-time external inputs via an affine-in-input dynamics $\dot{x} = f_0(x) + \sum_j g_j(x) u_j$. It provides sufficient contraction conditions using a uniform metric $M(x) = L(x)^T L(x)$ and the matrix $R = L J(x,u) L^{-1} + \dot{L} L^{-1}$ to ensure exponential convergence of trajectories, regardless of initialization. Through extensive experiments on mechanical, electrical, chemical, and PDE systems, including 1D/2D heat conduction and R-F dynamics, ICODEs consistently outperform NODE-based baselines (NODE, CDE, ANODE), especially under atypical or rapidly changing inputs, with Transformer-enhanced variants further boosting performance. The work demonstrates a practical and robust approach for learning physically informed dynamical systems with external inputs, offering stability guarantees and scalability for applications in robotics, physics, and control engineering.
Abstract
Learning models of dynamical systems with external inputs, which may be, for example, nonsmooth or piecewise, is crucial for studying complex phenomena and predicting future state evolution, which is essential for applications such as safety guarantees and decision-making. In this work, we introduce \emph{Input Concomitant Neural ODEs (ICODEs)}, which incorporate precise real-time input information into the learning process of the models, rather than treating the inputs as hidden parameters to be learned. The sufficient conditions to ensure the model's contraction property are provided to guarantee that system trajectories of the trained model converge to a fixed point, regardless of initial conditions across different training processes. We validate our method through experiments on several representative real dynamics: Single-link robot, DC-to-DC converter, motion dynamics of a rigid body, Rabinovich-Fabrikant equation, Glycolytic-glycogenolytic pathway model, and heat conduction equation. The experimental results demonstrate that our proposed ICODEs efficiently learn the ground truth systems, achieving superior prediction performance under both typical and atypical inputs. This work offers a valuable class of neural ODE models for understanding physical systems with explicit external input information, with potentially promising applications in fields such as physics and robotics. Our code is available online at https://github.com/EEE-ai59/ICODE.git.
