RLT4Rec: Reinforcement Learning Transformer for User Cold Start and Item Recommendation

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Rajapakse, Dilina Chandika, Leith, Douglas
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866916516654481408
author Rajapakse, Dilina Chandika
Leith, Douglas
author_facet Rajapakse, Dilina Chandika
Leith, Douglas
contents We introduce a new sequential transformer reinforcement learning architecture RLT4Rec and demonstrate that it achieves excellent performance in a range of item recommendation tasks. RLT4Rec uses a relatively simple transformer architecture that takes as input the user's (item,rating) history and outputs the next item to present to the user. Unlike existing RL approaches, there is no need to input a state observation or estimate. RLT4Rec handles new users and established users within the same consistent framework and automatically balances the "exploration" needed to discover the preferences of a new user with the "exploitation" that is more appropriate for established users. Training of RLT4Rec is robust and fast and is insensitive to the choice of training data, learning to generate "good" personalised sequences that the user tends to rate highly even when trained on "bad" data.
format Preprint
id arxiv_https___arxiv_org_abs_2412_07403
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle RLT4Rec: Reinforcement Learning Transformer for User Cold Start and Item Recommendation
Rajapakse, Dilina Chandika
Leith, Douglas
Information Retrieval
We introduce a new sequential transformer reinforcement learning architecture RLT4Rec and demonstrate that it achieves excellent performance in a range of item recommendation tasks. RLT4Rec uses a relatively simple transformer architecture that takes as input the user's (item,rating) history and outputs the next item to present to the user. Unlike existing RL approaches, there is no need to input a state observation or estimate. RLT4Rec handles new users and established users within the same consistent framework and automatically balances the "exploration" needed to discover the preferences of a new user with the "exploitation" that is more appropriate for established users. Training of RLT4Rec is robust and fast and is insensitive to the choice of training data, learning to generate "good" personalised sequences that the user tends to rate highly even when trained on "bad" data.
title RLT4Rec: Reinforcement Learning Transformer for User Cold Start and Item Recommendation
topic Information Retrieval
url https://arxiv.org/abs/2412.07403