高级检索

基于文档对话系统中一种参考知识识别新方法

A New Method of Reference Knowledge Identification in Document-based Dialogue Systems

  • 摘要: 在基于文档的对话系统中,参考知识识别要求根据对话历史从外部相关文档中识别出回复当前对话所依赖的相关信息。参考知识识别模型是基于文档的对话系统的必要模块,对生成回复的质量有重要影响。现有方法通常采用基于预训练的边界分类模型,然而,由于参考知识通常难以唯一确定,使用单一参考知识标签训练的模型面临鲁棒性差、泛化性低等问题。为此,该文提出了一种基于集成式自蒸馏的参考知识识别方法,该方法利用模型自身提供的参考知识软标签,依据提出的自蒸馏损失再次训练模型。集成后,模型在Doc2dial测试集上的F1值和完全匹配率分别达到了76.71%和63.92%,用预测的参考知识生成回复的BLEU值相较RoBERTa模型提升了1.7。

     

    Abstract: In document-based dialogue systems, reference knowledge identification aims to identify the user query related information from the relevant document given the dialogue history. Based on ensemble self-knowledge distillation, this paper proposes a reference knowledge identification method. It uses the soft labels of reference knowledge provided by model itself and retains the model based on the proposed distillation loss. Experiments on the Doc2dial test set demonstrate that the F1 measure and Exact Match rate reaches 76.71% and 63.92%, respectively, and the BLEU score of the reference knowledge powered by these reference knowledge is 1.7 higher than that by the RoBERTa model.

     

/

返回文章
返回