Mind2Dialogue: Training Human-Aware Language Models by Simulating User Mental States
Mind2Dialogue simulates users' mental states to create privileged supervision, boosting personalization metrics of Qwen, Llama, and OLMo assistants by up to 40.9 points.
The paper proposes a psychology-guided simulator that preserves personal characteristics while evolving shared mental states to generate coherent conversations, with an Oracle assistant grounded in those states. Privileged distillation then trains models on Oracle responses so deployed assistants help users without direct mental-state access. Training on the full corpus improves every personalization metric over Qwen, Llama, and OLMo instruction-tuned baselines, including 26.6 to 40.9 percentage point gains in preference-following generation and improved belief/action reasoning.
- Psychology-guided simulator generates coherent conversations from evolving user mental states
- Privileged distillation trains assistants that lack mental-state access at deployment
- 26.6 to 40.9 percentage point gains in preference-following over Qwen, Llama, OLMo baselines
- Evaluation combines personalization and theory-of-mind reasoning tasks
Full article259 words · extracted from arxiv.org · click to collapse
As language models become more capable, long-term collaboration in learning, reasoning, and decision-making calls for a deeper understanding of the people they serve. Yet training such human-aware language models faces a fundamental supervision gap because current datasets for LLM assistant training contain few if any well-informed responses explicitly grounded in users' unspoken beliefs and goals. Scaling such supervision is inherently constrained, as users' underlying states are not directly observable. We thus propose the Mind2Dialogue framework to mitigate this gap by simulating users' mental states and turning them into privileged supervision for human-aware training. Specifically, we first propose a psychology-guided simulator that preserves personal characteristics while updating mental states through interaction to generate coherent conversations. The key idea is to enforce a shared evolving mental state that drives user behavior and guides an Oracle assistant's responses. Our privileged distillation then trains models on the Oracle's well-informed responses to assist users without direct access to their mental states at deployment. Moreover, we propose to evaluate human-aware learning by combining personalization and theory of mind, examining how models understand people and act on that understanding. Training on the full Mind2Dialogue corpus improves every reported personalization metric over the corresponding Qwen, Llama, and OLMo instruction-tuned baselines, including gains of 26.6 to 40.9 percentage points in preference-following generation. The gains extend to belief and action reasoning on Qwen and Llama, beyond personalized assistance. Looking forward, Mind2Dialogue makes user simulation a foundation for genuine AI collaborators that understand beliefs and intentions behind people's words and support their long-term goals across education, work, and everyday life.
Text extracted automatically; images, tables and formatting may be missing. Original: https://arxiv.org/abs/2609.15972