mlsanityData to 7 Oct 2026

Count Bridges enable Modeling and Deconvolving Transcriptomic Data

arXiv 2603.04730 ↗First posted 5 Mar 2026cs.LGcited by 3

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

  1. 2024Generator Matching: Generative modeling with arbitrary Markov processes27 Oct
  2. 2024Discrete Flow Matching22 Jul
  3. 2024Dirichlet Flow Matching with Applications to DNA Sequence Design8 Feb
  4. 2024Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-Design7 Feb
  5. 2023Dirichlet Diffusion Score Model for Biological Sequence Generation18 May
  6. 2022Flow Matching for Generative Modeling6 Oct
  7. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow7 Sep

Other references · 32

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 8,964
  3. 2021Improved Denoising Diffusion Probabilistic Modelscited by 3,736
  4. 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,665
  5. 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
  6. 2024Simple and Effective Masked Diffusion Language Modelscited by 800
  7. 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 712
  8. 2024Simplified and Generalized Masked Diffusion for Discrete Datacited by 606
  9. 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modelingcited by 594
  10. 2021Tackling the Generative Learning Trilemma with Denoising Diffusion GANscited by 515
  11. 2025Dream 7B: Diffusion Large Language Modelscited by 515
  12. 2022All are Worth Words: A ViT Backbone for Diffusion Modelscited by 498
20 more
  1. 2022A Continuous Time Framework for Discrete Denoising Modelscited by 437
  2. 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Modelscited by 385
  3. 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Datacited by 371
  4. 2022Analog Bits: Generating Discrete Data using Diffusion Models with Self-Conditioningcited by 362
  5. 2022Analytic-DPM: an Analytic Estimate of the Optimal Reverse Variance in Diffusion Probabilistic Modelscited by 331
  6. 2013A survey of the Schr"odinger problem and some of its connections with optimal transportcited by 286
  7. 2021Likelihood Training of Schr"odinger Bridge using Forward-Backward SDEs Theorycited by 216
  8. 2024Scaling up Masked Diffusion Models on Textcited by 195
  9. 2025Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inferencecited by 173
  10. 2022Score-based Continuous-time Discrete Diffusion Modelscited by 167
  11. 2023Inversion by Direct Iteration: An Alternative to Denoising Diffusion for Image Restorationcited by 164
  12. 2021Solving Schr"odinger Bridges via Maximum Likelihoodcited by 140
  13. 2022Estimating the Optimal Covariance with Imperfect Mean in Diffusion Probabilistic Modelscited by 83
  14. 2023Non-Denoising Forward-Time Diffusionscited by 75
  15. 2024Learning Diffusion Priors from Observations by Expectation Maximizationcited by 60
  16. 2022From Denoising Diffusions to Denoising Markov Modelscited by 58
  17. 2023Blackout Diffusion: Generative Diffusion Models in Discrete-State Spacescited by 29
  18. 2025Distributional Diffusion Models with Scoring Rulescited by 25
  19. 2023DiSK: A Diffusion Model for Structured Knowledgecited by 16
  20. 2025Reverse Markov Learning: Multi-Step Generative Models for Complex Distributionscited by 7

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026CountsDiff: A Diffusion Model on the Natural Numbers for Generation and Imputation of Count-Based Data
  2. 2026Flow-Based Density Ratio Estimation for Intractable Distributions with Applications in Genomics
  3. 2026From Snapshots to Trajectories: Learning Single-Cell Gene Expression Dynamics via Conditional Flow Matching
  4. 2025Inference-Time Alignment in Diffusion Models with Reward-Guided Generation: Tutorial and Review
  5. 2026Bridge Matching Sampler: Scalable Sampling via Generalized Fixed-Point Diffusion Matching
  6. 2023Simulation-free Schr"odinger bridges via score and flow matching
  7. 2026Learning biophysical models of gene regulation with probability flow matching
  8. 2025Scalable Single-Cell Gene Expression Generation with Latent Diffusion Models