RLHF: Reinforcement Learning from Human Feedback
RLHF — Reinforcement Learning from Human Feedback — is a revolutionary technique that lets AI truly understand human intent. It uses reinforcement learning combined with human feedback to train large language models so they better align with human values.
Core Goals
- Align AI models with human values
- Through learning from human preferences, make model outputs safer, more helpful, and harmless
- Helpful
- Harmless
- Honest
- Data: instruction datasets (instruction-response pairs)
- Method: supervised learning with cross-entropy loss
- Goal: teach the model to understand and execute basic instructions
- Output: an SFT model with instruction-following capability
- Data: human preference data (chosen vs rejected)
- Method: learning-to-rank, maximizing the score gap between chosen and rejected responses
- Goal: learn to assess text quality and understand human preferences
- Output: a reward model that can score text
- Algorithm: PPO (Proximal Policy Optimization)
- Models: 4 models working together (Actor, Ref, Reward, Critic)
- Goal: maximize the scores given by the reward model
- Output: the final model aligned with human preferences
- DPO (Direct Preference Optimization): learns human preferences directly, without training an RM or running RL.
- Efficient training algorithms: lightweight architectures, distributed training, model compression.
- Enhanced safety: multi-layer safety checks, adversarial training, red-team testing.
The 3H Principles
Analogy: The Growth of an AI Student
1. Pre-training: like a student learning fundamentals, mastering language understanding and generation. 2. SFT: teaching the student how to read and solve problems — acquiring instruction-following ability. 3. RLHF: a teacher grades homework, and the student keeps improving based on feedback.
The Three-Stage RLHF Training Pipeline
Stage 1: Supervised Fine-Tuning (SFT)
Training an obedient base modelStage 2: Reward Model (RM) Training
Training a judge that can scoreStage 3: PPO Reinforcement Learning
Teaching the model to earn high scoresHow PPO Training Works
Four models collaborate:
1. Actor Model — executes the policy and generates responses 2. Reference Model — provides a baseline 3. Reward Model — evaluates quality 4. Critic Model — estimates expected returns
Technical Advantages
1. Alignment with human values
Following the 3H principles ensures outputs conform to human ethics and avoids harmful content.2. Better user experience
Generates more natural replies that fit user needs, improving conversation quality.3. Data efficiency
A small amount of high-quality feedback yields large performance gains: the 1.3B-parameter InstructGPT outperforms the 175B-parameter GPT-3.Challenges
1. Inconsistent human feedback — annotators' subjective judgments differ. 2. Reward model bias — the RM may not fully reflect true human preferences, limited by training data. 3. Compute cost — four models trained simultaneously require massive GPU resources (~240GB of VRAM) and long training times.
Future Directions
Why It Matters
RLHF is considered the most critical breakthrough behind ChatGPT compared to GPT-3 — the key technology on the path from GPT-3 to ChatGPT.
--- Source: Easy AI learning platform | Tutorial created for AI knowledge popularization