Incorporating Multiple Relations into Multi-task Learning for Session-Based Recommendation.

Yaohai Zeng,Bo Yang,Jie Gu
DOI: https://doi.org/10.1109/ICCCS57501.2023.10151094
2023-01-01
Abstract:Session-based recommendation aims to recommend the next item based on the information of sessions. There are multiple relations between items in sessions, e.g., the similar and sequential relation. How to learn these relations to effectively represent items and sessions is a major technical challenge of session-based recommendation. However, most of the existing session-based recommendation models do not distinguish the relations between items, thus difficult to learn the difference and relevance among relations. In this paper, we propose a model named MML-SR, which incorporates multiple relations into multi-task learning for session-based recommendation. In particular, we divide a session-based recommendation task into a similar relation recommendation task and a sequential relation recommendation task. The expert network layer and gate network layer are proposed to capture the representation of similar relations and sequential relations, then two proposed task towers are used to compute two recommendation results based on the similar relations and sequential relations of items in sessions respectively. We combine both recommendations results to get the final result. Experiments are conducted on three widely-used real-world datasets to evaluate the proposed MML-SR. The results demonstrate that MML-SR outperforms the state-of-the-art session-based recommendation models.
What problem does this paper attempt to address?