Concrete Dropout
Not on the map. Why?
Abstract (arXiv)
Dropout is used as a practical tool to obtain uncertainty estimates in large vision models and reinforcement learning (RL) tasks. But to obtain well-calibrated uncertainty estimates, a grid-search over the dropout probabilities is necessary - a prohibitive operation with large models, and an impossible one with RL. We propose a new dropout variant which gives improved performance and better calibrated uncertainties. Relying on recent developments in Bayesian deep learning, we use a continuous relaxation of dropout's discrete masks. Together with a principled optimisation objective, this allows for automatic tuning of the dropout probability in large models, and as a result faster experimentation cycles. In RL this allows the agent to adapt its uncertainty dynamically as more data is observed. We analyse the proposed variant extensively on a range of tasks, and give insights into common practice in the field where larger dropout probabilities are often used in deeper model layers.
Key facts. Cited by 227 papers in the map (data through 7 Oct 2026). First posted on arXiv 22 May 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2018: 2; 2019: 0; 2020: 0; 2021: 60; 2022: 44; 2023: 49; 2024: 26; 2025: 32; 2026: 14.
227 papers.
Sort:
By area:
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2018Scalable agent alignment via reward modeling: a research directioncited by 438
- 2021What Are Bayesian Neural Network Posteriors Really Like?cited by 307
- 2024A Survey on Uncertainty Quantification of Large Language Models: Taxonomy, Open Research Challenges, and Future Directionscited by 115
- 2021Dropout Q-Functions for Doubly Efficient Reinforcement Learningcited by 108
- 2021Machine Learning with a Reject Option: A surveycited by 107
- 2022CARD: Classification and Regression Diffusion Modelscited by 102
- 2021The Bayesian Learning Rulecited by 95
- 2021A Review of the Gumbel-max Trick and its Extensions for Discrete Stochasticity in Machine Learningcited by 82
- 2023DiffSTG: Probabilistic Spatio-Temporal Graph Forecasting with Denoising Diffusion Modelscited by 81
- 2022Source-Free Unsupervised Domain Adaptation: A Surveycited by 79
- 2021On Statistical Bias In Active Learning: How and When To Fix Itcited by 72
38 more
- 2024VL-Uncertainty: Detecting Hallucination in Large Vision-Language Model via Uncertainty Estimationcited by 71
- 2022Trustworthy clinical AI solutions: a unified review of uncertainty quantification in deep learning models for medical image analysiscited by 62
- 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformerscited by 59
- 2021Beyond Trivial Counterfactual Explanations with Diverse Valuable Explanationscited by 56
- 2021Quantifying Uncertainty in Deep Spatiotemporal Forecastingcited by 48
- 2022Uncertainty-guided Source-free Domain Adaptationcited by 46
- 2022Uncertainty Quantification of Collaborative Detection for Self-Drivingcited by 46
- 2021Improving Subgraph Recognition with Variational Graph Information Bottleneckcited by 45
- 2021Don't Just Blame Over-parametrization for Over-confidence: Theoretical Analysis of Calibration in Binary Classificationcited by 39
- 2022Robust Calibration with Multi-domain Temperature Scalingcited by 39
- 2021Correlated Input-Dependent Label Noise in Large-Scale Image Classificationcited by 37
- 2022Visual Attention Methods in Deep Learning: An In-Depth Surveycited by 37
- 2021Inspect, Understand, Overcome: A Survey of Practical Methods for AI Safetycited by 35
- 2023Mind the Label Shift of Augmentation-based Graph OOD Generalizationcited by 34
- 2021Calibrated and Sharp Uncertainties in Deep Learning via Density Estimationcited by 33
- 2023Making Substitute Models More Bayesian Can Enhance Transferability of Adversarial Examplescited by 32
- 2023Multi-agent Reinforcement Learning: A Comprehensive Surveycited by 30
- 2022Copula Conformal Prediction for Multi-step Time Series Forecastingcited by 28
- 2023Uncertainty Quantification in Machine Learning for Engineering Design and Health Prognostics: A Tutorialcited by 27
- 2021Contextual Dropout: An Efficient Sample-Dependent Dropout Modulecited by 26
- 2021A Survey on Epistemic (Model) Uncertainty in Supervised Learning: Recent Advances and Applicationscited by 24
- 2021Taxonomy of Machine Learning Safety: A Survey and Primercited by 22
- 2022GFlowOut: Dropout with Generative Flow Networkscited by 22
- 2023Dropout Reduces Underfittingcited by 21
- 2023Beyond Deep Ensembles: A Large-Scale Evaluation of Bayesian Deep Learning under Distribution Shiftcited by 20
- 2021Heterogeneous-Agent Trajectory Forecasting Incorporating Class Uncertaintycited by 19
- 2021Towards Lower-Dose PET using Physics-Based Uncertainty-Aware Multimodal Learning with Robustness to Out-of-Distribution Datacited by 18
- 2021Propagating State Uncertainty Through Trajectory Forecastingcited by 17
- 2023Optimal Training of Mean Variance Estimation Neural Networkscited by 17
- 2021How to Evaluate Uncertainty Estimates in Machine Learning for Regression?cited by 15
- 2023Dropout Injection at Test Time for Post Hoc Uncertainty Quantification in Neural Networkscited by 14
- 2021Pathologies in priors and inference for Bayesian transformerscited by 13
- 2024Scalable Bayesian Learning with posteriorscited by 13
- 2018Safe end-to-end imitation learning for model predictive controlcited by 12
- 2023Density-Softmax: Efficient Test-time Model for Uncertainty Estimation and Robustness under Distribution Shiftscited by 12
- 2024Looks Too Good To Be True: An Information-Theoretic Analysis of Hallucinations in Generative Restoration Modelscited by 12
- 2025Quantifying and Alleviating Co-Adaptation in Sparse-View 3D Gaussian Splattingcited by 12
- 2021STUaNet: Understanding uncertainty in spatiotemporal collective human mobilitycited by 11
50 most cited of 227 All 227 (JSON)
- 2026Towards Robust Time Series Learning via Capacity-Centric Modulation30 Sep 2026
- 2026Optimal Transport Dropout for Structured Predictive Uncertainty27 Sep 2026
- 2026Predictive Uncertainty for Neural CAE Surrogates21 Sep 2026
- 2026A Blueprint for Equilibrium-Based Differentiable Continuous-Variable Thermodynamic Computing17 Jul 2026
- 2026Uncertainty Quantification for AI-Driven Crash Simulation Surrogates: A Comparative Study of Monte Carlo Dropout and Deep Ensemble on Open-Source Bumper Beam Benchmark30 Jun 2026
- 2026Explicit Dropout: Deterministic Regularization for Transformer Architectures22 Apr 2026
- 2026Monte Carlo Stochastic Depth for Uncertainty Estimation in Deep Learning14 Apr 2026
- 2026DynamicGate MLP Conditional Computation via Learned Structural Dropout and Input Dependent Gating for Functional Plasticity17 Mar 2026
- 2026Deep Learning-Based Metamodeling of Nonlinear Stochastic Dynamic Systems under Parametric and Predictive Uncertainty12 Mar 2026
- 2026Confidence-Uncertainty Boundary Calibration for Bayesian Deep Learning in Medical Image Analysis12 Feb 2026
- 2026Data-driven registration and modeling of brain deformation for image-guided neurosurgery9 Feb 2026
- 2026Variance-Gated Ensembles: An Epistemic-Aware Framework for Uncertainty Estimation8 Feb 2026
38 more
- 2026DropoutTS: Sample-Adaptive Dropout for Robust Time Series Forecasting29 Jan 2026
- 2026Calibrated uncertainty quantification for prosumer flexibility aggregation in ancillary service markets21 Jan 2026
- 2025CLAPS: Aleatoric-Epistemic Scaling via Last-Layer Laplace for Conformal Regression1 Dec 2025
- 2025DARN: Dynamic Adaptive Regularization Networks for Efficient and Robust Foundation Model Adaptation6 Nov 2025
- 2025Convergence, design and training of continuous-time dropout as a random batch method15 Oct 2025
- 2025Clutch Control: An Attention-based Combinatorial Bandit for Efficient Mutation in JavaScript Engine Fuzzing14 Oct 2025
- 2025Moral Anchor System: A Predictive Framework for AI Value Alignment and Drift Prevention5 Oct 2025
- 2025Calibrating the Full Predictive Class Distribution of 3D Object Detectors for Autonomous Driving2 Oct 2025
- 2025Conformal Prediction for Time-series Forecasting with Change Points2 Sep 2025
- 2025Quantifying Out-of-Training Uncertainty of Neural-Network based Turbulence Closures23 Aug 2025
- 2025Anchor-MoE: A Mean-Anchored Mixture of Experts For Probabilistic Regression22 Aug 2025
- 2025Quantifying and Alleviating Co-Adaptation in Sparse-View 3D Gaussian Splatting18 Aug 2025
- 2025Synaptic Pruning: A Biological Inspiration for Deep Learning Regularization12 Aug 2025
- 2025Deep Neural Network Calibration by Reducing Classifier Shift with Stochastic Masking12 Aug 2025
- 2025OFAL: An Oracle-Free Active Learning Framework11 Aug 2025
- 2025UGOD: Uncertainty-Guided Differentiable Opacity and Soft Dropout for Enhanced Sparse-View 3DGS7 Aug 2025
- 2025Accelerating Hamiltonian Monte Carlo for Bayesian Inference in Neural Networks and Neural Operators19 Jul 2025
- 2025SGPMIL: Sparse Gaussian Process Multiple Instance Learning11 Jul 2025
- 2025Cross-regularization: Adaptive Model Complexity through Validation Gradients24 Jun 2025
- 2025On the Robustness of Human-Object Interaction Detection against Distribution Shift22 Jun 2025
- 2025Scaling Laws for Uncertainty in Deep Learning11 Jun 2025
- 2025Uncertainty Quantification for Physics-Informed Neural Networks with Extended Fiducial Inference25 May 2025
- 2025Enhancing Monte Carlo Dropout Performance for Uncertainty Quantification21 May 2025
- 2025MID-L: Matrix-Interpolated Dropout Layer with Layer-wise Neuron Selection16 May 2025
- 2025Uncertainty quantification with approximate variational learning for wearable photoplethysmography prediction tasks16 May 2025
- 2025An Analysis of Temporal Dropout in Earth Observation Time Series for Regression Tasks9 Apr 2025
- 2025POINT2: A Polymer Informatics Training and Testing Database30 Mar 2025
- 2025Neuroplasticity in Artificial Intelligence -- An Overview and Inspirations on Drop In & Out Learning27 Mar 2025
- 2025E-PINNs: Epistemic Physics-Informed Neural Networks25 Mar 2025
- 2025Explainable Bayesian deep learning through input-skip Latent Binary Bayesian Neural Networks13 Mar 2025
- 2025Foreclassing: A new machine learning perspective on human decision making with temporal data6 Mar 2025
- 2025Damper-B-PINN: Damper Characteristics-Based Bayesian Physics-Informed Neural Network for Vehicle State Estimation28 Feb 2025
- 2025Uncertainty Aware Human-machine Collaboration in Camouflaged Object Detection12 Feb 2025
- 2025Uncertainty-Aware Online Extrinsic Calibration: A Conformal Prediction Approach12 Jan 2025
- 2024Rate-In: Information-Driven Adaptive Dropout Rates for Improved Inference-Time Uncertainty Estimation10 Dec 2024
- 2024Safety Monitoring of Machine Learning Perception Functions: a Survey9 Dec 2024
- 2024A Survey on Uncertainty Quantification of Large Language Models: Taxonomy, Open Research Challenges, and Future Directions7 Dec 2024
- 2024A Review of Bayesian Uncertainty Quantification in Deep Probabilistic Image Segmentation25 Nov 2024
50 newest of 227 All 227 (JSON)
- 2022CARD: Classification and Regression Diffusion Modelscited by 102
- 2023DiffSTG: Probabilistic Spatio-Temporal Graph Forecasting with Denoising Diffusion Modelscited by 81
- 2024Diffusion Boosted Treescited by 2
- 2024General Intelligent Imaging and Uncertainty Quantification by Deterministic Diffusion Modelcited by 0
- 2024General Intelligent Imaging and Uncertainty Quantification by Deterministic Diffusion Model23 Aug 2024
- 2024Diffusion Boosted Trees3 Jun 2024
- 2023DiffSTG: Probabilistic Spatio-Temporal Graph Forecasting with Denoising Diffusion Models31 Jan 2023
- 2022CARD: Classification and Regression Diffusion Models15 Jun 2022
- 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformerscited by 59
- 2025Anchor-MoE: A Mean-Anchored Mixture of Experts For Probabilistic Regressioncited by 0
- 2026DynamicGate MLP Conditional Computation via Learned Structural Dropout and Input Dependent Gating for Functional Plasticitycited by 0
- 2026DynamicGate MLP Conditional Computation via Learned Structural Dropout and Input Dependent Gating for Functional Plasticity17 Mar 2026
- 2025Anchor-MoE: A Mean-Anchored Mixture of Experts For Probabilistic Regression22 Aug 2025
- 2023Sparse MoE as the New Dropout: Scaling Dense and Self-Slimmable Transformers2 Mar 2023
- 2021A Survey of Uncertainty in Deep Neural Networkscited by 675
- 2021What Are Bayesian Neural Network Posteriors Really Like?cited by 307
- 2024A Survey on Uncertainty Quantification of Large Language Models: Taxonomy, Open Research Challenges, and Future Directionscited by 115
- 2021Dropout Q-Functions for Doubly Efficient Reinforcement Learningcited by 108
- 2021Machine Learning with a Reject Option: A surveycited by 107
- 2021The Bayesian Learning Rulecited by 95
- 2021A Review of the Gumbel-max Trick and its Extensions for Discrete Stochasticity in Machine Learningcited by 82
- 2022Source-Free Unsupervised Domain Adaptation: A Surveycited by 79
- 2021On Statistical Bias In Active Learning: How and When To Fix Itcited by 72
- 2022Trustworthy clinical AI solutions: a unified review of uncertainty quantification in deep learning models for medical image analysiscited by 62
- 2021Beyond Trivial Counterfactual Explanations with Diverse Valuable Explanationscited by 56
- 2021Quantifying Uncertainty in Deep Spatiotemporal Forecastingcited by 48
12 most cited of 215 All 215 (JSON)
- 2026Towards Robust Time Series Learning via Capacity-Centric Modulation30 Sep 2026
- 2026Optimal Transport Dropout for Structured Predictive Uncertainty27 Sep 2026
- 2026Predictive Uncertainty for Neural CAE Surrogates21 Sep 2026
- 2026A Blueprint for Equilibrium-Based Differentiable Continuous-Variable Thermodynamic Computing17 Jul 2026
- 2026Uncertainty Quantification for AI-Driven Crash Simulation Surrogates: A Comparative Study of Monte Carlo Dropout and Deep Ensemble on Open-Source Bumper Beam Benchmark30 Jun 2026
- 2026Explicit Dropout: Deterministic Regularization for Transformer Architectures22 Apr 2026
- 2026Monte Carlo Stochastic Depth for Uncertainty Estimation in Deep Learning14 Apr 2026
- 2026Deep Learning-Based Metamodeling of Nonlinear Stochastic Dynamic Systems under Parametric and Predictive Uncertainty12 Mar 2026
- 2026Confidence-Uncertainty Boundary Calibration for Bayesian Deep Learning in Medical Image Analysis12 Feb 2026
- 2026Data-driven registration and modeling of brain deformation for image-guided neurosurgery9 Feb 2026
- 2026Variance-Gated Ensembles: An Epistemic-Aware Framework for Uncertainty Estimation8 Feb 2026
- 2026DropoutTS: Sample-Adaptive Dropout for Robust Time Series Forecasting29 Jan 2026
12 newest of 215 All 215 (JSON)