Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

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.

Benjamin Rhodes, Kai Xu, Michael U. Gutmann• 2020

Related benchmarks

TaskDatasetResultRank
Density EstimationMNIST (test)
NLL (bits/dim)1.09
69
Density EstimationMNIST mixture 784 dim (test)
MSE2
11
Density Estimationrandom GMM 20 0.1 (dim 10) reuse sampling scheme 1.0 (test)
MSE72.1
6
Energy-based modelingAlanine dipeptide (test)
IID JS Divergence0.0093
5
Energy-based Density EstimationChignolin
IID JS Divergence0.0593
5
Showing 5 of 5 rows

Other info

Follow for update