Reconstructing Noisy Gene Regulation Dynamics Using Extrinsic-Noise-Driven Neural Stochastic Differential Equations

Abstract

Proper regulation of cell signaling and gene expression is crucial for maintaining cellular function, development, and adaptation to environmental changes. Reaction dynamics in cell populations is often noisy because of (i) inherent stochasticity of intracellular biochemical reactions (``intrinsic noise'') and (ii) heterogeneity of cellular states across different cells that are influenced by external factors (``extrinsic noise''). In this work, we introduce an extrinsic-noise-driven neural stochastic differential equation (END-nSDE) framework that utilizes the Wasserstein distance to accurately reconstruct SDEs from trajectory data from a heterogeneous population of cells (extrinsic noise). We demonstrate the effectiveness of our approach using both simulated and experimental data from three different systems in cell biology: (i) circadian rhythms, (ii) RPA-DNA binding dynamics, and (iii) NF$\kappa$B signaling process. Our END-nSDE reconstruction method can model how cellular heterogeneity (extrinsic noise) modulates reaction dynamics in the presence of intrinsic noise. It also outperforms existing time-series analysis methods such as recurrent neural networks (RNNs) and long short-term memory networks (LSTMs). By inferring cellular heterogeneities from data, our END-nSDE reconstruction method can reproduce noisy dynamics observed in experiments. In summary, the reconstruction method we propose offers a useful surrogate modeling approach for complex biophysical processes, where high-fidelity mechanistic models may be impractical.

References in corpus (12)

• Pooling single-cell recordings: Scalable inference through heterogeneous kinetics
• Implicit energy regularization of neural ordinary-differential-equation control
• Learning force fields from stochastic trajectories
• Metastability in a stochastic neural network modeled as a velocity jump Markov process
• Neural Ordinary Differential Equation Control of Dynamics on Graphs
• Interpretable Polynomial Neural Ordinary Differential Equations
• Quantifying extrinsic noise in gene expression using the maximum entropy framework
• Near-optimal control of dynamical systems with neural ordinary differential equations
• Surrogate Modeling and Control of Medical Digital Twins
• NN2Poly: A polynomial representation for deep feed-forward artificial neural networks
• Visualizing high-dimensional loss landscapes with Hessian directions
• Kinetic theories of state- and generation-dependent cell populations