跳到主要导航 跳到搜索 跳到主要内容

Scale Down to Speed Up: Dynamic Data Selection for Reinforcement Learning

  • Zhuoyue Chen*
  • , Jihai Zhang
  • , Ben Liu
  • , Fangquan Lin
  • , Wotao Yin*
  • *此作品的通讯作者
  • Zhejiang University
  • Alibaba Group Holding Ltd.
  • Wuhan University

科研成果: 书/报告/会议事项章节 › 会议稿件 › 同行评审

摘要

Optimizing data utilization remains a central challenge in applying Reinforcement Learning (RL) to Large Language Models (LLMs), directly impacting sample efficiency, training stability, and final model performance. Current approaches often rely on massive static datasets, leading to computational inefficiency and redundant gradient updates. In this paper, we propose ScalingRL, a data-centric RL framework that dynamically selects the most informative training samples to optimize RL for mathematical reasoning. Specifically, ScalingRL introduces the Data Effectiveness Score (DES) that quantitatively ranks prompts according to three complementary factors: problem difficulty, Chain-of-Thought complexity, and reward adaptability. Then, ScalingRL employs an adaptive curriculum scheduler that progressively adjusts the overall scale and specific mix of training prompts—balancing exploration of new, challenging data with exploitation of previously learned concepts—thereby tailoring the data distribution to the model’s current learning trajectory and performance. Experimental results demonstrate that ScalingRL achieves comparable performance to full-data training methods while requiring only 1.5K samples instead of 220K, reducing training time from 13 days to just 4 hours on 8×A800 GPUs.

源语言英语
主期刊名EMNLP 2025 - 2025 Conference on Empirical Methods in Natural Language Processing, Findings of EMNLP 2025
编辑Christos Christodoulopoulos, Tanmoy Chakraborty, Carolyn Rose, Violet Peng
出版商Association for Computational Linguistics (ACL)
页7806-7817
页数12
ISBN(电子版)9798891763357
DOI
出版状态已出版 - 2025
已对外发布是
活动30th Conference on Empirical Methods in Natural Language Processing, EMNLP 2025 - Suzhou, 中国
期限: 4 11月 2025 → 9 11月 2025

丛书

姓名EMNLP 2025 - 2025 Conference on Empirical Methods in Natural Language Processing, Findings of EMNLP 2025

会议

会议30th Conference on Empirical Methods in Natural Language Processing, EMNLP 2025
国家/地区中国
市Suzhou
时期4/11/25 → 9/11/25

学术指纹

探究 'Scale Down to Speed Up: Dynamic Data Selection for Reinforcement Learning' 的科研主题。它们共同构成独一无二的学术指纹。

引用此