Flow-Based Density Ratio Estimation for Intractable Distributions with Applications in Genomics
Egor Antipov, Alessandro Palma, Lorenzo Consoli, Stephan Günnemann, Andrea Dittadi, Fabian Theis
Abstract
Estimating density ratios between pairs of intractable data distributions is a core problem in probabilistic modeling, enabling principled comparisons of sample likelihoods under different data-generating processes across conditions. While exact-likelihood models such as normalizing flows offer a promising approach to density ratio estimation, naive evaluations are computationally expensive and prone to discretization errors because they require simulating each distribution's likelihood independently. In this work, we leverage condition-aware flow matching to derive a single dynamical formulation for tracking density ratios along generative trajectories. We demonstrate competitive performance on simulated benchmarks for closed-form ratio estimation, and show that our method supports versatile tasks in single-cell genomics data analysis, where likelihood-based comparisons of cellular states across experimental conditions enable treatment effect estimation and batch correction evaluation.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 0e8039cf-cc14-45df-a97a-51e6cae95f3dBuilds on15
- Score-Based Generative Modeling through Stochastic Differential EquationsYang Song, Jascha Sohl-Dickstein, Diederik P. Kingma, Abhishek Kumar et al.ICLR 2021 · 1,270 citations
- Universal Guidance for Diffusion ModelsArpit Bansal, Hong-Min Chu, Avi Schwarzschild, Soumyadip Sengupta et al.ICLR 2024 · 436 citations
- Torsional Diffusion for Molecular Conformer GenerationBowen Jing, Gabriele Corso, Jeffrey Chang, Regina Barzilay et al.NeurIPS 2022 · 413 citations
- Posterior Network: Uncertainty Estimation without OOD Samples via Density-Based Pseudo-CountsBertrand Charpentier, Daniel Zügner, Stephan GünnemannNeurIPS 2020 · 263 citations
- Telescoping Density-Ratio EstimationBenjamin Rhodes, Kai Xu, Michael U. GutmannNeurIPS 2020 · 148 citations
Related papers
- Density Ratio Estimation with Conditional Probability PathsHanlin Yu, Arto Klami, Aapo Hyvärinen, Anna Korba et al.ICML 2025
- TrajectoryNet: A Dynamic Optimal Transport Network for Modeling Cellular DynamicsAlexander Tong, Jessie Huang, Guy Wolf, David van Dijk et al.ICML 2020 · 257 citations
- Multi-Modal and Multi-Attribute Generation of Single Cells with CFGenAlessandro Palma, Till Richter, Hanyi Zhang, Manuel Lubetzki et al.ICLR 2025
- Structured Flow Autoencoders: Learning Structured Probabilistic Representations with Flow MatchingYidan Xu, Yixin Wang, XuanLong NguyenICLR 2026
- Wasserstein Flow Matching: Generative Modeling Over Families of DistributionsDoron Haviv, Aram-Alexandre Pooladian, Dana Pe'er, Brandon AmosICML 2025
