Sequence and Distance Aware Transformer for Recommendation Systems

Runqiang Zang,Meiyun Zuo,Jilei Zhou,Yining Xue,Keman Huang
DOI: https://doi.org/10.1109/ICWS53863.2021.00027
2021-01-01
Abstract:Transformer has achieved admirable success in sequential tasks. However, the model only considers the order of items in the sequence, not the relative distances, which weakens the relevance between items. To this end, we propose a novel Sequence and Distance Aware Transformer (SDAT) for recommendation systems. Specifically, we first apply the Transformer to handle the interaction between the items effectively. Then, Gated Recurrent Unit can be designed to aggregate information on an item-by-item basis in sequential information, meanwhile, we adopt the attention mechanism to focus on items with smaller time intervals to indicate high relevance. We also add a time gain function to augment the influence weight of recent items. Finally, the processing result of the time information of our integrated items replaces the positional encoding representation of the original Transformer. Extensive experiments on three real-world datasets show that SDAT outperforms state-of-the-art methods.
What problem does this paper attempt to address?