MEGA Hub

ACRL: Adaptive Control of Training-Inference Discrepancy for Stable Reinforcement Learning

Authors

Do you know Wenwu Fan?You can claim authorship or link another user.Do you know Qihong Lin?You can claim authorship or link another user.Do you know Zhijie Xia?You can claim authorship or link another user.Do you know Zhuo Zheng?You can claim authorship or link another user.Do you know Sihao Wang?You can claim authorship or link another user.Do you know Qiang Chen?You can claim authorship or link another user.Do you know Liangsheng Zhu?You can claim authorship or link another user.

Abstract

Reinforcement Learning (RL) training for Large Language Models (LLMs) often suffers from instability due to the discrepancy between training and inference. This training-inference discrepancy stems from two primary factors: an architectural separation between training and inference engines, and the use of low-precision quantization in inference versus higher-precision computation in training. To address training instability issues caused by high training-inference discrepancy, we present the principles and methods for its adaptive control. We propose Adaptive Control Reinforcement Learning (ACRL), which adaptively maintains the training-inference discrepancy within a reasonable range to ensure stable RL training. Beyond stabilization, ACRL inherently increases policy entropy, thereby enhancing exploration and improving accuracy. The experimental results show that when the inference engine utilizes FP8 quantization, ACRL consistently maintains the training-inference discrepancy within a reasonable range and stabilizes RL training. Furthermore, ACRL not only matches the accuracy of the BF16 baseline but also outperforms importance sampling (IS) fixes.

Community

00