Constrain-KL for Variational Autoencoders
A constrained optimisation approach for setting exact KL targets in beta-VAE training.
This undergraduate research project introduced a Constrain-KL algorithm for beta-VAE training.
Instead of sweeping over beta values, the method enforces an exact KL target and matched NVAE accuracy on CIFAR-10 in the experiments described in my CV.
