- Published on
Bayesian Deep Learning for Autonomous Image-based Medical Diagnosis
- Authors
- Name
- Alberto Castelo
- @acaste10
In this project, I was curious to compare interesting approaches in Bayesian Deep Learning for Computer Vision:
- Use a pretrained ResNet model and stack a final Gaussian Process layer to provide the uncertainty estimation. This model was build on top of GPytorch.
- Consider a Convolutional Bayesian Neural Network and train it from zero using Bayes-to-backprop method. This model builds on top of this repository
Follow this links to know more: