Paper Overview
- Research area: AI
- Authors: Arthur Renard, Franck Gabriel, Valentin Hartmann, et al.
- Published: 2026-05-28
- arXiv: 2605.27701
Abstract (translated)
This paper proposes Frost Training, an improved method for Monte Carlo-based policy optimization, targeting a large family of LLM-as-a-judge tasks called Cross-Entropy Games. The core idea is to exploit the gradient of the reward function in embedding space. This signal is already used in the GCG jailbreaking technique; this paper is the first to show that it can also be used to enhance model training. The method is validated using GRPO training for maximum-likelihood infilling. Frost Training improves the model's ability to generate high-scoring outputs, reaches higher maximum scores in a best-of-k setting, and does so at increased speed.
Original Abstract
We present Frost Training, a method for improving Monte Carlo-based policy optimization for a large family of LLM-as-a-judge tasks called Cross-Entropy Games. The key idea is to exploit the gradient of the reward function in embedding space. This signal is used in the Greedy Coordinate Gradient (GCG) jailbreaking technique; we demonstrate for the first time that it can also be used to boost model training. We validate our method using GRPO training for maximum-likelihood infilling. Frost Training improves the model's ability to generate high-scoring outputs, reaching higher maximum scores in a best-of-k setting, and does so at an increased speed.
*Automatically collected on 2026-05-29.*