Skip to main content
[Preprint]. 2023 Nov 5:2023.09.27.559646. Originally published 2023 Sep 29. [Version 2] doi: 10.1101/2023.09.27.559646

Table 2:

Model details. Here, hierarchical means that there are parallel pathways for information to flow from the encoder to the decoder (Fig. 2), which is slightly different from the conventional notion. For variational models, this implies hierarchical dependencies between latents in a statistical sense [68]. This hierarchical dependence is reflected in the KL term for the cNVAE, where L is the number of hierarchical latent groups. See Supplementary section 9.3 for more details and section 9.1 for a derivation. All models have an equal # of latent dimensions (420, see Table 4), approximately the same # of convolutional layers, and # of parameters (~ 24 M). EPE, endpoint error.

Model Architecture Loss Kullback–Leibler term (KL)
cNVAE Hierarchical EPE +β * KL KL==1LEqz<xKL, where KL:=𝒟KLqzx,z<pzz<
VAE Non-hierarchical EPE +β * KL KL=𝒟KL[q(zx)p(z)]
cNAE Hierarchical EPE -
AE Non-hierarchical EPE -