arXiv:2505.00560cs.IR2025-05中稿 · as a full research…被引 4

通过动态剪枝实现百万级物品的高效推荐,推理速度提升64倍。

Efficient Recommendation with Millions of Items by Dynamic Pruning of Sub-Item Embeddings

  • 基于动态剪枝思想,仅计算部分候选子项嵌入以快速定位高分商品。
  • 在Tmall数据集上,220万商品仅需不到10毫秒完成评分,比基线快64倍。
  • 无需近似最近邻技术或GPU,适合大规模实时推荐系统部署。

大型商品目录是部署现代序列推荐模型的主要挑战,因其导致模型内存占用大、推理延迟高。一种有前景的方法是RecJPQ,它用子项嵌入替代商品嵌入。然而,由于需对全部商品进行打分才能找到最高分项,推理速度仍较慢。本文借鉴文档检索中的动态剪枝思想,提出RecJPQPrune算法,可在不计算所有商品分数的情况下高效找出前K个高分商品。该算法安全至第K位,理论上保证不会遗漏潜在高分商品,不影响推荐效果。在两个大规模数据集和三种推荐模型上的实验表明,RecJPQPrune可显著提升效率:例如,在包含220万商品的Tmall数据集上,中位数打分时间相比Transformer Default基线减少64倍,相比PQTopK方法减少5.3倍。结果证明,该算法能在无GPU、不依赖近似最近邻(ANN)技术的前提下,实现超百万级商品的高效推理,单次打分时间低于10毫秒。

原文摘要 · Abstract (English)

A large item catalogue is a major challenge for deploying modern sequential recommender models, since it makes the memory footprint of the model large and increases inference latency. One promising approach to address this is RecJPQ, which replaces item embeddings with sub-item embeddings. However, slow inference remains problematic because finding the top highest-scored items usually requires scoring all items in the catalogue, which may not be feasible for large catalogues. By adapting dynamic pruning concepts from document retrieval, we propose the RecJPQPrune dynamic pruning algorithm to efficiently find the top highest-scored items without computing the scores of all items in the catalogue. Our RecJPQPrune algorithm is safe-up-to-rank K since it theoretically guarantees that no potentially high-scored item is excluded from the final top K recommendation list, thereby ensuring no impact on effectiveness. Our experiments on two large datasets and three recommendation models demonstrate the efficiency achievable using RecJPQPrune: for instance, on the Tmall dataset with 2.2M items, we can reduce the median model scoring time by 64 times compared to the Transformer Default baseline, and 5.3 times compared to a recent scoring approach called PQTopK. Overall, this paper demonstrates the effective and efficient inference of Transformer-based recommendation models at catalogue scales not previously reported in the literature. Indeed, our RecJPQPrune algorithm can score 2 million items in under 10 milliseconds without GPUs, and without relying on Approximate Nearest Neighbour (ANN) techniques.

推荐系统动态剪枝高效推理序列建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。