Summary

The problems

The solution

Thoughts

The low codebook utilisation for VQVAEs has been an issue, and several training techniques have been proposed such as normalisation and re-initialisation which are seemingly not mentioned in the work. It would be interesting to see if the implemented training strategies alleviate the problem.