The Silent Vote: Improving Zero-Shot LLM Reliability by Aggregating Semantic Neighborhoods

· Source: Paper Index on ACL Anthology · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Data Science & Analytics · Depth: Expert, quick

Summary

A new inference-time layer, Semantic Softmax, significantly improves zero-shot Large Language Model reliability by addressing Renormalization Bias and the "Silent Vote" phenomenon. This bias occurs when standard constrained decoding in LLMs discards probability mass from semantic synonyms, leading to artificial overconfidence and poor calibration. Semantic Softmax recovers this lost information by aggregating scores from the semantic neighborhood of each target label. Evaluated on Qwen-3 and Phi-4-mini models using the GoEmotions and Civil Comments datasets, the approach consistently reduces Expected Calibration Error (ECE) and Brier Score. Simultaneously, it enhances discriminative performance, measured by AUROC and Macro-F1, offering a more calibrated and accurate alternative for zero-shot classification by accounting for linguistic nuances.

Key takeaway

For Machine Learning Engineers deploying zero-shot LLM classifiers, consider integrating Semantic Softmax to significantly enhance model reliability and calibration. Your models, especially Qwen-3 and Phi-4-mini, will exhibit reduced Expected Calibration Error and Brier Score, alongside improved AUROC and Macro-F1. This method directly addresses Renormalization Bias, ensuring your zero-shot classifications are more accurate and trustworthy by accounting for linguistic nuances.

Key insights

Semantic Softmax improves zero-shot LLM classification by aggregating semantic synonym scores, mitigating Renormalization Bias and enhancing calibration.

Principles

Method

Semantic Softmax is an inference-time layer that recovers lost probability mass by aggregating scores of semantic neighborhoods surrounding each target label in constrained decoding.

In practice

Topics

Best for: AI Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, NLP Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Paper Index on ACL Anthology.