arXiv:2602.22718cs.AIcs.DC2026-02被引 1

用无服务器架构提升强化学习人类反馈训练效率,降本提速。

RLHFless: Serverless Computing for Efficient RLHF

  • 基于无服务器计算动态适应资源需求,优化任务调度。
  • 实测比当前最优方案快1.35倍,成本降低44.8%。
  • 适合大规模语言模型对齐训练,尤其关注成本与速度的团队。

强化学习从人类反馈(RLHF)被广泛应用于大语言模型(LLM)后训练,以对齐模型输出与人类偏好。近期模型如DeepSeek-R1也证明了RLHF在复杂任务推理上的潜力。在强化学习中,推理与训练并存,导致整个流程资源需求动态变化。相比传统强化学习,RLHF因模型规模扩大和资源消耗增加,进一步挑战训练效率。现有框架虽试图平衡抽象灵活性与执行效率,但仍依赖有服务器基础设施,难以应对细粒度资源波动。同步训练中,各组件间或内部常出现空闲时间,造成开销与资源浪费。为此,我们提出首个基于无服务器环境的可扩展同步RLHF训练框架——RLHFless。该框架自适应整个RLHF流水线中的动态资源需求,预计算共享前缀以避免重复计算,并采用考虑响应长度变化的成本感知演员扩缩策略,找到成本更低、速度更快的平衡点。此外,通过高效分配任务,减少函数内负载不均与空闲时间。在真实测试平台及大规模模拟集群上的实验表明,相比当前最优基线,RLHFless实现最高1.35倍加速和44.8%成本降低。

原文摘要 · Abstract (English)

Reinforcement Learning from Human Feedback (RLHF) has been widely applied to Large Language Model (LLM) post-training to align model outputs with human preferences. Recent models, such as DeepSeek-R1, have also shown RLHF's potential to improve LLM reasoning on complex tasks. In RL, inference and training co-exist, creating dynamic resource demands throughout the workflow. Compared to traditional RL, RLHF further challenges training efficiency due to expanding model sizes and resource consumption. Several RLHF frameworks aim to balance flexible abstraction and efficient execution. However, they rely on serverful infrastructures, which struggle with fine-grained resource variability. As a result, during synchronous RLHF training, idle time between or within RL components often causes overhead and resource wastage. To address these issues, we present RLHFless, the first scalable training framework for synchronous RLHF, built on serverless computing environments. RLHFless adapts to dynamic resource demands throughout the RLHF pipeline, pre-computes shared prefixes to avoid repeated computation, and uses a cost-aware actor scaling strategy that accounts for response length variation to find sweet spots with lower cost and higher speed. In addition, RLHFless assigns workloads efficiently to reduce intra-function imbalance and idle time. Experiments on both physical testbeds and a large-scale simulated cluster show that RLHFless achieves up to 1.35x speedup and 44.8% cost reduction compared to the state-of-the-art baseline.

RLHF无服务器训练优化大模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。