arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2407.15411cs.IR
收藏

可扩展的动态嵌入尺寸搜索用于流式推荐

Scalable Dynamic Embedding Size Search for Streaming Recommendation

The University of Queensland(昆士兰大学);Southern University of Science and Technology(南方科技大学);City University of Hong Kong(香港城市大学);Griffith University(格里菲斯大学)

Yunke Qu, Liang Qu, Tong Chen, Xiangyu Zhao, Quoc Viet Hung Nguyen, Hongzhi Yin

AI总结:

本文提出SCALL方法,通过动态调整嵌入尺寸以优化流式推荐系统的内存使用效率。

摘要:

推荐系统通常通过学习用户和物品的嵌入来表示它们,这些嵌入通常设置为均匀维度,并主导模型参数。然而,现实中的推荐系统往往在流式推荐场景中运行,其中用户和物品的数量持续增长,导致这些嵌入的存储资源消耗显著。尽管一些方法尝试通过采用嵌入尺寸搜索策略来分配不同的嵌入维度以应对流式推荐,但它们假设嵌入尺寸随着用户/物品的频率增长,最终仍会超过预定义的内存预算。为了解决这个问题,本文提出了一种名为SCALL的可扩展轻量级嵌入学习方法,用于流式推荐,该方法可以在给定的内存预算内随时间适应地调整用户/物品的嵌入尺寸。具体来说,我们提出从概率分布中采样嵌入尺寸,以保证满足任何预定义的内存预算。通过固定内存预算,所提出的嵌入尺寸采样策略可以根据相应用户或物品的频率增加和减少嵌入尺寸。此外,我们开发了一种基于强化学习的搜索范式,通过均值池化对每个状态进行建模,以保持状态向量的长度固定,无论用户和物品数量如何变化。因此,所提出的方法可以为未见过的用户和物品提供嵌入尺寸。在两个公开数据集上的全面实证评估证实了我们提出的方法的优越有效性。

英文摘要:

Recommender systems typically represent users and items by learning their embeddings, which are usually set to uniform dimensions and dominate the model parameters. However, real-world recommender systems often operate in streaming recommendation scenarios, where the number of users and items continues to grow, leading to substantial storage resource consumption for these embeddings. Although a few methods attempt to mitigate this by employing embedding size search strategies to assign different embedding dimensions in streaming recommendations, they assume that the embedding size grows with the frequency of users/items, which eventually still exceeds the predefined memory budget over time. To address this issue, this paper proposes to learn Scalable Lightweight Embeddings for streaming recommendation, called SCALL, which can adaptively adjust the embedding sizes of users/items within a given memory budget over time. Specifically, we propose to sample embedding sizes from a probabilistic distribution, with the guarantee to meet any predefined memory budget. By fixing the memory budget, the proposed embedding size sampling strategy can increase and decrease the embedding sizes in accordance to the frequency of the corresponding users or items. Furthermore, we develop a reinforcement learning-based search paradigm that models each state with mean pooling to keep the length of the state vectors fixed, invariant to the changing number of users and items. As a result, the proposed method can provide embedding sizes to unseen users and items. Comprehensive empirical evaluations on two public datasets affirm the advantageous effectiveness of our proposed method.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑