Search papers, labs, and topics across Lattice.
This paper introduces Psy-CoT, a psychology-grounded chain-of-thought framework designed to enhance general role-playing agents by incorporating a structured reasoning process that includes Interaction Perception, Psychological Empathy, and Logical Construction. The authors identify limitations in traditional supervised fine-tuning, which often leads to superficial mimicry and poor generalization, and propose Role-Aware Policy Optimization (RAPO) to mitigate reward hacking by adjusting gradient signals based on profile-token mutual information. Experimental results show that Psy-CoT significantly outperforms existing methods on multiple benchmarks, demonstrating improved character fidelity and dynamic reasoning capabilities in role-playing agents.
Role-playing agents can achieve human-like character fidelity by integrating psychology-grounded reasoning with advanced reinforcement learning techniques.
Building general-purpose role-playing agents that faithfully portray any character from a natural-language profile remains challenging. The dominant paradigm -- supervised fine-tuning -- encourages behavioral mimicry without deep, human-like internal thought processes, resulting in poor out-of-distribution generalization. Therefore, we propose \textbf{Psy-CoT}, a psychology-grounded chain-of-thought framework that decomposes pre-response reasoning into three role-specific steps -- \emph{Interaction Perception}, \emph{Psychological Empathy}, and \emph{Logical Construction} -- so that the model \emph{thinks dynamically} from the profile rather than merely mimicking surface patterns. While structured reasoning provides a foundation, it alone is insufficient; reinforcement learning is essential to further align the model with character fidelity. However, we observe that under LLM-based reward models, both generic phrases that hack the reward model and genuinely role-specific phrases receive identical gradient signals -- this hacking accumulates over training, misleading the model into treating both as equally optimal choices. To address this, we propose \textbf{Role-Aware Policy Optimization (RAPO)}, which uses profile--token mutual information to weight gradients asymmetrically -- amplifying role-specific tokens under positive advantage while attenuating them under negative advantage. Experiments on CoSER, CharacterBench, and CharacterEval demonstrate that Psy-CoT outperforms existing role-playing CoT methods, and RAPO consistently surpasses GRPO across multiple model scales.