arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2005.12981cs.IRcs.AIcs.LG
收藏

深度兴趣与层次注意力网络的点击率预测

Deep Interest with Hierarchical Attention Network for Click-Through Rate Prediction

Alibaba Group(阿里巴巴集团)

Weinan Xu, Hengxu He, Minshi Tan, Yunming Li, Jun Lang, Dongbai Guo

摘要:

深度兴趣网络(DIN)是一种使用注意力机制从历史行为中捕获用户兴趣的最新模型。用户兴趣直观地遵循分层模式,使得用户通常从较高级别然后到较低级别抽象示出兴趣。在注意力网络中对这种兴趣层次进行建模可以从根本上提高用户行为的表示。因此,我们提出了一种改进的DIN模型任意的兴趣层次:层次注意力网络(DHAN)的深度兴趣。在该模型中,第一个关注层上引入多维层次结构,关注单个项目,同一维度下的后续关注层关注在较低层之上的更高层次。为了支持多维层次结构的建模,引入了一种扩展机制来捕获一个到多个层次结构。为了验证该模型的有效性,将简化的DHAN应用于点击率(CTR)预测,并在三个公共数据集上进行了实验,实验结果表明,DHAN模型能够有效地捕捉用户在不同维度(如类别、价格或品牌)上的兴趣。与DIN相比,DHAN的AUC从12%显著提高到21%,显示出其优越性。DHAN还与另一种最先进的模型深度兴趣进化网络(DIEN)进行了比较,该模型对时间兴趣进行了建模。简化的DHAN的AUC也比DIEN略有提高,从1.0%提高到1.7%。一个潜在的未来的工作可以结合DHAN和DIEN模型的时间和层次的利益。

英文摘要:

Deep Interest Network (DIN) is a state-of-the-art model which uses attention mechanism to capture user interests from historical behaviors. User interests intuitively follow a hierarchical pattern such that users generally show interests from a higher-level then to a lower-level abstraction. Modeling such an interest hierarchy in an attention network can fundamentally improve the representation of user behaviors. We, therefore, propose an improvement over DIN to model arbitrary interest hierarchy: Deep Interest with Hierarchical Attention Network (DHAN). In this model, a multi-dimensional hierarchical structure is introduced on the first attention layer which attends to an individual item, and the subsequent attention layers in the same dimension attend to higher-level hierarchy built on top of the lower corresponding layers. To enable modeling of multiple dimensional hierarchies, an expanding mechanism is introduced to capture one to many hierarchies. This design enables DHAN to attend different importance to different hierarchical abstractions thus can fully capture user interests at different dimensions (e.g. category, price, or brand).To validate our model, a simplified DHAN has applied to Click-Through Rate (CTR) prediction and our experimental results on three public datasets with two levels of the one-dimensional hierarchy only by category. It shows the superiority of DHAN with significant AUC uplift from 12% to 21% over DIN. DHAN is also compared with another state-of-the-art model Deep Interest Evolution Network (DIEN), which models temporal interest. The simplified DHAN also gets slight AUC uplift from 1.0% to 1.7% over DIEN. A potential future work can be a combination of DHAN and DIEN to model both temporal and hierarchical interests.

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

↑