RPAF: A Reinforcement Prediction-Allocation Framework for Cache Allocation in Large-Scale Recommender Systems

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Su, Shuo, Chen, Xiaoshuang, Wang, Yao, Wu, Yulin, Zhang, Ziqiang, Zhan, Kaiqiao, Wang, Ben, Gai, Kun
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929507890364416
author Su, Shuo
Chen, Xiaoshuang
Wang, Yao
Wu, Yulin
Zhang, Ziqiang
Zhan, Kaiqiao
Wang, Ben
Gai, Kun
author_facet Su, Shuo
Chen, Xiaoshuang
Wang, Yao
Wu, Yulin
Zhang, Ziqiang
Zhan, Kaiqiao
Wang, Ben
Gai, Kun
contents Modern recommender systems are built upon computation-intensive infrastructure, and it is challenging to perform real-time computation for each request, especially in peak periods, due to the limited computational resources. Recommending by user-wise result caches is widely used when the system cannot afford a real-time recommendation. However, it is challenging to allocate real-time and cached recommendations to maximize the users' overall engagement. This paper shows two key challenges to cache allocation, i.e., the value-strategy dependency and the streaming allocation. Then, we propose a reinforcement prediction-allocation framework (RPAF) to address these issues. RPAF is a reinforcement-learning-based two-stage framework containing prediction and allocation stages. The prediction stage estimates the values of the cache choices considering the value-strategy dependency, and the allocation stage determines the cache choices for each individual request while satisfying the global budget constraint. We show that the challenge of training RPAF includes globality and the strictness of budget constraints, and a relaxed local allocator (RLA) is proposed to address this issue. Moreover, a PoolRank algorithm is used in the allocation stage to deal with the streaming allocation problem. Experiments show that RPAF significantly improves users' engagement under computational budget constraints.
format Preprint
id arxiv_https___arxiv_org_abs_2409_13175
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle RPAF: A Reinforcement Prediction-Allocation Framework for Cache Allocation in Large-Scale Recommender Systems
Su, Shuo
Chen, Xiaoshuang
Wang, Yao
Wu, Yulin
Zhang, Ziqiang
Zhan, Kaiqiao
Wang, Ben
Gai, Kun
Machine Learning
Information Retrieval
Modern recommender systems are built upon computation-intensive infrastructure, and it is challenging to perform real-time computation for each request, especially in peak periods, due to the limited computational resources. Recommending by user-wise result caches is widely used when the system cannot afford a real-time recommendation. However, it is challenging to allocate real-time and cached recommendations to maximize the users' overall engagement. This paper shows two key challenges to cache allocation, i.e., the value-strategy dependency and the streaming allocation. Then, we propose a reinforcement prediction-allocation framework (RPAF) to address these issues. RPAF is a reinforcement-learning-based two-stage framework containing prediction and allocation stages. The prediction stage estimates the values of the cache choices considering the value-strategy dependency, and the allocation stage determines the cache choices for each individual request while satisfying the global budget constraint. We show that the challenge of training RPAF includes globality and the strictness of budget constraints, and a relaxed local allocator (RLA) is proposed to address this issue. Moreover, a PoolRank algorithm is used in the allocation stage to deal with the streaming allocation problem. Experiments show that RPAF significantly improves users' engagement under computational budget constraints.
title RPAF: A Reinforcement Prediction-Allocation Framework for Cache Allocation in Large-Scale Recommender Systems
topic Machine Learning
Information Retrieval
url https://arxiv.org/abs/2409.13175