arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2210.16637cs.CLcs.LG
收藏

超越聚类:通过聚类表示使预训练的语言模型成为更好的Zero-shot学习者

Beyond Prompting: Making Pre-trained Language Models Better Zero-shot Learners by Clustering Representations

ETH Zurich(苏黎世联邦理工学院);Peking University(北京大学)

Yu Fei, Ping Nie, Zhao Meng, Roger Wattenhofer, Mrinmaya Sachan

摘要:

最近的研究表明,预训练语言模型(PLM)是zero-shot学习器。然而,大多数现有的zero-shot方法涉及繁重的人体工程学或复杂的自我训练管道,阻碍了它们在新情况下的应用。在这项工作中,我们表明,zero-shot文本分类可以提高简单的聚类文本中的嵌入空间的PLM。具体来说,我们适合未标记的文本与贝叶斯高斯混合模型初始化后,使用类名称的聚类位置和形状。尽管它很简单,但这种方法在主题和情感分类数据集上都取得了优异或相当的性能,并且在不平衡数据集上显著优于先前的工作。我们进一步探索我们的聚类方法的适用性,通过评估它在14个数据集更多样化的主题,文本长度和类的数量。我们的方法实现了平均20%的绝对改善基于神经网络的零样本学习。最后,我们比较了不同的PLM嵌入空间,发现即使PLM没有显式地预先训练以生成有意义的句子嵌入,文本也能很好地按主题聚类。这项工作表明,PLM嵌入可以分类文本没有特定的任务微调,从而提供了一种新的方式来分析和利用他们的知识和零样本学习能力。

英文摘要:

Recent work has demonstrated that pre-trained language models (PLMs) are zero-shot learners. However, most existing zero-shot methods involve heavy human engineering or complicated self-training pipelines, hindering their application to new situations. In this work, we show that zero-shot text classification can be improved simply by clustering texts in the embedding spaces of PLMs. Specifically, we fit the unlabeled texts with a Bayesian Gaussian Mixture Model after initializing cluster positions and shapes using class names. Despite its simplicity, this approach achieves superior or comparable performance on both topic and sentiment classification datasets and outperforms prior works significantly on unbalanced datasets. We further explore the applicability of our clustering approach by evaluating it on 14 datasets with more diverse topics, text lengths, and numbers of classes. Our approach achieves an average of 20% absolute improvement over prompt-based zero-shot learning. Finally, we compare different PLM embedding spaces and find that texts are well-clustered by topics even if the PLM is not explicitly pre-trained to generate meaningful sentence embeddings. This work indicates that PLM embeddings can categorize texts without task-specific fine-tuning, thus providing a new way to analyze and utilize their knowledge and zero-shot learning ability.

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

↑