如何选择tensorflow推荐系统的嵌入大小

Question

我要建立一个推荐系统使用TensorFlow推荐和双塔模型.我想知道，如何选择嵌入维的大小。对于大规模推荐系统，是否有这方面的文件？例如，Google为电影推荐选择了32维的尺寸。我的词汇表包含了大约30,000个不同的项目。

我们非常感谢你的帮助！

Answer 1

一般来说，嵌入层的大小不是一个重要的超参数。

研究发现嵌入维低于19是不表演的.和~200维后的渐近改进。

为了提高硬件效率，选择了32倍的倍数。