Neural Ordinary Differential Equations Inspired Parameterization of Kinetic Models

Paul van Lent,Olga Bunkova,Léon Planken,Joep Schmitz,Thomas Abeel
DOI: https://doi.org/10.1101/2024.12.20.629595
2024-12-21
Abstract:Motivation: Metabolic kinetic models are widely used to model biological systems. Despite their widespread use, it remains challenging to parameterize these Ordinary Differential Equations (ODE) for large scale kinetic models. Recent work on neural ODEs has shown the potential for modeling time-series data using neural networks, and many methodological developments in this field can similarly be applied to kinetic models. Results: We have implemented a simulation and training framework for Systems Biology Markup Language (SBML) models using JAX/Diffrax, which we named jaxkineticmodel. JAX allows for automatic differentiation and just-in-time compilation capabilities to speed up the parameterization of kinetic models. We show the robust capabilities of training kinetic models using this framework on a large collection of SBML models with different degrees of prior information on parameter initialization. Finally, we showcase the training framework implementation on a complex model of glycolysis. These results show that our framework can be used to fit large metabolic kinetic models efficiently and provides a strong platform for modeling biological systems. Implementation: Implementation of jaxkineticmodel is available as a Python package at https://github.com/AbeelLab/jaxkineticmodel.
Systems Biology
What problem does this paper attempt to address?