Showing cs.LGShow all
2 papers · 1 filter
cs.LG2025
Stability-based Generalization Bounds for Variational Inference
Yadi Wei, Roni Khardon
Variational inference (VI) is widely used for approximate inference in Bayesian machine learning. In addition to this practical success, generalization bounds for variational infer…
cs.LG2024
Variational Inference on the Final-Layer Output of Neural Networks
Yadi Wei, Roni Khardon
Traditional neural networks are simple to train but they typically produce overconfident predictions. In contrast, Bayesian neural networks provide good uncertainty quantification…