Tracking large chemical reaction networks and rare events by neural networks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Weng, Jiayu, Zhu, Xinyi, Liu, Jing, Lü, Linyuan, Zhang, Pan, Tang, Ying
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915672303337472
author Weng, Jiayu
Zhu, Xinyi
Liu, Jing
Lü, Linyuan
Zhang, Pan
Tang, Ying
author_facet Weng, Jiayu
Zhu, Xinyi
Liu, Jing
Lü, Linyuan
Zhang, Pan
Tang, Ying
contents Chemical reaction networks are widely used to model stochastic dynamics in chemical kinetics, systems biology and epidemiology. Solving the chemical master equation that governs these systems poses a significant challenge due to the large state space exponentially growing with system sizes. The development of autoregressive neural networks offers a flexible framework for this problem; however, its efficiency is limited especially for high-dimensional systems and in scenarios with rare events. Here, we push the frontier of neural-network approach by exploiting faster optimizations such as natural gradient descent and time-dependent variational principle, achieving a 5- to 22-fold speedup, and by leveraging enhanced-sampling strategies to capture rare events. We demonstrate reduced computational cost and higher accuracy over the previous neural-network method in challenging reaction networks, including the mitogen-activated protein kinase (MAPK) cascade network, the hitherto largest biological network handled by the previous approaches of solving the chemical master equation. We further apply the approach to spatially extended reaction-diffusion systems, the Schlögl model with rare events, on two-dimensional lattices, beyond the recent tensor-network approach that handles one-dimensional lattices. The present approach thus enables efficient modeling of chemical reaction networks in general.
format Preprint
id arxiv_https___arxiv_org_abs_2512_10309
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Tracking large chemical reaction networks and rare events by neural networks
Weng, Jiayu
Zhu, Xinyi
Liu, Jing
Lü, Linyuan
Zhang, Pan
Tang, Ying
Molecular Networks
Machine Learning
Biological Physics
Chemical reaction networks are widely used to model stochastic dynamics in chemical kinetics, systems biology and epidemiology. Solving the chemical master equation that governs these systems poses a significant challenge due to the large state space exponentially growing with system sizes. The development of autoregressive neural networks offers a flexible framework for this problem; however, its efficiency is limited especially for high-dimensional systems and in scenarios with rare events. Here, we push the frontier of neural-network approach by exploiting faster optimizations such as natural gradient descent and time-dependent variational principle, achieving a 5- to 22-fold speedup, and by leveraging enhanced-sampling strategies to capture rare events. We demonstrate reduced computational cost and higher accuracy over the previous neural-network method in challenging reaction networks, including the mitogen-activated protein kinase (MAPK) cascade network, the hitherto largest biological network handled by the previous approaches of solving the chemical master equation. We further apply the approach to spatially extended reaction-diffusion systems, the Schlögl model with rare events, on two-dimensional lattices, beyond the recent tensor-network approach that handles one-dimensional lattices. The present approach thus enables efficient modeling of chemical reaction networks in general.
title Tracking large chemical reaction networks and rare events by neural networks
topic Molecular Networks
Machine Learning
Biological Physics
url https://arxiv.org/abs/2512.10309