2 In this way, the model initially learns to use the latent code but is then regularized towards the prior as training progresses.
this trick is not sufficient to avert KL collapse in all scenarios, particularly 2 Reweighting the KL term is also used in methods like β-VAE (Higgins et al., 2017) and InfoVAE (Zhao et al., 2017b).