RLPF: Reinforcement Learning from Prediction Feedback for User Summarization with LLMs

Authors

  • Jiaxing Wu Google DeepMind
  • Lin Ning Google DeepMind
  • Luyang Liu Google DeepMind
  • Harrison Lee Google DeepMind
  • Neo Wu Google DeepMind
  • Chao Wang Google DeepMind
  • Sushant Prakash Google DeepMind
  • Shawn O'Banion Google DeepMind
  • Bradley Green Google DeepMind
  • Jun Xie Google DeepMind

DOI:

https://doi.org/10.1609/aaai.v39i24.34738

Abstract

LLM-powered personalization agent systems employ Large Language Models (LLMs) to predict users’ behavior from their past activities. However, their effectiveness often hinges on the ability to effectively leverage extensive, long user historical data due to its inherent noise and length of such data. Existing pre-trained LLMs may generate summaries that are concise but lack the necessary context for downstream tasks, hindering their utility in personalization systems. To address these challenges, we introduce Reinforcement Learning from Prediction Feedback (RLPF). RLPF fine-tunes LLMs to generate concise, human-readable user summaries that are optimized for downstream task performance. By maximizing the usefulness of the generated summaries, RLPF effectively distills extensive user history data while preserving essential information for downstream tasks. Our empirical evaluation demonstrates significant improvements in both extrinsic downstream task utility and intrinsic summary quality, surpassing baseline methods by up to 22% and achieving an up to 84.59% win rate on Factuality, Abstractiveness, and Readability. RLPF also achieves a remarkable 74% reduction while improving performance on 16 out of 19 unseen tasks and/or datasets, showcasing its generalizability. This approach offers a promising solution for enhancing LLM personalization by effectively transforming long, noisy user histories into informative and human-readable representations.

Published

2025-04-11

How to Cite

Wu, J., Ning, L., Liu, L., Lee, H., Wu, N., Wang, C., … Xie, J. (2025). RLPF: Reinforcement Learning from Prediction Feedback for User Summarization with LLMs. Proceedings of the AAAI Conference on Artificial Intelligence, 39(24), 25488–25496. https://doi.org/10.1609/aaai.v39i24.34738

Issue

Section

AAAI Technical Track on Natural Language Processing III