English

Dequantified Diffusion-Schr{\"o}dinger Bridge for Density Ratio Estimation

Machine Learning 2025-11-04 v5 Machine Learning

Abstract

Density ratio estimation is fundamental to tasks involving ff-divergences, yet existing methods often fail under significantly different distributions or inadequately overlapping supports -- the density-chasm and the support-chasm problems. Additionally, prior approaches yield divergent time scores near boundaries, leading to instability. We design D3RE\textbf{D}^3\textbf{RE}, a unified framework for \textbf{robust}, \textbf{stable} and \textbf{efficient} density ratio estimation. We propose the dequantified diffusion bridge interpolant (DDBI), which expands support coverage and stabilizes time scores via diffusion bridges and Gaussian dequantization. Building on DDBI, the proposed dequantified Schr{\"o}dinger bridge interpolant (DSBI) incorporates optimal transport to solve the Schr{\"o}dinger bridge problem, enhancing accuracy and efficiency. Our method offers uniform approximation and bounded time scores in theory, and outperforms baselines empirically in mutual information and density estimation tasks.

Keywords

Cite

@article{arxiv.2505.05034,
  title  = {Dequantified Diffusion-Schr{\"o}dinger Bridge for Density Ratio Estimation},
  author = {Wei Chen and Shigui Li and Jiacheng Li and Junmei Yang and John Paisley and Delu Zeng},
  journal= {arXiv preprint arXiv:2505.05034},
  year   = {2025}
}
R2 v1 2026-06-28T23:25:26.910Z