通过激活分布峰度优化标签集,提升生成模型零样本分类效果
Label Set Optimization via Activation Distribution Kurtosis for Zero-shot Classification with Generative Models
- 基于神经元激活分布峰度筛选最优标签组合
- 在多个任务和语言上实现0.54~0.76的性能提升
- 无需标注数据与反向传播,单次前向即可完成
上下文学习(ICL)性能高度依赖提示设计,但零样本分类中类别标签选项(如词汇或顺序)的影响尚未深入研究。本文提出LOADS(Label set Optimization via Activation Distribution Kurtosis),一种针对大语言模型零样本ICL的后处理标签集优化方法。该方法基于首次系统性分析标签设计(词汇选择、顺序、扩展)对分类性能的影响,发现标签词汇选择显著影响模型表现及其对标签顺序的敏感性。进一步研究表明,表现更优的标签词会激活更少的异常神经元。LOADS利用峰度衡量神经元激活分布,仅需一次前向传播即可完成标签选择,无需梯度计算或标注数据。实验表明,LOADS选取的标签在多种分类任务、数据集、模型和语言上均表现稳定,相较传统使用原始标签词的方法,性能提升最高达0.76。
原文摘要 · Abstract (English)
In-context learning (ICL) performance is highly sensitive to prompt design, yet the impact of class label options (e.g. lexicon or order) in zero-shot classification remains underexplored. This study proposes LOADS (Label set Optimization via Activation Distribution kurtosiS), a post-hoc method for selecting optimal label sets in zero-shot ICL with large language models (LLMs). LOADS is built upon the observations in our empirical analysis, the first to systematically examine how label option design (i.e., lexical choice, order, and elaboration) impacts classification performance. This analysis shows that the lexical choice of the labels in the prompt (such as agree vs. support in stance classification) plays an important role in both model performance and model's sensitivity to the label order. A further investigation demonstrates that optimal label words tend to activate fewer outlier neurons in LLMs' feed-forward networks. LOADS then leverages kurtosis to measure the neuron activation distribution for label selection, requiring only a single forward pass without gradient propagation or labelled data. The LOADS-selected label words consistently demonstrate effectiveness for zero-shot ICL across classification tasks, datasets, models and languages, achieving maximum performance gain from 0.54 to 0.76 compared to the conventional approach of using original dataset label words.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。