Backpropagation through the Void: Optimizing control variates for black-box gradient estimation
Not on the map. Why?
Abstract (arXiv)
Gradient-based optimization is the foundation of deep learning and reinforcement learning. Even when the mechanism being optimized is unknown or not differentiable, optimization using high-variance or biased gradient estimates is still often the best strategy. We introduce a general framework for learning low-variance, unbiased gradient estimators for black-box functions of random variables. Our method uses gradients of a neural network trained jointly with model parameters or policies, and is applicable in both discrete and continuous settings. We demonstrate this framework for training discrete latent-variable models. We also give an unbiased, action-conditional extension of the advantage actor-critic reinforcement learning algorithm.
Key facts. Cited by 106 papers in the map (data through 14 Oct 2026). First posted on arXiv 31 Oct 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 2; 2020: 0; 2021: 28; 2022: 22; 2023: 20; 2024: 15; 2025: 8; 2026: 11.
106 papers.
Sort:
By area:
- 2024Black-Box Access is Insufficient for Rigorous AI Auditscited by 126
- 2022GFlowNets and variational inferencecited by 106
- 2021Implicit MLE: Backpropagating Through Discrete Exponential Family Distributionscited by 98
- 2023Soft Merging of Experts with Adaptive Routingcited by 90
- 2021A Review of the Gumbel-max Trick and its Extensions for Discrete Stochasticity in Machine Learningcited by 82
- 2022Ordered Subgraph Aggregation Networkscited by 57
- 2023Learning to Maximize Mutual Information for Dynamic Feature Selectioncited by 56
- 2021Smooth Normalizing Flowscited by 53
- 2021Variational Causal Networks: Approximate Bayesian Inference over Causal Structurescited by 43
- 2021RieszNet and ForestRiesz: Automatic Debiased Machine Learning with Neural Nets and Random Forestscited by 42
- 2022SIMPLE: A Gradient Estimator for k-Subset Samplingcited by 34
- 2023Bridging Discrete and Backpropagation: Straight-Through and Beyondcited by 31
38 more
- 2022Automatic Differentiation of Programs with Discrete Randomnesscited by 29
- 2021Direct Differentiable Augmentation Searchcited by 28
- 2021Multi-Facet Clustering Variational Autoencoderscited by 28
- 2023Neural Probabilistic Logic Programming in Discrete-Continuous Domainscited by 25
- 2021Mind Mappings: Enabling Efficient Algorithm-Accelerator Mapping Space Searchcited by 24
- 2022Gradient Estimation with Discrete Stein Operatorscited by 23
- 2023Efficient Attention via Control Variatescited by 23
- 2021CADDA: Class-wise Automatic Differentiable Data Augmentation for EEG Signalscited by 22
- 2022Sparse Graph Learning from Spatiotemporal Time Seriescited by 22
- 2021Self-Conditioned Probabilistic Learning of Video Rescalingcited by 19
- 2022SABLAS: Learning Safe Control for Black-box Dynamical Systemscited by 17
- 2023Sparse Backpropagation for MoE Trainingcited by 17
- 2021Learning Generalized Gumbel-max Causal Mechanismscited by 16
- 2024Imperative Learning: A Self-supervised Neuro-Symbolic Learning Framework for Robot Autonomycited by 16
- 2021Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learningcited by 15
- 2022Risk-Sensitive Reinforcement Learning with Exponential Criteriacited by 15
- 2021An overview of mixing augmentation methods and augmentation strategiescited by 14
- 2022Neural Set Function Extensions: Learning with Discrete Functions in High Dimensionscited by 14
- 2023Differentiable Sampling of Categorical Distributions Using the CatLog-Derivative Trickcited by 14
- 2021A unified view of likelihood ratio and reparameterization gradientscited by 12
- 2022A Survey of Automated Data Augmentation Algorithms for Deep Learning-based Image Classification Taskscited by 12
- 2022Adaptive Perturbation-Based Gradient Estimation for Discrete Latent Variable Modelscited by 12
- 2025Ideas in Inference-time Scaling can Benefit Generative Pre-training Algorithmscited by 12
- 2021Storchastic: A Framework for General Stochastic Automatic Differentiationcited by 11
- 2023GeoPhy: Differentiable Phylogenetic Inference via Geometric Gradients of Tree Topologiescited by 10
- 2021Model-free Policy Learning with Reward Gradientscited by 9
- 2021ARMS: Antithetic-REINFORCE-Multi-Sample Gradient for Binary Variablescited by 9
- 2021Coupled Gradient Estimators for Discrete Latent Variablescited by 9
- 2021Double Control Variates for Gradient Estimation in Discrete Latent Variable Modelscited by 9
- 2021Factored Policy Gradients: Leveraging Structure for Efficient Learning in MOMDPscited by 8
- 2021Leveraging Recursive Gumbel-Max Trick for Approximate Inference in Combinatorial Spacescited by 7
- 2022Gradient Estimation for Binary Latent Variables via Gradient Variance Clippingcited by 7
- 2023Meta-learning Control Variates: Variance Reduction with Limited Datacited by 7
- 2022Deep invariant networks with differentiable augmentation layerscited by 6
- 2023SLACK: Stable Learning of Augmentations with Cold-start and KL regularizationcited by 6
- 2024Variance reduction of diffusion model's gradients with Taylor approximation-based control variatecited by 6
- 2019Hierarchical Variational Imitation Learning of Control Programscited by 5
- 2022Variational Open-Domain Question Answeringcited by 5
50 most cited of 106 All 106 (JSON)
- 2026Variance-Optimal Control Variates for Learning with Black-box Feedback3 Oct 2026
- 2026CREDO: Variance-Guided Rubric Evolution for Replay-Corrected Credit Assignment21 Sep 2026
- 2026Gradient-estimator design overcomes trainability barriers in neural-network-based variational optimization16 Sep 2026
- 2026GFlowRL: Scaling Distribution-Matching RL to Large Language Models15 Jul 2026
- 2026Scoring Is Not Enough: Addressing Gaps in Utility-fairness Trade-offs for Ranking24 Jun 2026
- 2026Low-variance estimators overcome the phase-gradient bottleneck in complex-valued neural quantum states11 Jun 2026
- 2026DTO: a Differentiable Training Objective for Effective Counterfactual Story Rewriting24 May 2026
- 2026Conditional neural control variates for variance reduction in Bayesian inverse problems24 Feb 2026
- 2026Language Model Inversion through End-to-End Differentiation11 Feb 2026
- 2026Unbiased Single-Queried Gradient for Combinatorial Objective4 Feb 2026
- 2026Categorical Reparameterization with Denoising Diffusion models2 Jan 2026
- 2025Winning the Lottery by Preserving Network Training Dynamics with Concrete Ticket Search8 Dec 2025
38 more
- 2025Malliavin Calculus as Stochastic Backpropogation2 Nov 2025
- 2025Beyond Softmax: A Natural Parameterization for Categorical Random Variables29 Sep 2025
- 2025Discrete Variational Autoencoding via Policy Search29 Sep 2025
- 2025An Introduction to Discrete Variational Autoencoders15 May 2025
- 2025Training neural control variates using correlated configurations12 May 2025
- 2025Ideas in Inference-time Scaling can Benefit Generative Pre-training Algorithms10 Mar 2025
- 2025A Multi-Fidelity Control Variate Approach for Policy Gradient Estimation7 Mar 2025
- 2024DiffOP: Reinforcement Learning of Optimization-Based Control Policies via Implicit Policy Gradients12 Nov 2024
- 2024Adaptive Feedforward Gradient Estimation in Neural ODEs22 Sep 2024
- 2024Variance reduction of diffusion model's gradients with Taylor approximation-based control variate22 Aug 2024
- 2024How to Choose a Reinforcement-Learning Algorithm30 Jul 2024
- 2024DMTG: One-Shot Differentiable Multi-Task Grouping6 Jul 2024
- 2024Imperative Learning: A Self-supervised Neuro-Symbolic Learning Framework for Robot Autonomy23 Jun 2024
- 2024Simulating, Fast and Slow: Learning Policies for Black-Box Optimization6 Jun 2024
- 2024A Constraint-Enforcing Reward for Adversarial Attacks on Text Classifiers20 May 2024
- 2024Accelerating Legacy Numerical Solvers by Non-intrusive Gradient-based Meta-solving5 May 2024
- 2024Off-OAB: Off-Policy Policy Gradient Method with Optimal Action-Dependent Baseline4 May 2024
- 2024iMTSP: Solving Min-Max Multiple Traveling Salesman Problem with Imperative Learning1 May 2024
- 2024Probabilistic Directed Distance Fields for Ray-Based Shape Representations13 Apr 2024
- 2024Data augmentation with automated machine learning: approaches and performance comparison with classical data augmentation methods13 Mar 2024
- 2024What to Do When Your Discrete Optimization Is the Size of a Neural Network?15 Feb 2024
- 2024Black-Box Access is Insufficient for Rigorous AI Audits25 Jan 2024
- 2023Differentiable Sampling of Categorical Distributions Using the CatLog-Derivative Trick21 Nov 2023
- 2023Sparse Backpropagation for MoE Training1 Oct 2023
- 2023Deep Video Codec Control for Vision Models30 Aug 2023
- 2023Fast Slate Policy Optimization: Going Beyond Plackett-Luce3 Aug 2023
- 2023Efficient Learning of Discrete-Continuous Computation Graphs26 Jul 2023
- 2023Learning Disentangled Discrete Representations26 Jul 2023
- 2023Variational Prediction14 Jul 2023
- 2023GeoPhy: Differentiable Phylogenetic Inference via Geometric Gradients of Tree Topologies7 Jul 2023
- 2023SLACK: Stable Learning of Augmentations with Cold-start and KL regularization16 Jun 2023
- 2023Soft Merging of Experts with Adaptive Routing6 Jun 2023
- 2023Auto-CARD: Efficient and Robust Codec Avatar Driving for Real-time Mobile Telepresence24 Apr 2023
- 2023Bridging Discrete and Backpropagation: Straight-Through and Beyond17 Apr 2023
- 2023Anomaly Detection in Aeronautics Data with Quantum-compatible Discrete Deep Generative Model22 Mar 2023
- 2023Meta-learning Control Variates: Variance Reduction with Limited Data8 Mar 2023
- 2023Neural Probabilistic Logic Programming in Discrete-Continuous Domains8 Mar 2023
- 2023Implicit Bilevel Optimization: Differentiating through Bilevel Optimization Programming28 Feb 2023
50 newest of 106 All 106 (JSON)
- 2024How to Choose a Reinforcement-Learning Algorithmcited by 3
- 2025A Multi-Fidelity Control Variate Approach for Policy Gradient Estimationcited by 2
- 2024DiffOP: Reinforcement Learning of Optimization-Based Control Policies via Implicit Policy Gradientscited by 1
- 2024Off-OAB: Off-Policy Policy Gradient Method with Optimal Action-Dependent Baselinecited by 0
- 2025A Multi-Fidelity Control Variate Approach for Policy Gradient Estimation7 Mar 2025
- 2024DiffOP: Reinforcement Learning of Optimization-Based Control Policies via Implicit Policy Gradients12 Nov 2024
- 2024How to Choose a Reinforcement-Learning Algorithm30 Jul 2024
- 2024Off-OAB: Off-Policy Policy Gradient Method with Optimal Action-Dependent Baseline4 May 2024
- 2023Soft Merging of Experts with Adaptive Routingcited by 90
- 2023Sparse Backpropagation for MoE Trainingcited by 17
- 2023Sparse Backpropagation for MoE Training1 Oct 2023
- 2023Soft Merging of Experts with Adaptive Routing6 Jun 2023
- 2022Variational Open-Domain Question Answeringcited by 5
- 2022Variational Open-Domain Question Answering23 Sep 2022
- 2024Black-Box Access is Insufficient for Rigorous AI Auditscited by 126
- 2022GFlowNets and variational inferencecited by 106
- 2021Implicit MLE: Backpropagating Through Discrete Exponential Family Distributionscited by 98
- 2021A Review of the Gumbel-max Trick and its Extensions for Discrete Stochasticity in Machine Learningcited by 82
- 2022Ordered Subgraph Aggregation Networkscited by 57
- 2023Learning to Maximize Mutual Information for Dynamic Feature Selectioncited by 56
- 2021Smooth Normalizing Flowscited by 53
- 2021Variational Causal Networks: Approximate Bayesian Inference over Causal Structurescited by 43
- 2021RieszNet and ForestRiesz: Automatic Debiased Machine Learning with Neural Nets and Random Forestscited by 42
- 2022SIMPLE: A Gradient Estimator for k-Subset Samplingcited by 34
- 2023Bridging Discrete and Backpropagation: Straight-Through and Beyondcited by 31
- 2022Automatic Differentiation of Programs with Discrete Randomnesscited by 29
12 most cited of 92 All 92 (JSON)
- 2026Variance-Optimal Control Variates for Learning with Black-box Feedback3 Oct 2026
- 2026CREDO: Variance-Guided Rubric Evolution for Replay-Corrected Credit Assignment21 Sep 2026
- 2026Gradient-estimator design overcomes trainability barriers in neural-network-based variational optimization16 Sep 2026
- 2026Scoring Is Not Enough: Addressing Gaps in Utility-fairness Trade-offs for Ranking24 Jun 2026
- 2026Low-variance estimators overcome the phase-gradient bottleneck in complex-valued neural quantum states11 Jun 2026
- 2026DTO: a Differentiable Training Objective for Effective Counterfactual Story Rewriting24 May 2026
- 2026Conditional neural control variates for variance reduction in Bayesian inverse problems24 Feb 2026
- 2026Language Model Inversion through End-to-End Differentiation11 Feb 2026
- 2026Unbiased Single-Queried Gradient for Combinatorial Objective4 Feb 2026
- 2025Winning the Lottery by Preserving Network Training Dynamics with Concrete Ticket Search8 Dec 2025
- 2025Malliavin Calculus as Stochastic Backpropogation2 Nov 2025
- 2025Beyond Softmax: A Natural Parameterization for Categorical Random Variables29 Sep 2025
12 newest of 92 All 92 (JSON)