Boosting Reinforcement Learning with Verifiable Rewards via Randomly Selected Few-Shot Guidance
For practitioners training LLMs with RLVR, FEST reduces the need for large SFT datasets, making it more cost-effective.
FEST improves sample efficiency of RLVR for LLMs by using only 128 randomly selected demonstrations, outperforming baselines with magnitudes less SFT data and matching full-dataset performance on math and coding benchmarks.
Reinforcement Learning with Verifiable Rewards (RLVR) has achieved great success in developing Large Language Models (LLMs) with chain-of-thought rollouts for many tasks such as math and coding. Nevertheless, RLVR struggles with sample efficiency on difficult problems where correct rollouts are hard to generate. Prior works propose to address this issue via demonstration-guided RLVR, i.e., to conduct Supervised FineTuning (SFT) when RL fails; however, SFT often requires a lot of data, which can be expensive to acquire. In this paper, we propose FEST, a FEw-ShoT demonstration-guided RLVR algorithm. It attains compelling results with only 128 demonstrations randomly selected from an SFT dataset. We find that three components are vital for the success: supervised signal, on-policy signal, and decaying weights on the few-shot SFT dataset to prevent overfitting from multiple-epoch training. On several benchmarks, FEST outperforms baselines with magnitudes less SFT data, even matching their performance with full dataset.