2024-06-29 17:09 湖北
大型语言模型(LLM)在大型多类分类任务中表现出色,但仍然存在分类错误,甚至生成了词汇表外的类别标签。
为了解决这些问题,提出了一种称为“Paraphrase and AGgregate (PAG)-LLM”的方法:
LLM生成输入查询的多个释义(平行查询)
对原始查询和每个释义执行多类分类
根据置信度分数聚合所有分类标签
该方法特别适用于LLM不确定的困难示例,可以减少关键的误分类和幻觉标签生成错误。
左侧的图展示了PAG-LLM的流程。在Figure 1(A)中,LLM对原始查询进行分类,只有当分类置信度低于阈值τ时,原始查询才会被提供给LLM以生成释义,这些释义随后再次被提供给 LLM进行分类。最后,LLM聚合来自释义和原始查询的预测类别标签。
展示了来自CLINC的示例,其中LLM分类了错误的标签(顶部示例)和词汇表外(OOV)类别标签(底部示例)。在顶部示例中,由PAG-LLM生成的释义使得可以做出正确的分类决策,并具有高置信度分数。因此,即使是简单的多数投票聚合也会导致正确的类别预测。在底部示例中,只有PAG-LLM生成的释义2能够正确分类,而其余的释义和原始查询则有OOV类别标签。PAG-LLM聚合了输入文本、释义、它们的标签和置信度,最终预测正确的类别标签。
两个大型多类分类数据集上评估了PAG-LLM:CLINC和Banking,并分别展示了22.7%和15.1%的错误减少。
在领域内(ID)CLINC 和 Banking(50%) 数据集上的性能。也在自一致性运行中使用了相同的微调(SFT)LLaMa-7B(第2行和第3行)。"Rand-seed" 表示使用不同随机种子训练的 6 个不同的 SFT LLaMa 的集成。"Vote" 表示采用多数投票策略来选择最终标签。对于第 6 行和第 8 行,在开发数据上调整了置信度阈值(对于 CLINC 和 Banking 分别表示为 τ=0.98 和 0.90),并且只有分类置信度较低的查询被提供给 PAG-LLM。最佳数字以粗体显示。P1、P2、P3、P4 代表来自先前工作的 4 个上下文学习(ICL)基线。
PAG-LLM在减少词汇表外(OOV)类别标签生成错误方面特别有效。
在 CLINC 和 Banking(50%) 的整体测试数据集(包括领域内 ID 和领域外 OOD 输入)上的性能。符号与上表中的相同。"All F1" 表示所有 ID + 1 个 OOD 类别的宏观 F1 分数(即,对于 CLINC 是 151 个类别,对于银行是 38 个类别)。"Avg" 表示 ID 和 OOD F1 分数的平均值。
https://arxiv.org/pdf/2406.17163
Paraphrase and Aggregate with Large Language Models for Minimizing Intent Classification Errors
推荐阅读
• 对齐LLM偏好的直接偏好优化方法:DPO、IPO、KTO
• RAG全景图:从RAG启蒙到高级RAG之36技,再到终章Agentic RAG!
• Agent到多模态Agent再到多模态Multi-Agents系统的发展与案例讲解(1.2万字,20+文献,27张图)
欢迎关注我的公众号“PaperAgent”,每天一篇大模型(LLM)文章来锻炼我们的思维,简单的例子,不简单的方法,提升自己。