Dequantified Diffusion-Schr{ö}dinger Bridge for Density Ratio Estimation
By: Wei Chen , Shigui Li , Jiacheng Li and more
Potential Business Impact:
Helps computers understand different data better.
Density ratio estimation is fundamental to tasks involving $f$-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 $\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.
Similar Papers
Ambiguous Medical Image Segmentation Using Diffusion Schrödinger Bridge
CV and Pattern Recognition
Helps doctors see tiny sickness spots better.
Diffusion Bridge Variational Inference for Deep Gaussian Processes
Machine Learning (CS)
Makes AI learn faster and better from data.
Dynamic Diffusion Schrödinger Bridge in Astrophysical Observational Inversions
Instrumentation and Methods for Astrophysics
Helps predict how stars form in space.