# 对话通用理解模块DGU ## 目录 - [**模型简介**](#模型简介) - [**快速开始**](#快速开始) - [**进阶使用**](#进阶使用) - [**参考论文**](#参考论文) - [**版本更新**](#版本更新) ## 模型简介     对话相关的任务中，Dialogue System常常需要根据场景的变化去解决多种多样的任务。任务的多样性（意图识别、槽位解析、DA识别、DST等等），以及领域训练数据的稀少，给Dialogue System的研究和应用带来了巨大的困难和挑战，要使得dialogue system得到更好的发展，需要开发一个通用的对话理解模型。为此，我们给出了基于BERT的对话通用理解模块(DGU: DialogueGeneralUnderstanding)，通过实验表明，使用base-model(BERT)并结合常见的学习范式，就可以在几乎全部对话理解任务上取得比肩甚至超越各个领域业内最好的模型的效果，展现了学习一个通用对话理解模型的巨大潜力。 ## 快速开始 ### 安装说明 ####   a、环境依赖 - Python >= 2.7 - cuda >= 9.0 - cudnn >= 7.0 - PaddlePaddle >= 1.3.1，请参考[安装指南](http://www.paddlepaddle.org/#quick-start)进行安装, 由于模块内模型基于bert做finetuning, 训练速度较慢, 建议用户安装GPU版本PaddlePaddle进行训练。   注意：使用Windows GPU环境的用户，需要将示例代码中的[fluid.ParallelExecutor](http://paddlepaddle.org/documentation/docs/zh/1.4/api_cn/fluid_cn.html#parallelexecutor)替换为[fluid.Executor](http://paddlepaddle.org/documentation/docs/zh/1.4/api_cn/fluid_cn.html#executor)。 ####   b、下载代码     克隆代码库到本地 ``` git clone https://github.com/PaddlePaddle/models.git cd models/PaddleNLP/dialogue_model_toolkit/dialogue_general_understanding ``` ### 任务简介     本模块内共包含6个任务，内容如下： ``` udc: 使用Ubuntu Corpus V1公开数据集，实现对话匹配任务; atis_slot: 使用微软提供的公开数据集(Airline Travel Information System)，实现槽位识别任务； dstc2: 使用对话状态跟踪挑战（Dialog State Tracking Challenge）2公开数据集，实现对话状态追踪（DST）任务; atis_intent: 使用微软提供的公开数据集(Airline Travel Information System)，实现意图识别任务； mrda: 使用公开数据集Meeting Recorder Dialogue Act，实现DA识别任务; swda：使用公开数据集Switchboard Dialogue Act Corpus，实现DA识别任务; ``` 注意: 目前dgu模块内提供的训练好的官方模型及效果, 均是在GPU单卡上面训练和预测得到的, 用户如需复线效果, 可使用单卡相同的配置. ### 数据准备     数据集说明： ``` UDC: Ubuntu Corpus V1; ATIS: 微软提供的公开数据集(Airline Travel Information System)，模块内包含意图识别和槽位解析两个任务的数据; DSTC2: 对话状态跟踪挑战（Dialog State Tracking Challenge）2; MRDA: Meeting Recorder Dialogue Act; SWDA：Switchboard Dialogue Act Corpus; ```     数据集、相关模型下载： ``` cd dgu && bash prepare_data_and_model.sh ```     数据路径：data/input/data     预训练模型路径：data/pretrain_model     已训练模型路径：data/saved_models/trained_models     下载的数据集中已提供了训练集，测试集和验证集，用户如果需要重新生成某任务数据集的训练数据，可执行： ``` cd dgu/scripts && bash run_build_data.sh task_name 参数说明： task_name: udc, swda, mrda, atis, dstc2, 选择5个数据集选项中用户需要生成的数据名 各任务数据生成脚本描述： dgu/scripts/build_atis_dataset.py：将ATIS开源数据集转换成训练所需的意图识别（atis_intent）和槽位解析（atis_slot）训练数据 dgu/scripts/build_dstc2_dataset.py：将DSTC2开源数据集转换成训练所需数据格式； dgu/scripts/build_mrda_dataset.py：将MRDA开源数据集转换成训练所需数据格式； dgu/scripts/build_swda_dataset.py：将SWDA开源数据集转换成训练所需数据格式； ```     根据脚本构造的训练数据格式说明： ``` udc：数据组成，label和多轮对话(分为多轮上文和当前回复)，整体分割符为"\t" format: label \t conv1 \t conv2 \t conv3 \t ......\t response swda：数据组成，多轮对话id, 标签label, 发言人caller, 说话内容conversation_content，整体分割符为"\t" format： conversation_id \t label \t caller \t conversation_content mrda: 数据组成，多轮对话id, 标签label, 发言人caller, 说话内容conversation_content，整体分割符为"\t" format： conversation_id \t label \t caller \t conversation_content atis/atis_intent: 数据组成，标签label, 说话内容conversation_content，整体分割符为"\t" format： label \t conversation_content atis/atis_slot: 数据组成，说话内容conversation_content，标签序列 label_list（空格分割）, 其中标签和说话内容中token为一一对应关系，整体分割符为"\t" format： conversation_content \t label1 label2 label3 dstc2/dstc2: 数据组成，多轮对话id, 当前轮QA对(使用\1拼接)，标签(识别到的对话状态，从对话初始状态到当前轮累计的标签集合, 空格分割)，整体分割符为"\t" format：conversation_content \t question \1 answer \t state1 state2 state3...... ``` ### 模型配置     配置文件路径: data/config/dgu.yaml ``` task_name: 任务名称，可选udc、swda、mrda、atis_intent、atis_slot、dstc2 data_dir: 数据路径，如./data/input/data/udc bert_config_path: 预训练模型bert的网络配置./data/pretrain_model/uncased_L-12_H-768_A-12/bert_config.json init_from_checkpoint: 加载断点模型 init_from_params: 训练好的模型参数文件，一般用于预测 init_from_pretrain_model: 预训练模型路径，如bert的模型参数 inference_model_dir: inference model的保存路径 save_model_path: 训练产出模型的输出路径 save_checkpoint: 调用paddle的io接口save_persistables(把传入的层中所有参数以及优化器进行保存)来保存模型参数 save_param: 调用paddle的io接口save_params(从main_program中取出所有参数然后保存到文件中)来保存模型参数 lr_scheduler: learning rate scheduler weight_decay: learning rate 权重衰减因子 warmup_proportion: warmup比率 save_steps: 每隔save_steps个步数来保存一次模型 use_fp16: 是否使用fp16来训练模型 loss_scaling: loss权重因子 print_steps: 每隔print_steps个步数打印一次日志 evaluation_file: 参与评估的inference 标注文件 output_prediction_file: 输出的预测文件 vocab_path: 模型词表 max_seq_len: 输入bert内的最大序列长度 batch_size: 一个batch内输入的样本个数 do_lower_case: 是否进行大小写转换 random_seed: 随机种子设置 use_cuda: 是否使用cuda, 如果是gpu训练时，设置成true in_tokens: 是否采用in_tokens模式来计算batch_siz数量, 如果in_tokens为false, 则batch_size等于真实设置的batch_size大小, 如果in_tokens为true, 则batch_size=batch_size*max_seq_len，即按照token计数 do_save_inference_model: 是否保存inference model encable_ce: 是否开启ce ``` ### 单机训练 ####     方式一: 推荐直接使用模块内脚本训练 ``` bash run.sh task_name task_type 参数说明： task_name: udc, swda, mrda, atis_intent, atis_slot, dstc2，选择6个任务中任意一项； task_type: train，predict, evaluate, inference, all, 选择5个参数选项中任意一项(train: 只执行训练，predict: 只执行预测，evaluate：只执行评估过程，依赖预测的结果，inference: 保存inference model，all: 顺序执行训练、预测、评估、保存inference model的过程)； 训练示例： bash run.sh atis_intent train ```     如果为CPU训练: ``` 请将run.sh内参数设置为: 1、export CUDA_VISIBLE_DEVICES= ```     如果为GPU训练: ``` 请将run.sh内参数设置为: 1、如果为单卡训练（用户指定空闲的单卡）： export CUDA_VISIBLE_DEVICES=0 2、如果为多卡训练（用户指定空闲的多张卡）： export CUDA_VISIBLE_DEVICES=0,1,2,3 ``` ####     方式二: 执行训练相关的代码: ``` export FLAGS_sync_nccl_allreduce=0 export FLAGS_eager_delete_tensor_gb=1 #开启显存优化 export CUDA_VISIBLE_DEVICES=0 #GPU单卡训练 #export CUDA_VISIBLE_DEVICES=0,1,2,3 #GPU多卡训练 #export CUDA_VISIBLE_DEVICES= #CPU训练 if [ ! "$CUDA_VISIBLE_DEVICES" ] then use_cuda=false else use_cuda=true fi TASK_NAME="atis_intent" #指定训练的任务名称 BERT_BASE_PATH="data/pretrain_model/uncased_L-12_H-768_A-12" if [ -f "./data/saved_models/${TASK_NAME}" ]; then rm "./data/saved_models/${TASK_NAME}" fi if [ ! -d "./data/saved_models/${TASK_NAME}" ]; then mkdir "./data/saved_models/${TASK_NAME}" fi python -u main.py \ --task_name=${TASK_NAME} \ --use_cuda=${use_cuda} \ --do_train=true \ --in_tokens=true \ --epoch=20 \ --batch_size=4096 \ --do_lower_case=true \ --data_dir="./data/input/data/atis/${TASK_NAME}" \ --bert_config_path="${BERT_BASE_PATH}/bert_config.json" \ --vocab_path="${BERT_BASE_PATH}/vocab.txt" \ --init_from_pretrain_model="${BERT_BASE_PATH}/params" \ --save_model_path="./data/saved_models/${TASK_NAME}" \ --save_param="params" \ --save_steps=100 \ --learning_rate=2e-5 \ --weight_decay=0.01 \ --max_seq_len=128 \ --print_steps=10 ``` 注： 1) 采用方式二时，模型训练过程可参考run.sh内相关任务的参数设置 2) 用户进行模型训练、预测、评估等, 可通过修改data/config/dgu.yaml配置文件或者从命令行传入来进行参数配置, 优先推荐命令行参数传入; ### 模型预测 ####     方式一: 推荐直接使用模块内脚本预测 ``` bash run.sh task_name task_type 参数说明： task_name: udc, swda, mrda, atis_intent, atis_slot, dstc2，选择6个任务中任意一项； task_type: train，predict, evaluate, inference, all, 选择5个参数选项中任意一项(train: 只执行训练，predict: 只执行预测，evaluate：只执行评估过程，依赖预测的结果，inference: 保存inference model，all: 顺序执行训练、预测、评估、保存inference model的过程)； 预测示例： bash run.sh atis_intent predict ```     如果为CPU预测: ``` 请将run.sh内参数设置为: 1、export CUDA_VISIBLE_DEVICES= ```     如果为GPU预测: ``` 请将run.sh内参数设置为: 支持单卡预测（用户指定空闲的单卡）： export CUDA_VISIBLE_DEVICES=0 ``` 注：预测时，如采用方式一，用户可通过修改run.sh中init_from_params参数来指定自己训练好的需要预测的模型，目前代码中默认为加载官方已经训练好的模型; ####     方式二: 执行预测相关的代码: ``` export FLAGS_sync_nccl_allreduce=0 export FLAGS_eager_delete_tensor_gb=1 #开启显存优化 export CUDA_VISIBLE_DEVICES=0 #单卡预测 #export CUDA_VISIBLE_DEVICES= #CPU预测 if [ ! "$CUDA_VISIBLE_DEVICES" ] then use_cuda=false else use_cuda=true fi TASK_NAME="atis_intent" #指定预测的任务名称 BERT_BASE_PATH="./data/pretrain_model/uncased_L-12_H-768_A-12" python -u main.py \ --task_name=${TASK_NAME} \ --use_cuda=${use_cuda} \ --do_predict=true \ --in_tokens=true \ --batch_size=4096 \ --do_lower_case=true \ --data_dir="./data/input/data/atis/${TASK_NAME}" \ --init_from_params="./data/saved_models/trained_models/${TASK_NAME}/params" \ --bert_config_path="${BERT_BASE_PATH}/bert_config.json" \ --vocab_path="${BERT_BASE_PATH}/vocab.txt" \ --output_prediction_file="./data/output/pred_${TASK_NAME}" \ --max_seq_len=128 ``` 注：采用方式二时，模型预测过程可参考run.sh内具体任务的参数设置 ### 模型评估     模块中6个任务，各任务支持计算的评估指标内容如下： ``` udc: 使用R1@10、R2@10、R5@10三个指标评估匹配任务的效果; atis_slot: 使用F1指标来评估序列标注任务； dstc2: 使用joint acc 指标来评估DST任务的多标签分类结果; atis_intent: 使用acc指标来评估分类结果； mrda: 使用acc指标来评估DA任务分类结果; swda：使用acc指标来评估DA任务分类结果; ```     效果上，6个任务公开数据集评测效果如下表所示： | task_name | udc | udc | udc | atis_slot | dstc2 | atis_intent | swda | mrda | | :------ | :------ | :------ | :------ | :------| :------ | :------ | :------ | :------ | | 对话任务 | 匹配 | 匹配 | 匹配 | 槽位解析 | DST | 意图识别 | DA | DA | | 任务类型 | 分类 | 分类 | 分类 | 序列标注 | 多标签分类 | 分类 | 分类 | 分类 | | 任务名称 | udc | udc | udc| atis_slot | dstc2 | atis_intent | swda | mrda | | 评估指标 | R1@10 | R2@10 | R5@10 | F1 | JOINT ACC | ACC | ACC | ACC | | SOTA | 76.70% | 87.40% | 96.90% | 96.89% | 74.50% | 98.32% | 81.30% | 91.70% | | DGU | 82.03% | 90.59% | 97.73% | 97.14% | 91.23% | 97.76% | 80.37% | 91.53% | ####     方式一: 推荐直接使用模块内脚本评估 ``` bash run.sh task_name task_type 参数说明： task_name: udc, swda, mrda, atis_intent, atis_slot, dstc2，选择6个任务中任意一项； task_type: train，predict, evaluate, inference, all, 选择5个参数选项中任意一项(train: 只执行训练，predict: 只执行预测，evaluate：只执行评估过程，依赖预测的结果，inference: 保存inference model，all: 顺序执行训练、预测、评估、保存inference model的过程)； 评估示例： bash run.sh atis_intent evaluate ``` 注：评估计算ground_truth和predict_label之间的打分，默认CPU计算即可； ####     方式二: 执行评估相关的代码: ``` TASK_NAME="atis_intent" #指定预测的任务名称 python -u main.py \ --task_name=${TASK_NAME} \ --use_cuda=false \ --do_eval=true \ --evaluation_file="./data/input/data/atis/${TASK_NAME}/test.txt" \ --output_prediction_file="./data/output/pred_${TASK_NAME}" ``` ### 模型推断 ####     方式一: 推荐直接使用模块内脚本保存inference model ``` bash run.sh task_name task_type 参数说明： task_name: udc, swda, mrda, atis_intent, atis_slot, dstc2，选择6个任务中任意一项； task_type: train，predict, evaluate, inference, all, 选择5个参数选项中任意一项(train: 只执行训练，predict: 只执行预测，evaluate：只执行评估过程，依赖预测的结果，inference: 保存inference model，all: 顺序执行训练、预测、评估、保存inference model的过程)； 保存模型示例： bash run.sh atis_intent inference ```     如果为CPU执行inference model过程: ``` 请将run.sh内参数设置为: 1、export CUDA_VISIBLE_DEVICES= ```     如果为GPU执行inference model过程: ``` 请将run.sh内参数设置为: 1、单卡模型推断（用户指定空闲的单卡）： export CUDA_VISIBLE_DEVICES=0 ``` ####     方式二: 执行inference model相关的代码: ``` TASK_NAME="atis_intent" #指定预测的任务名称 BERT_BASE_PATH="./data/pretrain_model/uncased_L-12_H-768_A-12" export CUDA_VISIBLE_DEVICES=0 #单卡推断inference model #export CUDA_VISIBLE_DEVICES= #CPU预测 if [ ! "$CUDA_VISIBLE_DEVICES" ] then use_cuda=false else use_cuda=true fi python -u main.py \ --task_name=${TASK_NAME} \ --use_cuda=${use_cuda} \ --do_save_inference_model=true \ --init_from_params="./data/saved_models/trained_models/${TASK_NAME}/params" \ --bert_config_path="${BERT_BASE_PATH}/bert_config.json" \ --inference_model_dir="data/inference_models/${TASK_NAME}" ``` ### 预训练模型     支持PaddlePaddle官方提供的BERT及ERNIE相关模型作为预训练模型 | Model | Layers | Hidden size | Heads |Parameters | | :------| :------: | :------: |:------: |:------: | | [BERT-Base, Uncased](https://bert-models.bj.bcebos.com/uncased_L-12_H-768_A-12.tar.gz) | 12 | 768 |12 |110M | | [BERT-Large, Uncased](https://bert-models.bj.bcebos.com/uncased_L-24_H-1024_A-16.tar.gz) | 24 | 1024 |16 |340M | |[BERT-Base, Cased](https://bert-models.bj.bcebos.com/cased_L-12_H-768_A-12.tar.gz)|12|768|12|110M| |[BERT-Large, Cased](https://bert-models.bj.bcebos.com/cased_L-24_H-1024_A-16.tar.gz)|24|1024|16|340M| |[ERNIE, english](https://ernie.bj.bcebos.com/ERNIE_en_1.0.tgz)|24|1024|16|3.8G| ### 服务部署     模块内提供已训练好6个对话任务的inference_model模型，用户可根据自身业务情况进行下载使用。 #### 服务器部署     请参考PaddlePaddle官方提供的[服务器端部署](https://www.paddlepaddle.org.cn/documentation/docs/zh/1.5/advanced_usage/deploy/inference/index_cn.html)文档进行部署上线。 ## 进阶使用 ### 背景介绍     dialogue_general_understanding模块，针对数据集开发了相关的模型训练过程，支持分类，多标签分类，序列标注等任务，用户可针对自己的数据集，进行相关的模型定制；并取得了比肩业内最好模型的效果： ### 模型概览

    训练、预测、评估使用的数据可以由用户根据实际的对话应用场景，自己组织数据。输入网络的数据格式统一为，示例如下： ``` [CLS] token11 token12 token13 [INNER_SEP] token11 token12 token13 [SEP] token21 token22 token23 [SEP] token31 token32 token33 [SEP] ```     输入数据以[CLS]开始，[SEP]分割对话内容（上文、当前句、下文等），如果[SEP]分割的每部分内部由多轮组成的话，使用[INNER_SEP]进行分割；第二部分和第三部分部分皆可缺省；     目前dialogue_general_understanding模块内已将数据准备部分集成到代码内，用户可根据上面输入数据格式，组装自己的数据；     用户也可以根据自己的需求，组建自定义的模型，具体方法如下所示：     a、自定义数据       如用户目前有数据集为**task_name**, 则在**data/input/data**下定义**task_name**文件夹，将数据集存放进去；在**dgu/reader.py**中，新增自定义的数据处理的类，如**udc**数据集对应**UDCProcessor**; 在**train.py**内设置**task_name**和**processor**的对应关系(如**processors = {'udc': reader.UDCProcessor}**).     b、 自定义上层网络范式       如果用户自定义模型属于分类、多分类和序列标注这3种类型其中一个，则只需要在**dgu/define_paradigm.py** 内指明**task_name**和相应上层范式函数的对应关系即可，如用户自定义模型属于其他模型，则需要自定义上层范式函数并指明其与**task_name**之间的关系；     c、自定义预测封装接口       用户可在**dgu/define_predict_pack.py**内定义task_name和自定义封装预测接口的对应关系； ## 参考论文 1、Harshit Kumar, Arvind Agarwal, Riddhiman Dasgupta,Sachindra Joshi, and Arun Kumar. 2017. Dia-logue act sequence labeling using hierarchical en-coder with crf.arXiv preprint arXiv:1709.04250. 2、Changliang Li, Liang Li, and Ji Qi. 2018. A self-attentive model with gate mechanism for spoken lan-guage understanding. InProceedings of the 2018Conference on Empirical Methods in Natural Lan-guage Processing, pages 3824–3833. 3、Ryan Lowe, Nissan Pow, Iulian Serban, and JoellePineau. 2015. The ubuntu dialogue corpus: A largedataset for research in unstructured multi-turn dia-logue systems.arXiv preprint arXiv:1506.08909. 4、Tomas Mikolov, Ilya Sutskever, Kai Chen, Greg S Cor-rado, and Jeff Dean. 2013. Distributed representa-tions of words and phrases and their compositional-ity. InAdvances in neural information processingsystems, pages 3111–3119. 5、Hiroki Ouchi and Yuta Tsuboi. 2016. Addressee andresponse selection for multi-party conversation. InProceedings of the 2016 Conference on EmpiricalMethods in Natural Language Processing, pages2133–2143. 6、Elizabeth Shriberg, Raj Dhillon, Sonali Bhagat, JeremyAng, and Hannah Carvey. 2004. The icsi meetingrecorder dialog act (mrda) corpus. Technical report,INTERNATIONAL COMPUTER SCIENCE INSTBERKELEY CA. 7、Andreas Stolcke, Klaus Ries, Noah Coccaro, Eliza-beth Shriberg, Rebecca Bates, Daniel Jurafsky, PaulTaylor, Rachel Martin, Carol Van Ess-Dykema, andMarie Meteer. 2000. Dialogue act modeling for au-tomatic tagging and recognition of conversationalspeech.Computational linguistics, 26(3):339–373. 8、Ye-Yi Wang, Li Deng, and Alex Acero. 2005. Spo-ken language understanding.IEEE Signal Process-ing Magazine, 22(5):16–31.Jason Williams, Antoine Raux, Deepak Ramachan-dran, and Alan Black. 2013. The dialog state tracking challenge. InProceedings of the SIGDIAL 2013Conference, pages 404–413. 9、Yonghui Wu, Mike Schuster, Zhifeng Chen, Quoc VLe, Mohammad Norouzi, Wolfgang Macherey,Maxim Krikun, Yuan Cao, Qin Gao, KlausMacherey, et al. 2016. Google’s neural ma-chine translation system: Bridging the gap betweenhuman and machine translation.arXiv preprintarXiv:1609.08144.Kaisheng 10、Yao, Geoffrey Zweig, Mei-Yuh Hwang,Yangyang Shi, and Dong Yu. 2013. Recurrent neu-ral networks for language understanding. InInter-speech, pages 2524–2528. 11、Xiangyang Zhou, Lu Li, Daxiang Dong, Yi Liu, YingChen, Wayne Xin Zhao, Dianhai Yu, and Hua Wu.2018. Multi-turn response selection for chatbotswith deep attention matching network. InProceed-ings of the 56th Annual Meeting of the Associationfor Computational Linguistics (Volume 1: Long Pa-pers), volume 1, pages 1118–1127. 12、Su Zhu and Kai Yu. 2017. Encoder-decoder withfocus-mechanism for sequence labelling based spo-ken language understanding. In2017 IEEE Interna-tional Conference on Acoustics, Speech and SignalProcessing (ICASSP), pages 5675–5679. IEEE. 13、Jason Williams, Antoine Raux, Deepak Ramachan-dran, and Alan Black. 2013. The dialog state track-ing challenge. InProceedings of the SIGDIAL 2013Conference, pages 404–413. ## 版本更新 第一版：PaddlePaddle 1.4.0版本 主要功能：支持对话6个数据集上任务的训练、预测和评估 第二版：PaddlePaddle 1.6.0版本 更新功能：在第一版的基础上，根据PaddlePaddle的模型规范化标准，对模块内训练、预测、评估等代码进行了重构，提高易用性； ## 如何贡献代码     如果你可以修复某个issue或者增加一个新功能，欢迎给我们提交PR。如果对应的PR被接受了，我们将根据贡献的质量和难度进行打分（0-5分，越高越好）。如果你累计获得了10分，可以联系我们获得面试机会或者为你写推荐信。