arXivDaily arXiv每日学术速递 周一至周五更新
​通知:联邦节假日调休:哥伦布日 / 原住民日(Columbus Day / Indigenous Peoples' Day),今日(13日)无更新。
arXiv 2604.07853cs.LGcs.AI

QaRL:基于回放对齐的量化感知强化学习用于在训练-推理不匹配下的快速稳定训练

QaRL: Rollout-Aligned Quantization-Aware RL for Fast and Stable Training under Training--Inference Mismatch

The Hong Kong University of Science and Technology(香港科技大学);City University of Hong Kong(香港城市大学);Zhejiang University(浙江大学)

Hao Gu, Hao Wang, Jiacheng Liu, Lujun Li, Qiyuan Zhu, Bei Liu, Binxing Xu, Lei Wang, Xintong Yang, Sida Lin, Sirui Han, Yike Guo

AI总结:

本文提出QaRL,通过对齐训练和回放的量化过程,解决训练-推理不匹配问题,提升训练效率和稳定性,同时保持低比特吞吐量优势。

摘要:

大型语言模型(LLM)的强化学习(RL)流水线常因回放生成而成为瓶颈,导致端到端训练缓慢。近期工作通过在回放中运行量化来加速解码,这是RL循环中最昂贵的阶段。然而,这些设置通过放大训练-推理差距而破坏了优化:回放在低精度下运行,而学习更新在全精度下计算。为解决这一挑战,我们提出了QaRL(回放对齐量化感知强化学习),通过对齐训练端的前向计算与量化回放以最小化不匹配。我们进一步识别出量化回放中的失败模式:长形式响应倾向于产生重复、乱码的token(错误token)。为缓解这些问题,我们引入TBPO(信任带策略优化),一种序列级目标,具有双重剪裁用于负样本,旨在保持更新在信任区域内。在Qwen3-30B-A3B MoE用于数学问题上,QaRL在量化回放训练上表现更优,提升5.5%,同时提高稳定性和保留低比特吞吐量优势。

英文摘要:

Large language model (LLM) reinforcement learning (RL) pipelines are often bottlenecked by rollout generation, making end-to-end training slow. Recent work mitigates this by running rollouts with quantization to accelerate decoding, which is the most expensive stage of the RL loop. However, these setups destabilize optimization by amplifying the training-inference gap: rollouts are operated at low precision, while learning updates are computed at full precision. To address this challenge, we propose QaRL (Rollout Alignment Quantization-Aware RL), which aligns training-side forward with the quantized rollout to minimize mismatch. We further identify a failure mode in quantized rollouts: long-form responses tend to produce repetitive, garbled tokens (error tokens). To mitigate these problems, we introduce TBPO (Trust-Band Policy Optimization), a sequence-level objective with dual clipping for negative samples, aimed at keeping updates within the trust region. On Qwen3-30B-A3B MoE for math problems, QaRL outperforms quantized-rollout training by +5.5 while improving stability and preserving low-bit throughput benefits.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑