arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.08987cs.CLcs.AI

使用深度对齐聚类发现新意图

Discovering New Intents with Deep Aligned Clustering

Tsinghua University(清华大学);Beijing University of Posts and Telecommunications(北京邮电大学)

Hanlei Zhang, Hua Xu, Ting-En Lin, Rui Lyu

AI总结:

本文提出 Deep Aligned Clustering,利用已知意图预训练、k-means 伪标签、标签对齐和低置信聚类剔除,在未知新意图数量时更鲁棒地发现意图并显著超越现有方法。

摘要:

发现新意图是对话系统中的一项关键任务。大多数现有方法在将先验知识从已知意图迁移到新意图方面存在局限。它们也难以提供高质量的监督信号,以学习适合聚类的特征来对未标记意图进行分组。在本工作中,我们提出了一种有效的方法——Deep Aligned Clustering(深度对齐聚类),借助有限的已知意图数据来发现新意图。首先,我们利用少量带标签的已知意图样本作为先验知识来预训练模型。随后,我们执行 k-means 以生成聚类分配作为伪标签。此外,我们提出了一种对齐策略,以解决聚类分配过程中的标签不一致问题。最后,我们在对齐后的伪标签监督下学习意图表示。在新意图数量未知的情况下,我们通过消除低置信度的意图级聚类来预测意图类别的数量。在两个基准数据集上进行的大量实验表明,我们的方法更加鲁棒,并且相较于最先进方法取得了显著提升。代码已发布于 https://github.com/thuiar/DeepAligned-Clustering。

英文摘要:

Discovering new intents is a crucial task in dialogue systems. Most existing methods are limited in transferring the prior knowledge from known intents to new intents. They also have difficulties in providing high-quality supervised signals to learn clustering-friendly features for grouping unlabeled intents. In this work, we propose an effective method, Deep Aligned Clustering, to discover new intents with the aid of the limited known intent data. Firstly, we leverage a few labeled known intent samples as prior knowledge to pre-train the model. Then, we perform k-means to produce cluster assignments as pseudo-labels. Moreover, we propose an alignment strategy to tackle the label inconsistency problem during clustering assignments. Finally, we learn the intent representations under the supervision of the aligned pseudo-labels. With an unknown number of new intents, we predict the number of intent categories by eliminating low-confidence intent-wise clusters. Extensive experiments on two benchmark datasets show that our method is more robust and achieves substantial improvements over the state-of-the-art methods. The codes are released at https://github.com/thuiar/DeepAligned-Clustering.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑