HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Confidence Is All You Need: Few-Shot RL Fine-Tuning of Language Models

Li Pengyi Skripkin Matvey Zubrey Alexander Kuznetsov Andrey Oseledets Ivan

Confidence Is All You Need: Few-Shot RL Fine-Tuning of Language Models

Abstract

Large language models (LLMs) excel at reasoning, yet post-training remainscritical for aligning their behavior with task goals. Existing reinforcementlearning (RL) methods often depend on costly human annotations or externalreward models. We propose Reinforcement Learning via Self-Confidence (RLSC),which uses the model's own confidence as reward signals-eliminating the needfor labels, preference models, or reward engineering. Applied toQwen2.5-Math-7B with only 16 samples per question and 10 or 20 training steps,RLSC improves accuracy by +13.4% on AIME2024, +21.2% on MATH500, +21.7% onMinerva Math, +20.8% on Olympiadbench, and +9.7% on AMC23. RLSC provides asimple, scalable post-training method for inference models, requiring only asmall number of samples and unlabelled supervision.

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Confidence Is All You Need: Few-Shot RL Fine-Tuning of Language Models | Papers | HyperAI