mlsanityData to 7 Oct 2026

Mitigating Selection Bias with Node Pruning and Auxiliary Options

arXiv 2409.18857 ↗First posted 27 Sep 2024cs.AIcited by 12

Not on the map. Why?

Abstract (arXiv)

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.

Key facts. Cited by 12 papers in the map (data through 7 Oct 2026). Builds on 36 of its 45 references that are in the map. First posted on arXiv 27 Sep 2024.

Builds on

36 of 45 references.

References · 36

  1. 2020Language Models are Few-Shot Learnerscited by 32,199
  2. 2022Chain-of-Thought Prompting Elicits Reasoning in Large Language Modelscited by 12,302
  3. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  4. 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
  5. 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
  6. 2023Mistral 7Bcited by 3,065
  7. 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,291
  8. 2018CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledgecited by 1,793
  9. 2016ConceptNet 5.5: An Open Multilingual Graph of General Knowledgecited by 1,065
  10. 2023Large Language Models are not Fair Evaluatorscited by 1,050
  11. 2020RobustBench: a standardized adversarial robustness benchmarkcited by 659
  12. 2017To prune, or not to prune: exploring the efficacy of pruning for model compressioncited by 596
24 more
  1. 2019Importance Estimation for Neural Network Pruningcited by 531
  2. 2023Large Language Models Are Not Robust Multiple Choice Selectorscited by 454
  3. 2023DoLa: Decoding by Contrasting Layers Improves Factuality in Large Language Modelscited by 367
  4. 2023Large Language Models Sensitivity to The Order of Options in Multiple-Choice Questionscited by 249
  5. 2023Model Sparsity Can Simplify Machine Unlearningcited by 206
  6. 2022Leveraging Large Language Models for Multiple Choice Question Answeringcited by 199
  7. 2024Are We Done with MMLU?cited by 178
  8. 2023Do LLMs exhibit human-like response biases? A case study in survey designcited by 176
  9. 2023AnnoLLM: Making Large Language Models to Be Better Crowdsourced Annotatorscited by 171
  10. 2015Data-free parameter pruning for Deep Neural Networkscited by 112
  11. 2023CoAnnotating: Uncertainty-Guided Work Allocation between Human and Large Language Models for Data Annotationcited by 93
  12. 2023Split and Merge: Aligning Position Biases in LLM-based Evaluatorscited by 71
  13. 2024Artifacts or Abduction: How Do LLMs Answer Multiple-Choice Questions Without the Question?cited by 68
  14. 2024"My Answer is C": First-Token Probabilities Do Not Match Text Answers in Instruction-Tuned Language Modelscited by 66
  15. 2024Changing Answer Order Can Decrease MMLU Accuracycited by 41
  16. 2024Dissecting Language Models: Machine Unlearning via Selective Pruningcited by 38
  17. 2024Beyond Performance: Quantifying and Mitigating Label Bias in LLMscited by 24
  18. 2024Prompt-prompted Adaptive Structured Pruning for Efficient LLM Generationcited by 20
  19. 2024Unveiling Selection Biases: Exploring Order and Token Sensitivity in Large Language Modelscited by 20
  20. 2024AI Can Be Cognitively Biased: An Exploratory Study on Threshold Priming in LLM-Based Batch Relevance Assessmentcited by 20
  21. 2024Anchored Answers: Unravelling Positional Bias in GPT-2's Multiple-Choice Questionscited by 15
  22. 2024Position: Insights from Survey Methodology can Improve Training Datacited by 13
  23. 2024Strengthened Symbol Binding Makes Large Language Models Reliable Multiple-Choice Selectorscited by 13
  24. 2024Teacher-Student Training for Debiasing: General Permutation Debiasing for Large Language Modelscited by 8

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2025: 10; 2026: 2.

12 papers.