Uncertainty Estimation Using Variational Mixture of Gaussians Capsule Network for Health Image Classification

<div>Comparison of the prediction probabilities of the VMG-Routing CapsNet and the Multi-lane LBP-Gabor Capsule model on in-distribution COVID-19 and MNIST test images. These results express the certainty of each model in its prediction of the test image. Notice that the deterministic model produces higher probabilities as a way of expressing overconfidence.</div>

Computational Intelligence and Neuroscience

fig3

Figure 3

Figure 3: Uncertainty Estimation Using Variational Mixture of Gaussians Capsule Network for Health Image Classification 