Mitigating Selection Bias with Node Pruning and Auxiliary Options
Abstract
Large language models (LLMs) often exhibit systematic preferences for certain answer choices when responding to multiple-choice questions-a behavior known as selection bias. This bias reduces the accuracy and reliability of LLM outputs, limiting their usefulness in decision-critical applications. While prior work has focused on adjusting model inputs or outputs to mitigate this issue, our work takes a fundamentally different approach by identifying and removing the internal sources of bias. We introduce two methods: Bias Node Pruning (BNP), which prunes parameters that contribute to selection bias, and Auxiliary Option Injection (AOI), which introduces an additional answer choice to reduce bias in both white-box and black-box settings. To address the shortcomings of existing evaluation metrics, we propose Choice Kullback-Leibler Divergence (CKLD), a new metric that captures distributional imbalances in model predictions. Experiments on three LLMs across multiple datasets demonstrate that our methods consistently improve answer accuracy while reducing selection bias, providing a robust solution for both open- and closed-source models.
Cite
@article{arxiv.2409.18857,
title = {Mitigating Selection Bias with Node Pruning and Auxiliary Options},
author = {Hyeong Kyu Choi and Weijie Xu and Chi Xue and Stephanie Eckman and Chandan K. Reddy},
journal= {arXiv preprint arXiv:2409.18857},
year = {2025}
}
Comments
ACL 2025 Main