Telescoping Density-Ratio Estimation
About
Density-ratio estimation via classification is a cornerstone of unsupervised learning. It has provided the foundation for state-of-the-art methods in representation learning and generative modelling, with the number of use-cases continuing to proliferate. However, it suffers from a critical limitation: it fails to accurately estimate ratios p/q for which the two densities differ significantly. Empirically, we find this occurs whenever the KL divergence between p and q exceeds tens of nats. To resolve this limitation, we introduce a new framework, telescoping density-ratio estimation (TRE), that enables the estimation of ratios between highly dissimilar densities in high-dimensional spaces. Our experiments demonstrate that TRE can yield substantial improvements over existing single-ratio methods for mutual information estimation, representation learning and energy-based modelling.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Density Estimation | MNIST (test) | NLL (bits/dim)1.09 | 69 | |
| Density Estimation | MNIST mixture 784 dim (test) | MSE2 | 11 | |
| Density Estimation | random GMM 20 0.1 (dim 10) reuse sampling scheme 1.0 (test) | MSE72.1 | 6 | |
| Energy-based modeling | Alanine dipeptide (test) | IID JS Divergence0.0093 | 5 | |
| Energy-based Density Estimation | Chignolin | IID JS Divergence0.0593 | 5 |