Augmenting Sequential Recommendation with Pseudo-Prior Items via Reversely Pre-training Transformer

被引:95
作者
Liu, Zhiwei [1 ]
Fan, Ziwei [1 ]
Wang, Yu [1 ]
Yu, Philip S. [1 ]
机构
[1] Univ Illinois, Dept Comp Sci, Chicago, IL 60607 USA
来源
SIGIR '21 - PROCEEDINGS OF THE 44TH INTERNATIONAL ACM SIGIR CONFERENCE ON RESEARCH AND DEVELOPMENT IN INFORMATION RETRIEVAL | 2021年
关键词
Sequential Recommendation; Transformer; Cold-start; Augmentation; Pre-training;
D O I
10.1145/3404835.3463036
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Sequential Recommendation characterizes the evolving patterns by modeling item sequences chronologically. The essential target of it is to capture the item transition correlations. The recent developments of transformer inspire the community to design effective sequence encoders, e.g., SASRec and BERT4Rec. However, we observe that these transformer-based models suffer from the cold-start issue, i.e., performing poorly for short sequences. Therefore, we propose to augment short sequences while still preserving original sequential correlations. We introduce a new framework for Augmenting Sequential Recommendation with Pseudo-prior items (ASReP). We firstly pre-train a transformer with sequences in a reverse direction to predict prior items. Then, we use this transformer to generate fabricated historical items at the beginning of short sequences. Finally, we fine-tune the transformer using these augmented sequences from the time order to predict the next item. Experiments on two real-world datasets verify the effectiveness of ASReP.
引用
收藏
页码:1608 / 1612
页数:5
相关论文
共 33 条
[1]  
Devlin J, 2019, 2019 CONFERENCE OF THE NORTH AMERICAN CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS: HUMAN LANGUAGE TECHNOLOGIES (NAACL HLT 2019), VOL. 1, P4171
[2]   Sub-graph Contrast for Scalable Self-Supervised Graph Representation Learning [J].
Jiao, Yizhu ;
Xiong, Yun ;
Zhang, Jiawei ;
Zhang, Yao ;
Zhang, Tianqi ;
Zhu, Yangyong .
20TH IEEE INTERNATIONAL CONFERENCE ON DATA MINING (ICDM 2020), 2020, :222-231
[3]   Self-Attentive Sequential Recommendation [J].
Kang, Wang-Cheng ;
McAuley, Julian .
2018 IEEE INTERNATIONAL CONFERENCE ON DATA MINING (ICDM), 2018, :197-206
[4]   Predicting Dynamic Embedding Trajectory in Temporal Interaction Networks [J].
Kumar, Srijan ;
Zhang, Xikun ;
Leskovec, Jure .
KDD'19: PROCEEDINGS OF THE 25TH ACM SIGKDD INTERNATIONAL CONFERENCCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2019, :1269-1278
[5]  
Li JC, 2020, PROCEEDINGS OF THE 13TH INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING (WSDM '20), P322, DOI 10.1145/3336191.3371786
[6]  
Li JJ, 2019, AAAI CONF ARTIF INTE, P4189
[7]   Dynamic Graph Collaborative Filtering [J].
Li, Xiaohan ;
Zhang, Mengqi ;
Wu, Shu ;
Liu, Zheng ;
Wang, Liang ;
Yu, Philip S. .
20TH IEEE INTERNATIONAL CONFERENCE ON DATA MINING (ICDM 2020), 2020, :322-331
[8]   Basket Recommendation with Multi-Intent Translation Graph Neural Network [J].
Liu, Zhiwei ;
Li, Xiaohan ;
Fan, Ziwei ;
Guo, Stephen ;
Achan, Kannan ;
Yu, Philip S. .
2020 IEEE INTERNATIONAL CONFERENCE ON BIG DATA (BIG DATA), 2020, :728-737
[9]   BasConv: Aggregating Heterogeneous Interactions for Basket Recommendation with Graph Convolutional Neural Network [J].
Liu, Zhiwei ;
Wan, Mengting ;
Guo, Stephen ;
Achan, Kannan ;
Yu, Philip S. .
PROCEEDINGS OF THE 2020 SIAM INTERNATIONAL CONFERENCE ON DATA MINING (SDM), 2020, :64-72
[10]  
Liu ZW, 2019, IEEE INT CONF BIG DA, P850, DOI [10.1109/bigdata47090.2019.9006266, 10.1109/BigData47090.2019.9006266]