Robust Training of Vector Quantized Bottleneck Models