GAC: Noise-Aware Adaptive Mixing for Hybrid SFT-RL Post-Training
cs.LG, cs.AI
Submitted: 2026-05-25
Updated: 2026-10-10
Comments: 17 pages, 3 figures, 22 tables. Camera-ready update adds links to the released Qwen3.5-4B compact-backbone checkpoint, reference implementation, and project page; experimental results and conclusions are unchanged
Code: https://github.com/anonymous/GAC
License: http://creativecommons.org/licenses/by/4.0/
The gist: Hybrid post-training combining supervised fine-tuning (SFT) and reinforcement learning (RL) is the standard paradigm for aligning large language models, yet fixed mixing schedules cannot adapt when
Terminology
Abstract
Hybrid post-training combining supervised fine-tuning (SFT) and reinforcement learning (RL) is the standard paradigm for aligning large language models, yet fixed mixing schedules cannot adapt when the relative noise of the two signals evolves. We derive a noise-aware mixing weight mu* by minimizing an MSE upper bound on the mixed stochastic gradient, yielding a closed-form that balances gradient noise variance and SFT-RL disagreement. Building on the token-wise reweighting of CHORD, the practical Guided Adaptive Controller (GAC) adds EMA smoothing, a schedule prior, and capped updates around this estimator, with all statistics estimated online from existing training tensors. The noise-aware controller alone outperforms the best rule-based controller by +3.0 percentage points on AMC; the full system reaches +3.8 points over HPT across math, code, science, and logic benchmarks, while reducing KL-drift area by 28% and large absolute mixing-weight changes by more than 70%, at less than 1% overhead. Gains grow with model size from 1.5B to 14B. Resources: Qwen3.5-4B compact-backbone release: https://huggingface.co/YueLinHu/GAC-Qwen3.5-4B; reference implementation: https://github.com/deepnovacore/GAC; GAC project page: https://deepnovacore.github.io/GAC/
Sources
- Program Synthesis with Large Language Models
- Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback
- Evaluating Large Language Models Trained on Code
- DeepSeek-R1: Incentivizing Reasoning Capability in LLMs via Reinforcement Learning
- GPQA: A Graduate-Level Google-Proof Q&A Benchmark
- High-Dimensional Continuous Control Using Generalized Advantage Estimation
- Proximal Policy Optimization Algorithms
- DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Models
- Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Them
- SciBench: Evaluating College-Level Scientific Problem-Solving Abilities of Large Language Models
- On-Policy RL Meets Off-Policy Experts: Harmonizing Supervised Fine-Tuning and Reinforcement Learning via Dynamic Weighting
- SRFT: A Single-Stage Method with Supervised and Reinforcement Fine-Tuning for Reasoning
- Learning to Reason under Off-Policy Guidance
- Towards a Unified View of Large Language Model Post-Training
- Trust-Region Adaptive Policy Optimization
- Anchored Supervised Fine-Tuning
- On the Non-decoupling of Supervised Fine-tuning and Reinforcement Learning in Post-training
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks