arXiv:2505.17510cs.CL2025-05EMNLP被引 15

LLM做多标签分类时会抑制多数标签,新方法提升准确率和分布对齐。

Large Language Models Do Multi-Label Classification Differently

  • 分析每步生成的标签概率分布,发现初始分布与最终结果不符。
  • 模型规模越大,单标签置信度越高,标签排名更准确。
  • 通过分布对齐优化,无需额外计算即可提升分类性能。

多标签分类在真实场景中普遍,但大语言模型(LLM)在此任务上的行为研究不足。本文聚焦主观任务,分析自回归LLM在多标签分类中每步生成的输出分布。发现首个标签的初始概率分布往往不能反映最终输出,甚至相对顺序也不一致;且模型在每一步倾向于仅保留一个标签。随着模型规模增大,其标记分布熵降低、单标签置信度升高,内部标签相对排名改善。监督微调与强化学习进一步放大此现象。我们提出多标签场景下的分布对齐任务:将模型生成的标签分布与标注者响应的实证分布对齐。设计零样本与监督方法,均优于现有方案。其中一种方法——取所有生成步骤中最大概率标签而非仅初始分布——在不增加计算量的前提下,同时提升分布对齐与整体F1得分。

原文摘要 · Abstract (English)

Multi-label classification is prevalent in real-world settings, but the behavior of Large Language Models (LLMs) in this setting is understudied. We investigate how autoregressive LLMs perform multi-label classification, focusing on subjective tasks, by analyzing the output distributions of the models at each label generation step. We find that the initial probability distribution for the first label often does not reflect the eventual final output, even in terms of relative order and find LLMs tend to suppress all but one label at each generation step. We further observe that as model scale increases, their token distributions exhibit lower entropy and higher single-label confidence, but the internal relative ranking of the labels improves. Finetuning methods such as supervised finetuning and reinforcement learning amplify this phenomenon. We introduce the task of distribution alignment for multi-label settings: aligning LLM-derived label distributions with empirical distributions estimated from annotator responses in subjective tasks. We propose both zero-shot and supervised methods which improve both alignment and predictive performance over existing approaches. We find one method -- taking the max probability over all label generation distributions instead of just using the initial probability distribution -- improves both distribution alignment and overall F1 classification without adding any additional computation.

多标签分类大模型行为分布对齐语言模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。