Author Login Editor-in-Chief Peer Review Editor Work Office Work

Computer Engineering ›› 2023, Vol. 49 ›› Issue (2): 54-60. doi: 10.19678/j.issn.1000-3428.0063847

• Artificial Intelligence and Pattern Recognition • Previous Articles     Next Articles

Unknown Intent Detection for Task-Oriented Dialogs Based on Reconstruction Error

BI Ran1,2,3, WANG Yi1,3, ZHOU Xi1,3   

  1. 1. Xinjiang Technical Institute of Physics and Chemistry, Chinese Academy of Sciences, Urumqi 830011, China;
    2. University of Chinese Academy of Sciences, Beijing 100049, China;
    3. Xinjiang Laboratory of Minority Speech and Language Information Processing, Urumqi 830011, China
  • Received:2022-01-26 Revised:2022-03-08 Published:2022-07-05

基于重建误差的任务型对话未知意图检测

毕然1,2,3, 王轶1,3, 周喜1,3   

  1. 1. 中国科学院新疆理化技术研究所, 乌鲁木齐 830011;
    2. 中国科学院大学, 北京 100049;
    3. 新疆民族语音语言信息处理实验室, 乌鲁木齐 830011
  • 作者简介:毕然(1995-),男,硕士研究生,主研方向为任务型对话系统;王轶,副研究员、博士;周喜,研究员、博士。
  • 基金资助:
    中国科学院西部之光人才培养引进计划“基于回归分析的新疆高考志愿推荐方法研究”(2018-XBQNXZ-A-003);新疆维吾尔自治区重大专项“区块链共性关键技术研究”(2020A02001-1);新疆维吾尔自治区天山青年计划(2019Q030);新疆维吾尔自治区天山雪松计划“智适应学习中学习行为分析与个性化学习推荐技术研究”(2020XS10)。

Abstract: Existing unknown intent detection models map utterances to the vector space and use the Local Outlier Factor(LOF) algorithm to define the feature points with low density as the unknown intent.However, known intent feature clusters trained by the cross-entropy loss are narrower and longer, which causes the overall spacing in density, and the dispersion in the clusters is not sufficiently uniform for detection.This study proposes an unknown intent detection model based on the autoencoder reconstruction error to solve the above problems.During the training stage, the model uses a joint loss function with label knowledge to train a known intent classifier, which forces the distribution of known intent features to minimize the intraclass distance and maximize the interclass distance.It then uses these features to train an autoencoder that only contains information regarding the known intent.During the testing stage, samples with significant reconstruction errors are regarded as unknown intentions using the automatic encoder, and the other samples are regarded as normal classifications of known intentions.Experiments on the SNIPS dataset show that the Macro F1 score of the proposed model increases by 16.93%, 1.14%, and 2.37% compared with the Semantic-Enhanced large-margin Gaussian mixture loss(SEG) model of the best performance in the baseline models when proportions of the known intents are 25%, 50%, and 75%, respectively.Moreover, the proposed model can detect more unknown samples.Furthermore, the proposed model exhibits improved performance on the ATIS dataset, in which the intent distribution is highly unbalanced.

Key words: intent identification, task-oriented dialog, unknown intent detection, loss function, autoencoder, reconstruction error

摘要: 现有未知意图检测模型通常将语句映射到向量空间,并使用局部异常因子算法定义密度较小的特征点为未知意图,但经交叉熵损失训练的已知意图特征簇更加狭长,簇内的整体间距、密度和分散情况不均匀,进而增加了检测难度。针对上述问题,提出一种基于自动编码器重建误差的未知意图检测模型。在训练阶段,使用融入标签知识的联合损失函数训练已知意图分类器,使已知意图特征类间距离大且类内距离小,并利用这些特征训练一个仅能获取已知意图信息的自动编码器。在测试阶段,利用自动编码器将重建误差较大的样本视为未知意图,其余样本视为已知意图正常分类。在SNIPS数据集上的实验结果表明,在已知意图占比为25%、50%、75%时,该模型的Macro F1得分相比于表现最优的增强语义的高斯混合损失基线模型分别提升了16.93%、1.14%和2.37%,能够检测到更多的未知意图样本,同时在类别分布极不平衡的ATIS数据集上也有较好的性能表现。

关键词: 意图识别, 任务型对话, 未知意图检测, 损失函数, 自动编码器, 重建误差

CLC Number: