DEMO
收藏资源简介:
DEMO是一个专为对话建模设计的基准,旨在促进对话模型的全面建模和评估。该数据集涵盖了从开场白到对话结束的系统性对话框架,并定义了一个创新的研究任务:对话元素建模。任务重点在于模型的元素意识和对话代理交互能力。数据集包含适用于英语和中文的多方面元素,并结合了模仿学习的专家经验。
DEMO is a benchmark specifically designed for dialogue modeling, aiming to facilitate comprehensive modeling and evaluation of dialogue models. This dataset covers a systematic dialogue framework spanning from opening lines to the conclusion of a conversation, and defines an innovative research task: Dialogue Element Modeling. The task focuses on the model's element awareness and the interactive capabilities of dialogue agents. This dataset includes diverse elements applicable to both English and Chinese, and incorporates expert experience from imitation learning.
DEMO: Reframing Dialogue Interaction with Fine-grained Element Modeling
概述
该仓库包含我们论文DEMO的代码:通过细粒度元素建模重构对话交互。我们系统地从前奏到对话再到尾声构建对话框架,并定义了一个创新的研究任务:对话元素建模。此外,我们引入了一个量身定制的基准DEMO,以促进全面的对话建模和评估。具体来说,我们提出的任务侧重于模型的两个核心能力:(1)元素意识,涉及逆向工程以分解对话元素,以及(2)对话代理交互,涉及由元素驱动的目标导向的多轮对话建模。我们精心设计了一个数据合成框架,为对话建模贡献了一个包含多方面元素的新基准,适用于英语和中文。此外,受模仿学习的启发,我们收集了大量的专家经验,并构建了一个具有对话元素建模能力的DEMO代理。
更新
- [2024.12.07] DEMO发布!我们发布了论文、代码、模型和数据。
使用方法
步骤1 下载DEMO并解压数据 shell git clone https://github.com/MozerWang/DEMO.git cd DEMO unzip data/DEMO.zip -d data/
步骤2 创建conda环境并安装其他依赖项 shell conda create --name loong python=3.9 -y conda activate DEMO pip install -r requirements.txt
步骤3 准备模型
-
(必须)在config/gpt_4o.yaml中设置您的OPENAI密钥 shell api_key: "Your OPENAI key"
-
如果您使用基于API的大型语言模型 shell
首先,在config/*.yaml中设置您的密钥
api_key: "Your API key"
- 如果您使用开源的大型语言模型 shell
我们推荐使用vLLM。我们使用实现OpenAI的Completions和Chat API的HTTP服务器。
在config/*.yaml中设置您的vLLM设置
步骤4 评估 shell sh run.sh
引用
@article{wang2024demo, title={DEMO: Reframing Dialogue Interaction with Fine-grained Element Modeling}, author={Minzheng Wang and Xinghua Zhang and Kun Chen and Nan Xu and Haiyang Yu and Fei Huang and Wenji Mao and Yongbin Li}, year={2024}, journal={arXiv preprint arXiv:2412.04905}, url={https://arxiv.org/abs/2412.04905}, year={2023} }




