Beyond Truncation: Rethinking LLM Decoding as Ensemble Pruning

Dunyao Xue, Chengshuo Du, Zhengbo Wang, Wenlin Dai, Cheng Meng


Abstract
We introduce Mahalanobis-Ensemble Decoding (ME-Decoding), a novel Large Language Model (LLM) decoding framework that frames candidate token selection as ensemble pruning. Existing selection strategies rely predominantly on scalar probabilities, ignoring geometric semantic relationships and causing candidate redundancy. Meanwhile, current geometry-aware methods often require complex optimization or directly reweighting the original token probabilities, leading to significant computational overhead or inference instability. To address this, we formulate decoding as a subset optimization problem using a Mahalanobis distance-driven objective to enhance semantic diversity while preserving high probabilities. Specifically, we dynamically discount redundant generation paths using a token similarity matrix, constructed via an adaptive-bandwidth kernel over token embeddings. We further devise an efficient greedy selection algorithm with near-linear complexity in the candidate size under early stopping, while establishing its theoretical approximation guarantees. This renders ME-Decoding a robust, plug-and-play module with negligible inference overhead. Extensive experiments across diverse reasoning and generation tasks demonstrate that our method consistently achieves strong performance.
Anthology ID:
2026.emnlp-main.743
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
17229–17253
Language:
URL:
https://aclanthology.org/2026.emnlp-main.743/
DOI:
Bibkey:
Cite (ACL):
Dunyao Xue, Chengshuo Du, Zhengbo Wang, Wenlin Dai, and Cheng Meng. 2026. Beyond Truncation: Rethinking LLM Decoding as Ensemble Pruning. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 17229–17253, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Beyond Truncation: Rethinking LLM Decoding as Ensemble Pruning (Xue et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.743.pdf