Optimal Variance Control of the Score-Function Gradient Estimator for Importance Weighted Bounds

Publikation: Bidrag til tidsskriftKonferenceartikelForskningfagfællebedømt

This paper introduces novel results for the score function gradient estimator of the importance weighted variational bound (IWAE). We prove that in the limit of large K (number of importance samples) one can choose the control variate such that the Signal-to-Noise ratio (SNR) of the estimator grows as pK. This is in contrast to the standard pathwise gradient estimator where the SNR decreases as 1/pK. Based on our theoretical findings we develop a novel control variate that extends on VIMCO. Empirically, for the training of both continuous and discrete generative models, the proposed method yields superior variance reduction, resulting in an SNR for IWAE that increases with K without relying on the reparameterization trick. The novel estimator is competitive with state-of-the-art reparameterization-free gradient estimators such as Reweighted Wake-Sleep (RWS) and the thermodynamic variational objective (TVO) when training generative models.

OriginalsprogEngelsk
TidsskriftAdvances in Neural Information Processing Systems
Vol/bind2020-December
Antal sider12
ISSN1049-5258
StatusUdgivet - 2020
Begivenhed34th Conference on Neural Information Processing Systems, NeurIPS 2020 - Virtual, Online
Varighed: 6 dec. 202012 dec. 2020

Konference

Konference34th Conference on Neural Information Processing Systems, NeurIPS 2020
ByVirtual, Online
Periode06/12/202012/12/2020
SponsorApple, et al., Microsoft, PDT Partners, Sony Group Corporation, Tenstorrent

Bibliografisk note

Funding Information:
The PhD program supporting Valentin Liévin is partially funded by Google. supported by the NVIDIA Corporation with the donation of GPUs.

Publisher Copyright:
© 2020 Neural information processing systems foundation. All rights reserved.

ID: 276208639