Efficient Hyperparameter Optimization for LLM Reinforcement Learning
Reinforcement learning (RL) for large language models (LLMs) is highly sensitive to hyperparameter configurations, making hyperparameter optimization (HPO) essential yet computationally expensive. Existing multi-fidelity HPO methods remain inefficient for LLM RL due to the massive model scale and resource-intensive training cycles. In this paper, we propose Joint Fidelity Hyperparameter Optimization (JF-HPO), which simultaneously adapts both model size and training budget as fidelity. JF-HPO is empowered by: (i) it leverages a small proxy model of the target LLM for efficient training and evaluation in each HPO trial; (ii) it integrates carefully designed early-stopping strategies based on training dynamics; (iii) it introduces an efficient checkpointing mechanism to eliminate redundant computations. Compared with existing HPO methods, JF-HPO significantly improves the computational efficiency of each trial (up to 14.9 times), while achieving better or competitive predictive accuracy under the same time budget. Notably, compared with utilizing hyperparameter configurations from the VeRL Recipe, JF-HPO delivers performance improvements ranging from 5.8% to 111.6%.
Code (0)
등록된 구현이 없습니다.
Tasks
Hyperparameter OptimizationComputational EfficiencyReinforcement LearningSimilar Papers 제목 키워드 기반
Quantity vs. Quality: On Hyperparameter Optimization for Deep Reinforcement Learning
Reinforcement learning algorithms can show strong variation in performance between training runs with different random seeds. In this paper we explore how this affects hyperparameter optimization when the goal is to find…
Bayesian OptimizationDeep Reinforcement LearningHyperparameter Optimizationreinforcement-learning+2HyperController: A Hyperparameter Controller for Fast and Stable Training of Reinforcement Learning Neural Networks
We introduce Hyperparameter Controller (HyperController), a computationally efficient algorithm for hyperparameter optimization during training of reinforcement learning neural networks. HyperController optimizes hyperpa…
Hyperparameter Optimizationreinforcement-learningReinforcement LearningHyperparameter Optimization for Driving Strategies Based on Reinforcement Learning
This paper focuses on hyperparameter optimization for autonomous driving strategies based on Reinforcement Learning. We provide a detailed description of training the RL agent in a simulation environment. Subsequently, w…
Autonomous DrivingBayesian Optimizationglobal-optimizationHyperparameter Optimization+2Hyperparameter Optimization for Multi-Objective Reinforcement Learning
Reinforcement learning (RL) has emerged as a powerful approach for tackling complex problems. The recent introduction of multi-objective reinforcement learning (MORL) has further expanded the scope of RL by enabling agen…
Hyperparameter OptimizationMulti-Objective Reinforcement Learningreinforcement-learningReinforcement Learning+1Metaoptimization on a Distributed System for Deep Reinforcement Learning
Training intelligent agents through reinforcement learning is a notoriously unstable procedure. Massive parallelization on GPUs and distributed systems has been exploited to generate a large amount of training experience…
Atari GamesDeep Reinforcement Learningreinforcement-learningReinforcement Learning+1