English

Tagging the Thought: Unlocking Personalization Reasoning via Reinforcement Learning

Computation and Language 2025-09-30 v1

Abstract

Recent advancements have endowed Large Language Models (LLMs) with impressive general reasoning capabilities, yet they often struggle with personalization reasoning - the crucial ability to analyze user history, infer unique preferences, and generate tailored responses. To address this limitation, we introduce TagPR, a novel training framework that significantly enhances an LLM's intrinsic capacity for personalization reasoning through a tagging the thought approach. Our method first develops a data-driven pipeline to automatically generate and semantically label reasoning chains, creating a structured dataset that fosters interpretable reasoning. We then propose a synergistic training strategy that begins with Supervised Fine-Tuning (SFT) on this tagged data to establish foundational reasoning patterns, followed by a multi-stage reinforcement learning (RL) process. This RL phase is guided by a unique composite reward signal, which integrates tag-based constraints and a novel Personalization Reward Model with User Embeddings (PRMU) to achieve fine-grained alignment with user-specific logic. Extensive experiments on the public LaMP benchmark and a self-constructed dataset demonstrate that our approach achieves state-of-the-art results, delivering an average improvement of 32.65% over the base model across all tasks. Our work validates that structured, interpretable reasoning is a highly effective pathway to unlocking genuine personalization capabilities in LLMs.

Keywords

Cite

@article{arxiv.2509.23140,
  title  = {Tagging the Thought: Unlocking Personalization Reasoning via Reinforcement Learning},
  author = {Song Jin and Juntian Zhang and Yong Liu and Xun Zhang and Yufei Zhang and Fei Jiang and Guojun Yin and Wei Lin and Rui Yan},
  journal= {arXiv preprint arXiv:2509.23140},
  year   = {2025}
}
R2 v1 2026-07-01T06:00:25.231Z