Propensity Straight-Through Gradients for Discrete Stochastic Systems

📅 2026-08-26
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
研究解决了连续时间马尔可夫链与梯度机器学习结合的问题,通过区分归一化反应倾向性来获得精确一步条件均值敏感性,定义了倾向直通估计器。
📝 Abstract
Continuous-time Markov chains (CTMCs) provide the backbone for modeling discrete stochastic dynamics across applied, physical, and biological sciences. Their integration with modern gradient-based machine learning, however, is limited by the hard categorical event selection intrinsic to Gillespie-type simulation algorithms. We exploit the affine state update to obtain the exact one-step conditional-mean sensitivity by differentiating normalized reaction propensities. We pair this backward rule with exact forward trajectories to define the propensity straight-through (PST) estimator. At the trajectory level, we show that one-step sensitivities composed across events can depart from the exact multistep sensitivity. We derive the resulting per-step discrepancy in closed form and prove that it vanishes identically for affine downstream dependence. PST matches the accuracy of Gumbel-Softmax straight-through across all benchmarks: reversible dimerization (0.06% error), a genetic oscillator (1.7% error), a 50-task repressilator suite (0.17% median error), and patch-clamp ion-channel recordings ($R^2$ = 0.988). Under matched settings, PST converges 3.0-fold faster on the oscillator and 2.1-fold faster on the ion channel. At deep-learning scale, PST trains a 203,796-parameter stochastic reaction network with hard sampling, reaching 98.22% MNIST digit classification accuracy. By differentiating an exact conditional mean rather than a relaxed sample, PST offers a temperature- and Gumbel-free path to scalable gradient-based learning through exact stochastic trajectories.
Problem

Research questions and friction points this paper is trying to address.

Continuous-time Markov chains
Gradient-based machine learning
Stochastic dynamics
Innovation

Methods, ideas, or system contributions that make the work stand out.

Propensity Straight-Through
Continuous-time Markov chains
Gradient-based learning
Stochastic systems
Exact conditional mean
🔎 Similar Papers
💼 Related Jobs
No related jobs found.
J
Jose M. G. Vilar
Biofisika Institute (CSIC, UPV/EHU), University of the Basque Country, P.O. Box 644, 48080 Bilbao, Spain; IKERBASQUE, Basque Foundation for Science, 48011 Bilbao, Spain
L
Leonor Saiz
Department of Biomedical Engineering, University of California, 451 East Health Sciences Drive, Davis, CA 95616, USA