Logit Separability-Driven Samples and Multiple Class-Related Words Selection for Advancing In-Context Learning

Zixiao Zhu, Zijian Feng, Hanzhang Zhou, Junlang Qian, Kezhi Mao


Abstract
Effective organization of in-context learning (ICL) demonstrations is key to improving the quality of large language model (LLM) responses. To create better sample-label pairs that instruct LLM understanding, we introduce logit separability, a criterion to assess the clarity of both samples and class-related words at the logit level. This facilitates the optimization of sample and label selection, enhancing the precision of information provided in ICL demonstrations. Additionally, we find that incorporating multiple class-related words for each sample, rather than relying on a single class name, improves performance by offering a broader range of label information. Building on these insights, we propose LICL, a logit separability-based method that jointly organizes samples and integrates multiple class-related words into each sample-label pair. Evaluations across seven classification datasets show that this approach significantly improves ICL performance by providing clearer instructions and richer label information.
Anthology ID:
2025.naacl-long.343
Volume:
Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers)
Month:
April
Year:
2025
Address:
Albuquerque, New Mexico
Editors:
Luis Chiruzzo, Alan Ritter, Lu Wang
Venue:
NAACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
6739–6759
Language:
URL:
https://preview.aclanthology.org/fix-sig-urls/2025.naacl-long.343/
DOI:
Bibkey:
Cite (ACL):
Zixiao Zhu, Zijian Feng, Hanzhang Zhou, Junlang Qian, and Kezhi Mao. 2025. Logit Separability-Driven Samples and Multiple Class-Related Words Selection for Advancing In-Context Learning. In Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers), pages 6739–6759, Albuquerque, New Mexico. Association for Computational Linguistics.
Cite (Informal):
Logit Separability-Driven Samples and Multiple Class-Related Words Selection for Advancing In-Context Learning (Zhu et al., NAACL 2025)
Copy Citation:
PDF:
https://preview.aclanthology.org/fix-sig-urls/2025.naacl-long.343.pdf