← Search

Shaohang Wei

8 accepted papers

2026

Experience Augmented Policy Optimization for LLM Reasoning

ICML 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) is a powerful paradigm for improving the reasoning capabilities of large language models (LLMs). However, existing RLVR methods typically rely on on-policy optimization from scratch, resulting in high sampling costs and inefficient utilization of…

Cited by 0SourceScholar
2026

One-Way Policy Optimization for Self-Evolving LLMs

ICML 2026poster

Reinforcement Learning with Verifiable Rewards (RLVR) has become a promising paradigm for scaling reasoning capabilities of Large Language Models (LLMs). However, the sparsity of binary verifier rewards often leads to low efficiency and optimization instability. To stabilize training, existing metho…

Cited by 0SourceScholar
2026

Probability-Entropy Calibration: An Elastic Indicator for Adaptive Fine-tuning

ICML 2026poster

Token-level reweighting is a simple yet effective mechanism for controlling supervised fine-tuning, but common indicators are largely one-dimensional: the ground-truth probability reflects downstream alignment, while token entropy reflects intrinsic uncertainty induced by the pre-training prior. Ign…

Cited by 0SourceScholar
2026

Sparse but Critical: A Token-Level Analysis of Distributional Shifts in RLVR Fine-Tuning of LLMs

ICLR 2026poster

Reinforcement learning with verifiable rewards (RLVR) has significantly improved reasoning in large language models (LLMs), yet the token-level mechanisms through which they reshape model behavior remain unclear. We present a systematic empirical study of RLVR’s distributional effects across three c…

Cited by 0SourceScholar
2025

MeNTi: Bridging Medical Calculator and LLM Agent with Nested Tool Calling

NAACL 2025long

Integrating tools into Large Language Models (LLMs) has facilitated the widespread application. Despite this, in specialized downstream task contexts, reliance solely on tools is insufficient to fully address the complexities of the real world. This particularly restricts the effective deployment of…

Cited by 1SourcePDFScholar
2025

Odysseus Navigates the Sirens’ Song: Dynamic Focus Decoding for Factual and Diverse Open-Ended Text Generation

ACL 2025long

Large Language Models (LLMs) are increasingly required to generate text that is both factually accurate and diverse across various open-ended applications. However, current stochastic decoding methods struggle to balance such objectives. We introduce Dynamic Focus Decoding (DFD), a novel plug-and-pl…

Cited by 0SourcePDFScholar
2025

TimE: A Multi-level Benchmark for Temporal Reasoning of LLMs in Real-World Scenarios

NeurIPS 2025spotlight

Temporal reasoning is pivotal for Large Language Models (LLMs) to comprehend the real world. However, existing works neglect the real-world challenges for temporal reasoning: (1) intensive temporal information, (2) fast-changing event dynamics, and (3) complex temporal dependencies in social interac…

Cited by 0SourcecodeScholar
2025

Well Begun is Half Done: Low-resource Preference Alignment by Weak-to-Strong Decoding

ACL 2025finding

Large Language Models (LLMs) require alignment with human preferences to avoid generating offensive, false, or meaningless content. Recently, low-resource methods for LLM alignment have been popular, while still facing challenges in obtaining both high-quality and aligned content. Motivated by the o…