Modelling Chemical Reaction Networks using Neural Ordinary Differential Equations
Anna C. M. Thöni, William E. Robinson, Yoram Bachrach, Wilhelm T. S. Huck, Tal Kachman
TL;DR
This work tackles the challenge that ODEs derived from chemical reaction networks (CRNs) via mass-action can be incomplete due to hidden dynamics. It introduces neural ODEs (nODEs) that augment the theoretical CRN vector field with a data-driven term, yielding a hybrid model $\dfrac{dy}{dt}=h_\kappa(t,y)+f_\theta(t,y)$ that better fits experimental concentration trajectories and reveals missing interactions. Through experiments on an oscillatory CRN, the authors show that nODEs improve predictive accuracy for both aperiodic and oscillatory data and enable interpretation of residual dynamics, including the transfer of learned dynamics across experiments. While the nODE often surpasses the purely theoretical model in period estimation and residual correction, it may not reliably classify oscillation regimes without more data, underscoring the value of jointly guiding experiments and model refinement. The approach provides a practical framework to diagnose model misspecification, quantify missing reactions, and inform the design of future CRNs with improved dynamic fidelity.
Abstract
In chemical reaction network theory, ordinary differential equations are used to model the temporal change of chemical species concentration. As the functional form of these ordinary differential equations systems is derived from an empirical model of the reaction network, it may be incomplete. Our approach aims to elucidate these hidden insights in the reaction network by combining dynamic modelling with deep learning in the form of neural ordinary differential equations. Our contributions not only help to identify the shortcomings of existing empirical models but also assist the design of future reaction networks.
