arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2305.14299cs.CLcs.AI

基于模板的对话句嵌入对比学习

Template-assisted Contrastive Learning of Task-oriented Dialogue Sentence Embeddings

Stanford University(斯坦福大学);Zhejiang University(浙江大学);Alibaba Qwen Pilot(阿里巴巴Qwen Pilot)

Minsik Oh, Jiwei Li, Guoyin Wang

AI总结:

本文提出TaDSE方法,利用模板信息通过自监督对比学习学习对话句嵌入,通过合成数据集增强效果,并在五个对话基准数据集上验证了其有效性。

摘要:

学习高质量对话句嵌入对于解决多种对话任务至关重要,但标注和获取对话中句子关系困难,而token级标注如实体、槽位和模板更容易获得。其他句嵌入方法通常是句子级自监督框架,无法利用token级额外知识。我们引入了Template-aware Dialogue Sentence Embedding (TaDSE),一种新的增强方法,通过自监督对比学习框架利用模板信息学习句子嵌入。我们进一步通过合成增强数据集增强效果,其中槽位填充是初步步骤。我们在五个下游对话基准数据集上评估了TaDSE的性能。实验结果表明,TaDSE在对话任务中相比先前SOTA方法有显著提升。我们还引入了一种新的语义压缩测试分析工具,发现其与统一性和对齐性相关。我们的代码可在https://github.com/minsik-ai/Template-Contrastive-Embedding获取。

英文摘要:

Learning high quality sentence embeddings from dialogues has drawn increasing attentions as it is essential to solve a variety of dialogue-oriented tasks with low annotation cost. Annotating and gathering utterance relationships in conversations are difficult, while token-level annotations, \eg, entities, slots and templates, are much easier to obtain. Other sentence embedding methods are usually sentence-level self-supervised frameworks and cannot utilize token-level extra knowledge. We introduce Template-aware Dialogue Sentence Embedding (TaDSE), a novel augmentation method that utilizes template information to learn utterance embeddings via self-supervised contrastive learning framework. We further enhance the effect with a synthetically augmented dataset that diversifies utterance-template association, in which slot-filling is a preliminary step. We evaluate TaDSE performance on five downstream benchmark dialogue datasets. The experiment results show that TaDSE achieves significant improvements over previous SOTA methods for dialogue. We further introduce a novel analytic instrument of semantic compression test, for which we discover a correlation with uniformity and alignment. Our code is available at https://github.com/minsik-ai/Template-Contrastive-Embedding

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

↑