Incremental Sampling Without Replacement for Sequence Models
Kensen Shi, David Bieber, Charles Sutton
Abstract
Sampling is a fundamental technique, and sampling without replacement is often desirable when duplicate samples are not beneficial. Within machine learning, sampling is useful for generating diverse outputs from a trained model. We present an elegant procedure for sampling without replacement from a broad class of randomized programs, including generative neural models that construct outputs sequentially. Our procedure is efficient even for exponentially-large output spaces. Unlike prior work, our approach is incremental, i.e., samples can be drawn one at a time, allowing for increased flexibility. We also present a new estimator for computing expectations from samples drawn without replacement. We show that incremental sampling without replacement is applicable to many domains, e.g., program synthesis and combinatorial optimization.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Cited by top-tier papers9
- CrossBeam: Learning to Search in Bottom-Up Program SynthesisKensen Shi, Hanjun Dai, Kevin Ellis, Charles SuttonICLR 2022 · 28 citations
- Scaling Neural Program Synthesis with Distribution-Based SearchNathanaël Fijalkow, Guillaume Lagarde, Théo Matricon, Kevin Ellis et al.AAAI 2022 · 11 citations
- LambdaBeam: Neural Program Search with Higher-Order Functions and LambdasKensen Shi, Hanjun Dai, Wen-Ding Li, Kevin Ellis et al.NeurIPS 2023 · 8 citations
- Predictive Querying for Autoregressive Neural Sequence ModelsAlex Boyd, Samuel Showalter, Stephan Mandt, Padhraic SmythNeurIPS 2022 · 6 citations
- Towards Generalizable Multi-Policy Optimization with Self-Evolution for Job SchedulingInguk Choi, Woo-Jin Shin, Sang-Hyun Cho, Hyun-Jung KimNeurIPS 2025 · 4 citations
Builds on1
Related papers
- Improving Molecular Design by Stochastic Iterative Target AugmentationKevin Yang, Wengong Jin, Kyle Swanson, Regina Barzilay et al.ICML 2020 · 31 citations
- Estimating Gradients for Discrete Random Variables by Sampling without ReplacementWouter Kool, Herke van Hoof, Max WellingICLR 2020 · 59 citations
- Robust and Scalable SDE Learning: A Functional PerspectiveScott Alexander Cameron, Tyron Luke Cameron, Arnu Pretorius, Stephen J. RobertsICLR 2022 · 2 citations
- Foundation Posteriors for Approximate Probabilistic InferenceMike Wu, Noah D. GoodmanNeurIPS 2022 · 9 citations
- Outsourced Diffusion Sampling: Efficient Posterior Inference in Latent Spaces of Generative ModelsSiddarth Venkatraman, Mohsin Hasan, Minsu Kim, Luca Scimeca et al.ICML 2025
