Label Set Optimization via Activation Distribution Kurtosis for Zero-Shot Classification with Generative Models

Yue Li, Zhixue Zhao, Carolina Scarton


Abstract
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.
Anthology ID:
2025.emnlp-main.1617
Volume:
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Month:
November
Year:
2025
Address:
Suzhou, China
Editors:
Christos Christodoulopoulos, Tanmoy Chakraborty, Carolyn Rose, Violet Peng
Venue:
EMNLP
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
31712–31729
Language:
URL:
https://preview.aclanthology.org/ingest-emnlp/2025.emnlp-main.1617/
DOI:
Bibkey:
Cite (ACL):
Yue Li, Zhixue Zhao, and Carolina Scarton. 2025. Label Set Optimization via Activation Distribution Kurtosis for Zero-Shot Classification with Generative Models. In Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing, pages 31712–31729, Suzhou, China. Association for Computational Linguistics.
Cite (Informal):
Label Set Optimization via Activation Distribution Kurtosis for Zero-Shot Classification with Generative Models (Li et al., EMNLP 2025)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-emnlp/2025.emnlp-main.1617.pdf
Checklist:
 2025.emnlp-main.1617.checklist.pdf