arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 1902.09757cs.LGcs.IRcs.SIstat.ML
收藏

Interaction-aware Factorization Machines for Recommender Systems

Advertising and Marketing Services, Corporate Development Group, Tencent Inc.(腾讯公司广告与营销服务部、企业发展集团);zju.edu.cn(浙江大学(注:此为邮箱域名,非明确机构名,按要求不应提取,但原HTML中作者邮箱部分可能存在格式问题,根据提取范围,仅提取明确的机构名,此处原文本中唯一明确的机构是腾讯公司的部门,故仅保留该机构))

Fuxing Hong, Dongbo Huang, Ge Chen

英文摘要:

Factorization Machine (FM) is a widely used supervised learning approach by effectively modeling of feature interactions. Despite the successful application of FM and its many deep learning variants, treating every feature interaction fairly may degrade the performance. For example, the interactions of a useless feature may introduce noises; the importance of a feature may also differ when interacting with different features. In this work, we propose a novel model named \emph{Interaction-aware Factorization Machine} (IFM) by introducing Interaction-Aware Mechanism (IAM), which comprises the \emph{feature aspect} and the \emph{field aspect}, to learn flexible interactions on two levels. The feature aspect learns feature interaction importance via an attention network while the field aspect learns the feature interaction effect as a parametric similarity of the feature interaction vector and the corresponding field interaction prototype. IFM introduces more structured control and learns feature interaction importance in a stratified manner, which allows for more leverage in tweaking the interactions on both feature-wise and field-wise levels. Besides, we give a more generalized architecture and propose Interaction-aware Neural Network (INN) and DeepIFM to capture higher-order interactions. To further improve both the performance and efficiency of IFM, a sampling scheme is developed to select interactions based on the field aspect importance. The experimental results from two well-known datasets show the superiority of the proposed models over the state-of-the-art methods.

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

↑