DATABANK

Prover-Verifier Games improve legibility of LLM outputs

Jan Hendrik Kirchner, Yining Chen, Harri Edwards, Jan Leike, Nat McAleese, Yuri Burda

One way to increase confidence in the outputs of Large Language Models (LLMs) is to support them with reasoning that is clear and easy to check -- a property we call legibility. We study legibility in the context of solving grade-school math problems and show that optimizing chain-of-thought solutions only for answer correctness can make them less legible. To mitigate the loss in legibility, we propose a training algorithm inspired by Prover-Verifier Game from Anil et al. (2021). Our algorithm iteratively trains small verifiers to predict solution correctness, "helpful" provers to produce correct solutions that the verifier accepts, and "sneaky" provers to produce incorrect solutions that fool the verifier. We find that the helpful prover's accuracy and the verifier's robustness to adversarial attacks increase over the course of training. Furthermore, we show that legibility training transfers to time-constrained humans tasked with verifying solution correctness. Over course of LLM training human accuracy increases when checking the helpful prover's solutions, and decreases when checking the sneaky prover's solutions. Hence, training for checkability by small verifiers is a plausible technique for increasing output legibility. Our results suggest legibility training against small verifiers as a practical avenue for increasing legibility of large LLMs to humans, and thus could help with alignment of superhuman models.

Figure 1: Key result. Optimizing for solution correctness leads to high accuracy, but poor legibility. Initialization (pretrained on a variety of human-written math derivations) has higher legibility, but worse accuracy. Our proposed method retains high legibility with an accuracy in between the initialization and the correctness-only baseline. The resulting trade-off illustrates a legibility tax. From Jan Hendrik Kirchner et al. (2024), arXiv license.
Figure 2: Checkability training produces legible and well structured solutions. Three sample solutions produced by the helpful prover from different rounds of checkability training. From Jan Hendrik Kirchner et al. (2024), arXiv license.

Not packaged. A REPS package would add charts and verified data.How to package an experiment