Learning and Policy Search in Stochastic Dynamical Systems with Bayesian Neural Networks
Not on the map. Why?
Abstract (arXiv)
We present an algorithm for model-based reinforcement learning that combines Bayesian neural networks (BNNs) with random roll-outs and stochastic optimization for policy learning. The BNNs are trained by minimizing α-divergences, allowing us to capture complicated statistical patterns in the transition dynamics, e.g. multi-modality and heteroskedasticity, which are usually missed by other common modeling approaches. We illustrate the performance of our method by solving a challenging benchmark where model-based approaches usually fail and by obtaining promising results in a real-world scenario for controlling a gas turbine.
Key facts. Cited by 49 papers in the map (data through 14 Oct 2026). First posted on arXiv 23 May 2016.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 2; 2021: 13; 2022: 3; 2023: 14; 2024: 9; 2025: 3; 2026: 5.
49 papers.
Sort:
By area:
- 2020MOPO: Model-based Offline Policy Optimizationcited by 636
- 2023Reward Model Ensembles Help Mitigate Overoptimizationcited by 231
- 2021RvS: What is Essential for Offline RL via Supervised Learning?cited by 185
- 2024Uncertainty-aware Reward Model: Teaching Reward Models to Know What is Unknowncited by 74
- 2021Vector Quantized Models for Planningcited by 48
- 2020Overcoming Model Bias for Robust Offline Deep Reinforcement Learningcited by 43
- 2021Continuous-Time Model-Based Reinforcement Learningcited by 41
- 2021Few-shot Language Coordination by Modeling Theory of Mindcited by 32
- 2021Physics-informed Dyna-Style Model-Based Deep Reinforcement Learning for Dynamic Controlcited by 23
- 2021Heterogeneous-Agent Trajectory Forecasting Incorporating Class Uncertaintycited by 19
- 2024Trust the Model Where It Trusts Itself -- Model-Based Actor-Critic with Uncertainty-Aware Rollout Adaptioncited by 18
- 2021Propagating State Uncertainty Through Trajectory Forecastingcited by 17
37 more
- 2022Gluformer: Transformer-Based Personalized Glucose Forecasting with Uncertainty Quantificationcited by 17
- 2021Function-space Inference with Sparse Implicit Processescited by 13
- 2022User-Interactive Offline Reinforcement Learningcited by 11
- 2022Uncertainty Quantification for Deep Unrolling-Based Computational Imagingcited by 10
- 2023Normalizing Flow Ensembles for Rich Aleatoric and Epistemic Uncertainty Modelingcited by 10
- 2023Improving Offline RL by Blending Heuristicscited by 10
- 2023Efficient Uncertainty Quantification and Reduction for Over-Parameterized Neural Networkscited by 10
- 2023How to Learn and Generalize From Three Minutes of Data: Physics-Constrained and Uncertainty-Aware Neural Stochastic Differential Equationscited by 10
- 2021Certification of Iterative Predictions in Bayesian Neural Networkscited by 8
- 2024On Divergence Measures for Training GFlowNetscited by 6
- 2023Probabilistic Reach-Avoid for Bayesian Neural Networkscited by 5
- 2025Conformalized Generative Bayesian Imaging: An Uncertainty Quantification Framework for Computational Imagingcited by 5
- 2024FACTS: A Factored State-Space Framework For World Modellingcited by 4
- 2021Measuring Data Quality for Dataset Selection in Offline Reinforcement Learningcited by 3
- 2023Learning Control Policies for Variable Objectives from Offline Datacited by 3
- 2023Mind the Uncertainty: Risk-Aware and Actively Exploring Model-Based Reinforcement Learningcited by 3
- 2023CROP: Conservative Reward for Model-based Offline Policy Optimizationcited by 3
- 2025Look Before Leap: Look-Ahead Planning with Uncertainty in Reinforcement Learningcited by 3
- 2021Learning to Reweight Imaginary Transitions for Model-Based Reinforcement Learningcited by 2
- 2021Federated Ensemble Model-based Reinforcement Learning in Edge Computingcited by 2
- 2023DiSProD: Differentiable Symbolic Propagation of Distributions for Planningcited by 2
- 2023A Review of Symbolic, Subsymbolic and Hybrid Methods for Sequential Decision Makingcited by 2
- 2023Sampling-Free Probabilistic Deep State-Space Modelscited by 2
- 2026Variational Neural Belief Parameterizations for Robust Dexterous Grasping under Multimodal Uncertaintycited by 2
- 2021Adaptive learning for financial markets mixing model-based and model-free RL for volatility targetingcited by 1
- 2023Towards Dependable Autonomous Systems Based on Bayesian Deep Learning Componentscited by 1
- 2024Learning World Models With Hierarchical Temporal Abstractions: A Probabilistic Perspectivecited by 1
- 2024Mamba as Decision Maker: Exploring Multi-scale Sequence Modeling in Offline Reinforcement Learningcited by 1
- 2024Bounded Exploration with World Model Uncertainty in Soft Actor-Critic Reinforcement Learning Algorithmcited by 1
- 2025Combining Bayesian Inference and Reinforcement Learning for Agent Decision Making: A Reviewcited by 1
- 2026Uncertainty-Aware Reward Discounting for Mitigating Reward Hackingcited by 1
- 2023Quantifying Uncertainty in Deep Learning Classification with Noise in Discrete Inputs for Risk-Based Decision Makingcited by 0
- 2024Generative modeling of density regression through tree flowscited by 0
- 2024Mitigating Relative Over-Generalization in Multi-Agent Reinforcement Learningcited by 0
- 2026Stochastic Actor-Critic: Mitigating Overestimation via Temporal Aleatoric Uncertaintycited by 0
- 2026All Models are Wrong, Knowing Where is Useful: On Model Uncertainty in Reinforcement Learningcited by 0
- 2026Full-Covariance Smoothing of Bayesian Neural Networks for Online Adaptationcited by 0
- 2026Full-Covariance Smoothing of Bayesian Neural Networks for Online Adaptation23 Sep 2026
- 2026All Models are Wrong, Knowing Where is Useful: On Model Uncertainty in Reinforcement Learning31 May 2026
- 2026Uncertainty-Aware Reward Discounting for Mitigating Reward Hacking29 Apr 2026
- 2026Variational Neural Belief Parameterizations for Robust Dexterous Grasping under Multimodal Uncertainty28 Apr 2026
- 2026Stochastic Actor-Critic: Mitigating Overestimation via Temporal Aleatoric Uncertainty2 Jan 2026
- 2025Combining Bayesian Inference and Reinforcement Learning for Agent Decision Making: A Review12 May 2025
- 2025Conformalized Generative Bayesian Imaging: An Uncertainty Quantification Framework for Computational Imaging10 Apr 2025
- 2025Look Before Leap: Look-Ahead Planning with Uncertainty in Reinforcement Learning26 Mar 2025
- 2024Bounded Exploration with World Model Uncertainty in Soft Actor-Critic Reinforcement Learning Algorithm9 Dec 2024
- 2024Mitigating Relative Over-Generalization in Multi-Agent Reinforcement Learning17 Nov 2024
- 2024FACTS: A Factored State-Space Framework For World Modelling28 Oct 2024
- 2024On Divergence Measures for Training GFlowNets12 Oct 2024
37 more
- 2024Uncertainty-aware Reward Model: Teaching Reward Models to Know What is Unknown1 Oct 2024
- 2024Generative modeling of density regression through tree flows7 Jun 2024
- 2024Mamba as Decision Maker: Exploring Multi-scale Sequence Modeling in Offline Reinforcement Learning4 Jun 2024
- 2024Trust the Model Where It Trusts Itself -- Model-Based Actor-Critic with Uncertainty-Aware Rollout Adaption29 May 2024
- 2024Learning World Models With Hierarchical Temporal Abstractions: A Probabilistic Perspective24 Apr 2024
- 2023CROP: Conservative Reward for Model-based Offline Policy Optimization26 Oct 2023
- 2023Quantifying Uncertainty in Deep Learning Classification with Noise in Discrete Inputs for Risk-Based Decision Making9 Oct 2023
- 2023Reward Model Ensembles Help Mitigate Overoptimization4 Oct 2023
- 2023Probabilistic Reach-Avoid for Bayesian Neural Networks3 Oct 2023
- 2023Sampling-Free Probabilistic Deep State-Space Models15 Sep 2023
- 2023Mind the Uncertainty: Risk-Aware and Actively Exploring Model-Based Reinforcement Learning11 Sep 2023
- 2023Learning Control Policies for Variable Objectives from Offline Data11 Aug 2023
- 2023How to Learn and Generalize From Three Minutes of Data: Physics-Constrained and Uncertainty-Aware Neural Stochastic Differential Equations10 Jun 2023
- 2023Efficient Uncertainty Quantification and Reduction for Over-Parameterized Neural Networks9 Jun 2023
- 2023Improving Offline RL by Blending Heuristics1 Jun 2023
- 2023A Review of Symbolic, Subsymbolic and Hybrid Methods for Sequential Decision Making20 Apr 2023
- 2023DiSProD: Differentiable Symbolic Propagation of Distributions for Planning3 Feb 2023
- 2023Normalizing Flow Ensembles for Rich Aleatoric and Epistemic Uncertainty Modeling2 Feb 2023
- 2023Towards Dependable Autonomous Systems Based on Bayesian Deep Learning Components12 Jan 2023
- 2022Gluformer: Transformer-Based Personalized Glucose Forecasting with Uncertainty Quantification9 Sep 2022
- 2022Uncertainty Quantification for Deep Unrolling-Based Computational Imaging2 Jul 2022
- 2022User-Interactive Offline Reinforcement Learning21 May 2022
- 2021RvS: What is Essential for Offline RL via Supervised Learning?20 Dec 2021
- 2021Measuring Data Quality for Dataset Selection in Offline Reinforcement Learning26 Nov 2021
- 2021Function-space Inference with Sparse Implicit Processes14 Oct 2021
- 2021Propagating State Uncertainty Through Trajectory Forecasting7 Oct 2021
- 2021Federated Ensemble Model-based Reinforcement Learning in Edge Computing12 Sep 2021
- 2021Physics-informed Dyna-Style Model-Based Deep Reinforcement Learning for Dynamic Control31 Jul 2021
- 2021Few-shot Language Coordination by Modeling Theory of Mind12 Jul 2021
- 2021Vector Quantized Models for Planning8 Jun 2021
- 2021Certification of Iterative Predictions in Bayesian Neural Networks21 May 2021
- 2021Heterogeneous-Agent Trajectory Forecasting Incorporating Class Uncertainty26 Apr 2021
- 2021Adaptive learning for financial markets mixing model-based and model-free RL for volatility targeting19 Apr 2021
- 2021Learning to Reweight Imaginary Transitions for Model-Based Reinforcement Learning9 Apr 2021
- 2021Continuous-Time Model-Based Reinforcement Learning9 Feb 2021
- 2020Overcoming Model Bias for Robust Offline Deep Reinforcement Learning12 Aug 2020
- 2020MOPO: Model-based Offline Policy Optimization27 May 2020
- 2020MOPO: Model-based Offline Policy Optimizationcited by 636
- 2021RvS: What is Essential for Offline RL via Supervised Learning?cited by 185
- 2020Overcoming Model Bias for Robust Offline Deep Reinforcement Learningcited by 43
- 2022User-Interactive Offline Reinforcement Learningcited by 11
- 2023Improving Offline RL by Blending Heuristicscited by 10
- 2021Measuring Data Quality for Dataset Selection in Offline Reinforcement Learningcited by 3
- 2023Learning Control Policies for Variable Objectives from Offline Datacited by 3
- 2023CROP: Conservative Reward for Model-based Offline Policy Optimizationcited by 3
- 2023CROP: Conservative Reward for Model-based Offline Policy Optimization26 Oct 2023
- 2023Learning Control Policies for Variable Objectives from Offline Data11 Aug 2023
- 2023Improving Offline RL by Blending Heuristics1 Jun 2023
- 2022User-Interactive Offline Reinforcement Learning21 May 2022
- 2021RvS: What is Essential for Offline RL via Supervised Learning?20 Dec 2021
- 2021Measuring Data Quality for Dataset Selection in Offline Reinforcement Learning26 Nov 2021
- 2020Overcoming Model Bias for Robust Offline Deep Reinforcement Learning12 Aug 2020
- 2020MOPO: Model-based Offline Policy Optimization27 May 2020
- 2024Trust the Model Where It Trusts Itself -- Model-Based Actor-Critic with Uncertainty-Aware Rollout Adaptioncited by 18
- 2025Look Before Leap: Look-Ahead Planning with Uncertainty in Reinforcement Learningcited by 3
- 2025Combining Bayesian Inference and Reinforcement Learning for Agent Decision Making: A Reviewcited by 1
- 2026Stochastic Actor-Critic: Mitigating Overestimation via Temporal Aleatoric Uncertaintycited by 0
- 2026All Models are Wrong, Knowing Where is Useful: On Model Uncertainty in Reinforcement Learningcited by 0
- 2026All Models are Wrong, Knowing Where is Useful: On Model Uncertainty in Reinforcement Learning31 May 2026
- 2026Stochastic Actor-Critic: Mitigating Overestimation via Temporal Aleatoric Uncertainty2 Jan 2026
- 2025Combining Bayesian Inference and Reinforcement Learning for Agent Decision Making: A Review12 May 2025
- 2025Look Before Leap: Look-Ahead Planning with Uncertainty in Reinforcement Learning26 Mar 2025
- 2024Trust the Model Where It Trusts Itself -- Model-Based Actor-Critic with Uncertainty-Aware Rollout Adaption29 May 2024
- 2021Vector Quantized Models for Planningcited by 48
- 2021Continuous-Time Model-Based Reinforcement Learningcited by 41
- 2021Few-shot Language Coordination by Modeling Theory of Mindcited by 32
- 2021Physics-informed Dyna-Style Model-Based Deep Reinforcement Learning for Dynamic Controlcited by 23
- 2021Heterogeneous-Agent Trajectory Forecasting Incorporating Class Uncertaintycited by 19
- 2021Propagating State Uncertainty Through Trajectory Forecastingcited by 17
- 2022Gluformer: Transformer-Based Personalized Glucose Forecasting with Uncertainty Quantificationcited by 17
- 2021Function-space Inference with Sparse Implicit Processescited by 13
- 2022Uncertainty Quantification for Deep Unrolling-Based Computational Imagingcited by 10
- 2023Normalizing Flow Ensembles for Rich Aleatoric and Epistemic Uncertainty Modelingcited by 10
- 2023Efficient Uncertainty Quantification and Reduction for Over-Parameterized Neural Networkscited by 10
- 2023How to Learn and Generalize From Three Minutes of Data: Physics-Constrained and Uncertainty-Aware Neural Stochastic Differential Equationscited by 10
12 most cited of 28 All 28 (JSON)
- 2026Full-Covariance Smoothing of Bayesian Neural Networks for Online Adaptation23 Sep 2026
- 2026Variational Neural Belief Parameterizations for Robust Dexterous Grasping under Multimodal Uncertainty28 Apr 2026
- 2025Conformalized Generative Bayesian Imaging: An Uncertainty Quantification Framework for Computational Imaging10 Apr 2025
- 2024Mitigating Relative Over-Generalization in Multi-Agent Reinforcement Learning17 Nov 2024
- 2024On Divergence Measures for Training GFlowNets12 Oct 2024
- 2024Generative modeling of density regression through tree flows7 Jun 2024
- 2023Quantifying Uncertainty in Deep Learning Classification with Noise in Discrete Inputs for Risk-Based Decision Making9 Oct 2023
- 2023Probabilistic Reach-Avoid for Bayesian Neural Networks3 Oct 2023
- 2023Sampling-Free Probabilistic Deep State-Space Models15 Sep 2023
- 2023Mind the Uncertainty: Risk-Aware and Actively Exploring Model-Based Reinforcement Learning11 Sep 2023
- 2023How to Learn and Generalize From Three Minutes of Data: Physics-Constrained and Uncertainty-Aware Neural Stochastic Differential Equations10 Jun 2023
- 2023Efficient Uncertainty Quantification and Reduction for Over-Parameterized Neural Networks9 Jun 2023
12 newest of 28 All 28 (JSON)