cs.CLFeb 22, 2026

Learning to Reason for Multi-Step Retrieval of Personal Context in Personalized Question Answering

Authors: Maryam Amirizaniani, Alireza Salemi, Hamed Zamani

Organizations: University of Washington Seattle, WA, USA · University of Massachusetts Amherst Amherst, MA, USA

Abstract

Personalization in Question Answering (QA) requires answers that are both accurate and aligned with users' background, preferences, and historical context. Existing state-of-the-art methods primarily rely on retrieval-augmented generation (RAG) solutions that construct personal context by retrieving relevant items from the user's profile. Existing methods use the user's query directly to retrieve personal documents, and such strategies often lead to surface-level personalization. We propose PR2 (Personalized Retrieval-Augmented Reasoning), a reinforcement learning framework that integrates reasoning and retrieval from personal context for personalization. PR2 learns adaptive retrieval-reasoning policies, determining when to retrieve, what evidence to retrieve from user profiles, and how to incorporate it into intermediate reasoning steps. By optimizing multi-turn reasoning trajectories under a personalized reward function, the framework reinforces reasoning paths that better align with user-specific preferences and contextual signals reflected by the reward model. Extensive experiments on the LaMP-QA benchmark using three LLMs show that PR2 consistently outperforms strong baselines, achieving an average relative improvement of 8.8%-12% in personalized QA.

Figures & tables

Explore similar work

CardsList
  1. Training LLMs with Reinforcement Learning for Intent-Aware Personalized Question Answering

    May 12, 2026Maryam Amirizaniani, Benjamin Charles Germain Lee, Jevin West +1RL for Language ModelsLLM Personalization

  2. CoPA: Benchmarking Personalized Question Answering with Data-Informed Cognitive Factors

    Apr 16, 2026Hang Su, Zequn Liu, Chen Hu +3LLM PersonalizationQuestion Answering