paper-with-me

Papers

Speculative Reward Model Boosts Decision Making Ability of LLMs Cost-Effectively

2025-05-31 · Jiawei Gu, Shangsong Liang

Effective decision-making in Large Language Models (LLMs) is essential for handling intricate tasks. However, existing approaches prioritize performance but often overlook the balance between effectiveness and computational cost. To address this, we first introduce the 3E Criteria to systematically assess the cost-effectiveness of search strategies, revealing that existing methods often trade significant efficiency for marginal performance gains. To improve LLM decision-making while maintaining efficiency, we propose the Speculative Reward Model (SRM), a plug-and-play framework that seamlessly integrates with existing search strategies. Specifically, SRM employs an external reward assigner to predict optimal actions, reducing reliance on LLMs' internal self-evaluation. And a speculative verification mechanism is used to prune suboptimal choices and guide the search toward more promising steps. We evaluate SRM on several complex decision-making tasks including mathematical reasoning, planning and numerical reasoning in specialized domains. Experimental results show that SRM reduces costs to 1/10 of the original search framework on average while maintaining effectiveness.

📄 PDF Abstract BibTeX arXiv:2506.00396

Code (1)

kuvvius/speculative-rm 공식 구현 pytorch

Tasks

Decision MakingMathematical Reasoning

Methods 이 논문이 사용한 방법론

SRM SRM combines style transfer with an attention mechanism. Its main contribution is style pooling which utilizes both mean and standard deviation of the input features to improve…

Similar Papers 제목 키워드 기반

Decision-making with Speculative Opponent Models

2022-11-22 · Jing Sun, Shuo Chen, Cong Zhang, Yining Ma 외

Opponent modelling has proven effective in enhancing the decision-making of the controlled agent by constructing models of opponent agents. However, existing methods often rely on access to the observations and actions o…

Decision MakingSMACSMAC+Starcraft

Overcoming Joint Intractability with Lossless Hierarchical Speculative Decoding

2026-01-09 · Yuxuan Zhou, Fei Huang, Heng Li, Fengyi Wu 외 arxiv

Verification is a key bottleneck in improving inference speed while maintaining distribution fidelity in Speculative Decoding. Recent work has shown that sequence-level verification leads to a higher number of accepted t…

Speculative Decoding Reimagined for Multimodal Large Language Models

2025-05-20 · Luxi Lin, Zhihang Lin, Zhanpeng Zeng, Rongrong Ji

This paper introduces Multimodal Speculative Decoding (MSD) to accelerate Multimodal Large Language Models (MLLMs) inference. Speculative decoding has been shown to accelerate Large Language Models (LLMs) without sacrifi…

Language ModelingLanguage Modelling

Out-of-Vocabulary Sampling Boosts Speculative Decoding

2025-06-02 · Nadav Timor, Jonathan Mamou, Oren Pereg, Hongyang Zhang 외

Speculative decoding relies on fast and accurate drafters. Recent state-of-the-art language models employ larger and larger vocabularies, which significantly slows down drafters. One promising approach to boost the effic…

Guided Speculative Inference for Efficient Test-Time Alignment of LLMs

2025-06-04 · Jonathan Geuter, Youssef Mroueh, David Alvarez-Melis

We propose Guided Speculative Inference (GSI), a novel algorithm for efficient reward-guided decoding in large language models. GSI combines soft best-of-$n$ test-time scaling with a reward model $r(x,y)$ and speculative…

Math