DialToM
收藏资源简介:
DialToM是由新加坡管理大学和澳大利亚国立大学联合构建的心理理论推理评测基准,基于真实人类对话数据构建并经过人工验证。该数据集包含5943条上下文窗口,覆盖动机访谈、情感支持和说服对话三个领域,每条数据包含6个回顾性推理问题和1个前瞻性预测问题。数据来源于AnnoMI、ESConv和PersuasionForGood三个高质量对话语料库,通过分段和标准化处理确保数据一致性。该数据集旨在评估语言模型从心理状态推断到社会轨迹预测的完整推理链条,特别关注信任维度在复杂社交互动中的作用,为对话系统和社交预测研究提供重要基准。
DialToM is a Theory of Mind reasoning evaluation benchmark jointly developed by the Singapore Management University and the Australian National University, constructed based on real human conversational data and manually validated. This dataset comprises 5,943 context windows, covering three domains: motivational interviewing, emotional support, and persuasive dialogue. Each entry contains six retrospective reasoning questions and one prospective prediction question. The data is sourced from three high-quality conversational corpora: AnnoMI, ESConv, and PersuasionForGood, and processed through segmentation and standardization to ensure data consistency. This dataset aims to evaluate the complete reasoning chain of language models, spanning from mental state inference to social trajectory prediction, with particular focus on the role of the trust dimension in complex social interactions, serving as an important benchmark for research on conversational systems and social prediction.
根据您提供的页面地址和README文件内容,以下是对数据集详情页的总结:
数据集概述
DialToM 是一个用于预测状态驱动对话轨迹的“心智理论”(Theory of Mind)基准数据集,相关论文已投稿至KDD 2026 DnB轨道。
数据集内容
- 数据文件:位于
data目录下,包含经人工验证版本的 DialToM 数据集。 - 反事实数据:位于
counterfactual_data目录下,包含三个文件,每个数据集对应所有生成的反事实样本,用于反事实消融研究。
基准测试任务
数据集支持以下三种评估任务:
-
回顾性任务(Retrospective)
- 运行命令:
python benchmark.py --model [模型] --task retrospective --filename retrospective.csv
- 运行命令:
-
前瞻性任务(Prospective)
- 运行命令:
python benchmark.py --model [模型] --task prospective --exp [实验类型] --filename prospective.csv - 支持四种实验类型:
normal(默认基线)、easy(简单集评估)、NOTA和CoT(两种消融实验)。 - 输出文件名会根据实验类型动态变为
{filename}_{实验类型}.csv。
- 运行命令:
-
书面任务(Written)
- 运行命令:
python benchmark.py --model [模型] --task written --filename written.csv
- 运行命令:
其他研究模块
- 反事实测试:运行
python counterfactual_test.py --model [模型] --filename counter.csv - 记忆性试点研究:运行
python memorization_pilot.py --model [模型] --filename memorize.csv
依赖环境
需要安装以下Python包:
google-genaiopenaisacrebleurougebert-score




