Output-Constrained Bayesian Neural Networks
arXiv (Cornell University)2019
Citations Over Time
Wanqian Yang, Lars Lorch, Moritz A. Graule, Srivatsan Srinivasan, Anirudh Suresh, Jiayu Yao, Melanie F. Pradier, Finale Doshi‐Velez
Abstract
Bayesian neural network (BNN) priors are defined in parameter space, making it hard to encode prior knowledge expressed in function space. We formulate a prior that incorporates functional constraints about what the output can or cannot be in regions of the input space. Output-Constrained BNNs (OC-BNN) represent an interpretable approach of enforcing a range of constraints, fully consistent with the Bayesian framework and amenable to black-box inference. We demonstrate how OC-BNNs improve model robustness and prevent the prediction of infeasible outputs in two real-world applications of healthcare and robotics.
Related Papers
- → Physician-Friendly Machine Learning: A Case Study with Cardiovascular Disease Risk Prediction(2019)71 cited
- → A brief review of Bayesian belief network(2019)14 cited
- Fault bayesian network based method for risk assessment of slope collapse accident(2012)
- Artificial Neural Network (ANN) Method for Predicting of Hydrochemical Types of Salt Lakes(2005)
- Prediction Study of Farmer Income Based on the Artificial Neural Network(2005)