Optimal Budgeted Rejection Sampling for Generative Models
Not on the map. Why?
Abstract (arXiv)
Rejection sampling methods have recently been proposed to improve the performance of discriminator-based generative models. However, these methods are only optimal under an unlimited sampling budget, and are usually applied to a generator trained independently of the rejection procedure. We first propose an Optimal Budgeted Rejection Sampling (OBRS) scheme that is provably optimal with respect to textitany f-divergence between the true distribution and the post-rejection distribution, for a given sampling budget. Second, we propose an end-to-end method that incorporates the sampling scheme into the training procedure to further enhance the model's overall performance. Through experiments and supporting theory, we show that the proposed methods are effective in significantly improving the quality and diversity of the samples.
Key facts. Cited by 5 papers in the map (data through 14 Oct 2026). Builds on 19 of its 26 references that are in the map. First posted on arXiv 1 Nov 2023.
Builds on
19 of 26 references.
References · 19
- 2022Generative Adversarial Networkscited by 5,627
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 3,045
- 2018Large Scale GAN Training for High Fidelity Natural Image Synthesiscited by 2,626
- 2017Visualizing the Loss Landscape of Neural Netscited by 1,130
- 2019Improved Precision and Recall Metric for Assessing Generative Modelscited by 944
- 2016f-GAN: Training Generative Neural Samplers using Variational Divergence Minimizationcited by 506
- 2020Reliable Fidelity and Diversity Metrics for Generative Modelscited by 434
- 2018Assessing Generative Models via Precision and Recallcited by 406
- 2016Adversarially Learned Inferencecited by 216
- 2022Refining Generative Process with Discriminator Guidance in Score-based Diffusion Modelscited by 100
- 2020Your GAN is Secretly an Energy-based Model and You Should use Discriminator Driven Latent Samplingcited by 71
- 2018Discriminator Rejection Samplingcited by 49
7 more
- 2020Refining Deep Generative Models via Discriminator Gradient Flowcited by 45
- 2021Resampling Base Distributions of Normalizing Flowscited by 42
- 2018Metropolis-Hastings Generative Adversarial Networkscited by 41
- 2019Discriminator optimal transportcited by 25
- 2023TopP&R: Robust Support Estimation Approach for Evaluating Fidelity and Diversity in Generative Modelscited by 14
- 2023Precision-Recall Divergence Optimization for Generative Modeling with GANs and Normalizing Flowscited by 9
- 2021Latent reweighting, an almost free improvement for GANscited by 6
Built on by
Citing papers per year: 2025: 2; 2026: 3.
5 papers.
- 2025Improving Diversity in Language Models: When Temperature Fails, Change the Losscited by 7
- 2026Jackpot: Optimal Budgeted Rejection Sampling for Extreme Actor-Policy Mismatch Reinforcement Learningcited by 5
- 2025Improving Discriminator Guidance in Diffusion Modelscited by 2
- 2026Multimodal Diffusion to Mutually Enhance Polarized Light and Low Resolution EBSD Datacited by 0
- 2026Implicit Variational Rejection Samplingcited by 0
- 2026Implicit Variational Rejection Sampling12 Jun 2026
- 2026Multimodal Diffusion to Mutually Enhance Polarized Light and Low Resolution EBSD Data24 Apr 2026
- 2026Jackpot: Optimal Budgeted Rejection Sampling for Extreme Actor-Policy Mismatch Reinforcement Learning5 Feb 2026
- 2025Improving Diversity in Language Models: When Temperature Fails, Change the Loss13 Aug 2025
- 2025Improving Discriminator Guidance in Diffusion Models20 Mar 2025