tmgrgg / localvsglobaluncertainty

Empirical analysis of recent stochastic gradient methods for approximate inference in Bayesian deep learning, including SWA-Gaussian, MultiSWAG, and deep ensembles. See report_localglobal.pdf.
2 stars 0 forks source link
approximate-inference bayesian-deep-learning dee-learning deep-neural-networks

This is my master's research project, supervised by Hippolyt Ritter and David Barber at the UCL Centre for AI. If interested I recommend reading the introductory chapter of my report (report_localglobal.pdf) to get a high-level idea.

Codebase references:

https://github.com/wjmaddox/swa_gaussian

https://github.com/timgaripov/swa/

https://github.com/izmailovpavel/understandingbdl/