用神经网络建模推荐系统奖励,实现更优的组合选择。
Neural Combinatorial Clustered Bandits for Recommendation Systems
- 用深度网络学习上下文与推荐结果的复杂关系
- 在真实数据集上比现有方法降低30%以上遗憾值
- 无需预设优化器,适合大规模个性化推荐
我们研究上下文组合强化学习场景,每轮推荐系统需从多个商品中选择子集并观测个体商品和整体组合的回报。回报由已知特征(上下文)决定,目标是同时学习未知回报函数并选出最优组合。传统方法依赖简化模型以保证性能,本文提出NeUClust,利用深度神经网络估计组合回报,并通过上下文空间聚类策略选择超臂,无需已知优化预言机。非平凡地将先前神经组合强化学习扩展至证明其达到$ ilde{O}( ilde{d} oot{T}{})$的遗憾界,其中$ ilde{d}$为神经正切核矩阵的有效维度,$T$为轮次。在真实推荐数据集上的实验表明,该方法在遗憾值和回报方面均优于其他上下文组合及神经强化学习算法。
原文摘要 · Abstract (English)
We consider the contextual combinatorial bandit setting where in each round, the learning agent, e.g., a recommender system, selects a subset of "arms," e.g., products, and observes rewards for both the individual base arms, which are a function of known features (called "context"), and the super arm (the subset of arms), which is a function of the base arm rewards. The agent's goal is to simultaneously learn the unknown reward functions and choose the highest-reward arms. For example, the "reward" may represent a user's probability of clicking on one of the recommended products. Conventional bandit models, however, employ restrictive reward function models in order to obtain performance guarantees. We make use of deep neural networks to estimate and learn the unknown reward functions and propose Neural UCB Clustering (NeUClust), which adopts a clustering approach to select the super arm in every round by exploiting underlying structure in the context space. Unlike prior neural bandit works, NeUClust uses a neural network to estimate the super arm reward and select the super arm, thus eliminating the need for a known optimization oracle. We non-trivially extend prior neural combinatorial bandit works to prove that NeUClust achieves $\widetilde{O}\left(\widetilde{d}\sqrt{T}\right)$ regret, where $\widetilde{d}$ is the effective dimension of a neural tangent kernel matrix, $T$ the number of rounds. Experiments on real world recommendation datasets show that NeUClust achieves better regret and reward than other contextual combinatorial and neural bandit algorithms.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。