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

<div>Comparison of the uncertainty between the VMG-Routing and Multi-lane CapsNets on in-distribution test images. The VMG-Routing CapsNet shows uncertainty over the ML model on the MNIST dataset. On the COVID-19 dataset, the uncertainty for both models is approximately the same.</div>

Computational Intelligence and Neuroscience

fig4

Figure 4

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