ACALBERTO CASTELO
Published on

Bayesian Deep Learning for Autonomous Image-based Medical Diagnosis

Authors

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: