Tagging the Thought: Unlocking Personalization Reasoning via Reinforcement Learning
Jin, Song, Zhang, Juntian, Liu, Yong, Zhang, Xun, Zhang, Yufei, Jiang, Fei, Yin, Guojun, Lin, Wei, Yan, Rui
–arXiv.org Artificial Intelligence
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.
arXiv.org Artificial Intelligence
Sep-30-2025
- Country:
- North America > United States (0.46)
- Europe > Austria (0.28)
- Asia (0.28)
- Genre:
- Research Report (0.64)
- Technology: