4 个月前

通过约束深度自适应聚类及聚类优化发现新意图

通过约束深度自适应聚类及聚类优化发现新意图

摘要

识别新用户意图是对话系统中的一个重要任务。然而,由于意图的定义强烈依赖于先验知识,因此很难获得令人满意的聚类结果。现有的方法通过密集的特征工程来融入先验知识,这不仅容易导致过拟合,还使得聚类结果对聚类数量非常敏感。在本文中,我们提出了一种带有聚类优化的约束深度自适应聚类(CDAC+)方法,这是一种端到端的聚类方法,可以自然地将成对约束作为先验知识纳入聚类过程。此外,我们通过强制模型从高置信度分配中学习来优化聚类结果。在消除低置信度分配后,我们的方法对聚类数量表现出惊人的不敏感性。在三个基准数据集上的实验结果显示,我们的方法相比强大的基线模型取得了显著的改进。

代码仓库

thuiar/CDAC-plus
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
open-intent-discovery-on-atisCDAC+
ACC: 91.66
ARI: 89.41
NMI: 94.74
open-intent-discovery-on-snipsCDAC+
ACC: 93.63
ARI: 86.82
NMI: 89.3
open-intent-discovery-on-stackoverflowCDAC+
ACC: 73.48
ARI: 52.59
NMI: 69.84

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
通过约束深度自适应聚类及聚类优化发现新意图 | 论文 | HyperAI超神经