tf-vqvae icon indicating copy to clipboard operation
tf-vqvae copied to clipboard

Hi,is there any guarantee not only a few of embeddings being selected?

Open jiqizaisikao opened this issue 7 years ago • 3 comments

Hi,i have run your source code,it seems that it works well,but i have one question,for the weights of the embeddings are inited randomly,and in tranning it always select the nearest neighbors,so is there any guarantee that not a few of embeddings always being selected?

jiqizaisikao avatar Jan 23 '18 14:01 jiqizaisikao

It's a good question. I do not have an answer though since I am not the authors of the paper. In my humble opinion, describing in the conceptual level, embedding is also be trained to maximize interpretability of training examples, so it could become the most informative supports for a dataset.

hiwonjoon avatar Jan 23 '18 14:01 hiwonjoon

When I run the cifar example, it always selects a fixed subset of the embedding, so the issue indeed exists.

leao1995 avatar Mar 11 '19 05:03 leao1995

Kaiser et al. addressed this issue in their paper (https://arxiv.org/abs/1803.03382) and they called it index collapse.

bckim92 avatar Apr 13 '19 08:04 bckim92