DEMO
收藏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} }




