模型

Mind2Dialogue框架训练人类感知语言模型

Mind2Dialogue: Training Human-Aware Language Models by Simulating User Mental States

精选理由

这个新框架挺有意思,它通过模拟用户心理状态来训练AI,让AI能更好地理解我们说话背后的真实想法和目标,比单纯看表面文字更懂人。

Mind2Dialogue通过模拟用户心理状态来训练语言模型,解决现有数据集缺乏用户未言明的信念和目标响应的问题。该框架使用心理学指导的模拟器生成对话,并利用模拟器生成的‘上帝视角’响应来训练模型,使其能更好地理解用户意图。在Qwen、Llama和OLMo等基线模型上,该框架在个性化指标上取得显著提升,尤其在偏好遵循生成任务上提升26.6到40.9个百分点。

原文 · arXiv cs.LG

Mind2Dialogue: Training Human-Aware Language Models by Simulating User Mental States

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.