TransparencyTrustworthinessSafetycs.CL
Improving LLM First-Token Predictions in Multiple-Choice Question Answering via Output Prefilling
Silvia Cappelletti, Tobia Poppi, Samuele Poppi, Zheng-Xin Yong, Diego Garcia-Olano, Marcella Cornia, Lorenzo Baraldi, Rita Cucchiara
May 21, 2025
International Conference on Pattern Recognition
8 citations
Key Takeaways
-
Current evaluation methods for AI systems can be misleading: The paper demonstrates that when LLMs answer multiple-choice questions, they often assign high probability to irrelevant tokens or use tokens as generic filler rather than genuine answers, meaning standard evaluation metrics may overestimate or misrepresent actual model performance.
-
Simple prompting techniques can significantly improve AI reliability: Adding a structured prefix like "The correct option is:" before the model's response substantially improved accuracy and consistency across multiple AI systems and benchmarks, suggesting that careful prompt design is a practical way to make AI systems more trustworthy without expensive retraining.
-
Efficiency and reliability don't have to be trade-offs: The prefilling approach achieved performance comparable to more computationally expensive evaluation methods while being "significantly more efficient," indicating that responsible AI evaluation doesn't necessarily require choosing between speed and accuracy.
-
Evaluation methods themselves shape how we understand AI capabilities: The paper reveals that how we measure AI performance (first-token probability vs. other approaches) can substantially affect our conclusions about what the model actually knows or can do, highlighting the importance of rigorous evaluation methodology in responsible AI development.
-
Low-cost interventions can address AI safety concerns: Originally developed for AI safety purposes, the prefilling technique was repurposed to improve evaluation reliability, demonstrating that solutions to AI robustness problems don't always require complex technical fixes and can be adapted across different applications.