RPRA: Predicting an LLM-Judge for Efficient but Performant Inference

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Ashley, Dylan R., Lan, Gaël Le, Zhao, Changsheng, Dhingra, Naina, Cai, Zhipeng, Chang, Ernie, Zhuge, Mingchen, Shi, Yangyang, Chandra, Vikas, Schmidhuber, Jürgen
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866917407083200512
author Ashley, Dylan R.
Lan, Gaël Le
Zhao, Changsheng
Dhingra, Naina
Cai, Zhipeng
Chang, Ernie
Zhuge, Mingchen
Shi, Yangyang
Chandra, Vikas
Schmidhuber, Jürgen
author_facet Ashley, Dylan R.
Lan, Gaël Le
Zhao, Changsheng
Dhingra, Naina
Cai, Zhipeng
Chang, Ernie
Zhuge, Mingchen
Shi, Yangyang
Chandra, Vikas
Schmidhuber, Jürgen
contents Large language models (LLMs) face a fundamental trade-off between computational efficiency (e.g., number of parameters) and output quality, especially when deployed on computationally limited devices such as phones or laptops. One way to address this challenge is by following the example of humans and have models ask for help when they believe they are incapable of solving a problem on their own; we can overcome this trade-off by allowing smaller models to respond to queries when they believe they can provide good responses, and deferring to larger models when they do not believe they can. To this end, in this paper, we investigate the viability of Predict-Answer/Act (PA) and Reason-Predict-Reason-Answer/Act (RPRA) paradigms where models predict -- prior to responding -- how an LLM judge would score their output. We evaluate three approaches: zero-shot prediction, prediction using an in-context report card, and supervised fine-tuning. Our results show that larger models (particularly reasoning models) perform well when predicting generic LLM judges zero-shot, while smaller models can reliably predict such judges well after being fine-tuned or provided with an in-context report card. Altogether, both approaches can substantially improve the prediction accuracy of smaller models, with report cards and fine-tuning achieving mean improvements of up to 55% and 52% across datasets, respectively. These findings suggest that models can learn to predict their own performance limitations, paving the way for more efficient and self-aware AI systems.
format Preprint
id arxiv_https___arxiv_org_abs_2604_12634
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle RPRA: Predicting an LLM-Judge for Efficient but Performant Inference
Ashley, Dylan R.
Lan, Gaël Le
Zhao, Changsheng
Dhingra, Naina
Cai, Zhipeng
Chang, Ernie
Zhuge, Mingchen
Shi, Yangyang
Chandra, Vikas
Schmidhuber, Jürgen
Artificial Intelligence
Computation and Language
Machine Learning
Multiagent Systems
68T07
I.2.6; I.2.7; I.2.11
Large language models (LLMs) face a fundamental trade-off between computational efficiency (e.g., number of parameters) and output quality, especially when deployed on computationally limited devices such as phones or laptops. One way to address this challenge is by following the example of humans and have models ask for help when they believe they are incapable of solving a problem on their own; we can overcome this trade-off by allowing smaller models to respond to queries when they believe they can provide good responses, and deferring to larger models when they do not believe they can. To this end, in this paper, we investigate the viability of Predict-Answer/Act (PA) and Reason-Predict-Reason-Answer/Act (RPRA) paradigms where models predict -- prior to responding -- how an LLM judge would score their output. We evaluate three approaches: zero-shot prediction, prediction using an in-context report card, and supervised fine-tuning. Our results show that larger models (particularly reasoning models) perform well when predicting generic LLM judges zero-shot, while smaller models can reliably predict such judges well after being fine-tuned or provided with an in-context report card. Altogether, both approaches can substantially improve the prediction accuracy of smaller models, with report cards and fine-tuning achieving mean improvements of up to 55% and 52% across datasets, respectively. These findings suggest that models can learn to predict their own performance limitations, paving the way for more efficient and self-aware AI systems.
title RPRA: Predicting an LLM-Judge for Efficient but Performant Inference
topic Artificial Intelligence
Computation and Language
Machine Learning
Multiagent Systems
68T07
I.2.6; I.2.7; I.2.11
url https://arxiv.org/abs/2604.12634