arXiv · 2512.14069
RADAR: Accelerate Large Language Model Inference With RL-Based Dynamic Draft Trees
Abstract
Inference with modern Large Language Models (LLMs) is expensive and slow, and speculative sampling has emerged as an effective solution to this problem. However, the number of calls to the draft model for generating candidate tokens in speculative sampling is a preset hyperparameter, lacking flexibility. To generate and utilize the candidate tokens more effectively, we propose RADAR, a novel speculative sampling method with RL-based dynamic draft trees. RADAR formulates the draft tree generation process as a Markov Decision Process (MDP) and employs offline reinforcement learning to train a prediction model, which enables real-time decisions on calls to the draft model, reducing redundant computations and further accelerating inference. Evaluations across three LLMs and four tasks show that RADAR achieves a speedup of 3.17x-4.82x over the auto-regressive decoding baseline. The code is available at https://github.com/minaduki-sora/RADAR.
Explore related subjects
Keep this discovery
Junjie Ma, Jinlong Li, Jiajun Luo. 2025-12-16. RADAR: Accelerate Large Language Model Inference With RL-Based Dynamic Draft Trees. https://arxiv.org/abs/2512.14069
Cite the original work for its findings. Save a collection to share your selection of sources.