mlsanityData to 7 Oct 2026

A hitchhiker's guide to Poisson gradient estimation

arXiv 2602.03896 ↗First posted 3 Feb 2026stat.ML cs.LG q-bio.NCcited by 1

Not on the map. Why?

Abstract (arXiv)

Poisson-distributed latent variable models are widely used in computational neuroscience, but differentiating through discrete stochastic samples remains challenging. Two approaches address this: *Exponential Arrival Time* (EAT) simulation and *Gumbel-SoftMax* (GSM) relaxation. We provide the first systematic comparison of these methods, along with practical guidance for practitioners. Our main technical contribution is a modification to the EAT method that theoretically guarantees an unbiased first moment (exactly matching the firing rate), and reduces second-moment bias. We evaluate these methods on their distributional fidelity, gradient quality, and performance on two tasks: (1) variational autoencoders with Poisson latents, and (2) partially observable generalized linear models, where latent neural connectivity must be inferred from observed spike trains. Across all metrics, our modified EAT method exhibits better overall performance (often comparable to exact gradients), and substantially higher robustness to hyperparameter choices. These results extend to over-dispersed Negative Binomial latents, where modified EAT again performs best. However, only GSM generalizes to arbitrary non-Poisson distributions, including the under-dispersed regime. Together, our results clarify the trade-offs between these methods and offer concrete recommendations for practitioners working with Poisson latent variable models.

Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 14 of its 47 references that are in the map. First posted on arXiv 3 Feb 2026.

Builds on

14 of 47 references.

References · 14

  1. 2013Estimating or Propagating Gradients Through Stochastic Neurons for Conditional Computationcited by 2,031
  2. 2016Variational Inference: A Review for Statisticianscited by 1,466
  3. 2016The Concrete Distribution: A Continuous Relaxation of Discrete Random Variablescited by 1,135
  4. 2015Gradient Estimation Using Stochastic Computation Graphscited by 120
  5. 2018Iterative Amortized Inferencecited by 93
  6. 2018Semi-Amortized Variational Autoencoderscited by 83
  7. 2014Reweighted Wake-Sleepcited by 66
  8. 2024Poisson Variational Autoencodercited by 11
  9. 2022Markov Chain Score Ascent: A Unifying Framework of Variational Inference with Markovian Gradientscited by 6
  10. 2024Brain-like Variational Inferencecited by 6
  11. 2023Forward chi2 Divergence Based Variational Importance Samplingcited by 4
  12. 2024A Differentiable Partially Observable Generalized Linear Model with Forward-Backward Message Passingcited by 3
2 more
  1. 2025Negative Binomial Variational Autoencoders for Overdispersed Latent Modelingcited by 2
  2. 2025Inferring response times of perceptual decisions with Poisson variational autoencoderscited by 2

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 1.

1 paper.