阿里二项箔召回《Approximate Nearest Neighbor Search under Neural Similarity Metric for Large-Scale Recommendation》

这里阿里继TDM之后提出的新的突破双塔限制的召回算法,TDM索引部分采用的索引是树结构,其非叶子结点是虚拟结点(虚拟结点不好加入item side info信息),在这篇论文里采用了HNSW来作为索引,HNSW最下面一层保护了所有item,上面一层是下面一层的随机采样

1. 模型训练完后可以得到item embedding,用item embedding构建HNSW索引

2. serving时,对每层计算topk item

 

参考资料

https://zhuanlan.zhihu.com/p/443113850

https://arxiv.org/pdf/2202.10226

posted @ 2024-05-13 20:02  AI_Engineer  阅读(350)  评论(0)    收藏  举报