Compute as Teacher: Turning Inference Compute Into Reference-Free Supervision
Dulhan Jayalath, Shashwat Goel, Thomas Foster, Parag Jain, Suchin Gururangan, Cheng Zhang, Anirudh Goyal, Alan Schelten
TL;DR
This work introduces Compute as Teacher (CaT), a framework that enables RL in settings lacking ground-truth references by turning inference compute into supervision. It combines reference estimation (e.g., synthesis) to produce a pseudo-reference and reward derivation (verifiable or rubric-based) to guide training, allowing parallel rollouts to inform learning without human labels. The approach is validated on HealthBench (non-verifiable) and MATH-500 (verifiable), showing CaT can match or exceed inference-time aggregation while reducing test-time compute by up to 9×, and achieving up to 30% relative improvement over the initial policy in non-verifiable domains. A key innovation is self-proposed rubrics for non-verifiable tasks, which, scored by a judge, provide stable, auditable rewards without human annotations. Across three model families, CaT demonstrates strong scalability and versatility, acting as a drop-in mechanism that extends RL learning to domains where programmatic checkers are unavailable or non-existent.
Abstract
Where do learning signals come from when there is no ground truth in post-training? We show that inference compute itself can serve as supervision. By generating parallel rollouts and converting them into reference estimates, models can learn without human labels-critically, even in non-verifiable domains like healthcare guidance where no programmatic checker exists. We call this framework Compute as Teacher (CaT) and it turns inference-time compute from parallel rollouts into supervision for RL training. The framework has two components: (1) reference estimation which aggregates rollouts into a pseudo-reference answer, and (2) reward derivation which converts that pseudo-reference into RL rewards. For (1), we explore a simple method we call synthesis, but the framework admits any aggregator. For (2), we introduce self-proposed rubrics for non-verifiable domains. These are binary, auditable criteria generated from the pseudo-reference and scored by an LLM judge. On HealthBench, models trained with CaT match or exceed inference-time aggregation quality while using 9x less test-time compute. Here, CaT also competes with learning from expert physician annotations, yielding up to +30% relative improvement over the initial policy. The framework extends naturally to verifiable rewards, matching the best existing baselines on MATH-500 in test-time RL and demonstrating 'drop-in' versatility across both types of domains.
