FIGURE 2.

Probabilistic graphical depiction of vaeabmil. Given the latent variables ${\mathbf{Z}}^b$, the bag label $y^b$ is independent of the observed bag ${\text{X}}^b$. We use $q_{\varphi }\left({\mathbf{Z}}^b|{\mathbf{X}}^b\right)$ for both the classification and ood detection tasks.