Count Bridges enable Modeling and Deconvolving Transcriptomic Data
On the map: Diffusion models → Flow matching, time series and science
Abstract (arXiv)
Many modern biological assays, including RNA sequencing, yield integer-valued counts that reflect the number of molecules detected. These measurements are often not at the desired resolution: while the unit of interest is typically a single cell, many measurement technologies produce counts aggregated over sets of cells. Although recent generative frameworks such as diffusion and flow matching have been extended to non-Euclidean and discrete settings, it remains unclear how best to model integer-valued data or how to systematically deconvolve aggregated observations. We introduce Count Bridges, a stochastic bridge process on the integers that provides an exact, tractable analogue of diffusion-style models for count data, with closed-form conditionals for efficient training and sampling. We extend this framework to enable direct training from aggregated measurements via an Expectation-Maximization-style approach that treats unit-level counts as latent variables. We demonstrate state-of-the-art performance on integer distribution matching benchmarks, comparing against flow matching and discrete flow matching baselines across various metrics. We then apply Count Bridges to two large-scale problems in biology: modeling single-cell gene expression data at the nucleotide resolution, with applications to deconvolving bulk RNA-seq, and resolving multicellular spatial transcriptomic spots into single-cell count profiles. Our methods offer a principled foundation for generative modeling and deconvolution of biological count data across scales and modalities.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Flow matching, time series and science, one of 14 directions in Diffusion models. Builds on 39 of its 65 references that are in the map. First posted on arXiv 5 Mar 2026.
Builds on
39 of 65 references.
Closest ancestors · 7
Same direction, newest first
- 2024Generator Matching: Generative modeling with arbitrary Markov processes27 Oct
- 2024Discrete Flow Matching22 Jul
- 2024Dirichlet Flow Matching with Applications to DNA Sequence Design8 Feb
- 2024Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-Design7 Feb
- 2023Dirichlet Diffusion Score Model for Biological Sequence Generation18 May
- 2022Flow Matching for Generative Modeling6 Oct
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow7 Sep
Other references · 32
- 2020Denoising Diffusion Probabilistic Modelscited by 20,851
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
- 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
- 2024Simple and Effective Masked Diffusion Language Modelscited by 800
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 712
- 2024Simplified and Generalized Masked Diffusion for Discrete Datacited by 606
- 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modelingcited by 594
- 2021Tackling the Generative Learning Trilemma with Denoising Diffusion GANscited by 515
- 2025Dream 7B: Diffusion Large Language Modelscited by 515
- 2022All are Worth Words: A ViT Backbone for Diffusion Modelscited by 498
20 more
- 2022A Continuous Time Framework for Discrete Denoising Modelscited by 437
- 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Modelscited by 385
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Datacited by 371
- 2022Analog Bits: Generating Discrete Data using Diffusion Models with Self-Conditioningcited by 362
- 2022Analytic-DPM: an Analytic Estimate of the Optimal Reverse Variance in Diffusion Probabilistic Modelscited by 331
- 2013A survey of the Schr"odinger problem and some of its connections with optimal transportcited by 286
- 2021Likelihood Training of Schr"odinger Bridge using Forward-Backward SDEs Theorycited by 216
- 2024Scaling up Masked Diffusion Models on Textcited by 195
- 2025Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inferencecited by 173
- 2022Score-based Continuous-time Discrete Diffusion Modelscited by 167
- 2023Inversion by Direct Iteration: An Alternative to Denoising Diffusion for Image Restorationcited by 164
- 2021Solving Schr"odinger Bridges via Maximum Likelihoodcited by 140
- 2022Estimating the Optimal Covariance with Imperfect Mean in Diffusion Probabilistic Modelscited by 83
- 2023Non-Denoising Forward-Time Diffusionscited by 75
- 2024Learning Diffusion Priors from Observations by Expectation Maximizationcited by 60
- 2022From Denoising Diffusions to Denoising Markov Modelscited by 58
- 2023Blackout Diffusion: Generative Diffusion Models in Discrete-State Spacescited by 29
- 2025Distributional Diffusion Models with Scoring Rulescited by 25
- 2023DiSK: A Diffusion Model for Structured Knowledgecited by 16
- 2025Reverse Markov Learning: Multi-Step Generative Models for Complex Distributionscited by 7
Built on by
Citing papers per year: 2026: 3.
3 papers.
- 2026Flow Matching for Count Datacited by 2
- 2026Stochastic Flow Map for Count Datacited by 0
- 2026CRNDiff: Count-Native Diffusion Framework via Chemical Reaction Networkscited by 0
- 2026CRNDiff: Count-Native Diffusion Framework via Chemical Reaction Networks25 Sep 2026
- 2026Stochastic Flow Map for Count Data20 Sep 2026
- 2026Flow Matching for Count Data8 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026CountsDiff: A Diffusion Model on the Natural Numbers for Generation and Imputation of Count-Based Data
- 2026Flow-Based Density Ratio Estimation for Intractable Distributions with Applications in Genomics
- 2026From Snapshots to Trajectories: Learning Single-Cell Gene Expression Dynamics via Conditional Flow Matching
- 2025Inference-Time Alignment in Diffusion Models with Reward-Guided Generation: Tutorial and Review
- 2026Bridge Matching Sampler: Scalable Sampling via Generalized Fixed-Point Diffusion Matching
- 2023Simulation-free Schr"odinger bridges via score and flow matching
- 2026Learning biophysical models of gene regulation with probability flow matching
- 2025Scalable Single-Cell Gene Expression Generation with Latent Diffusion Models