← Search

Zhepei Wei

10 accepted papers

2026

TruthRL: Incentivizing Truthful LLMs via Reinforcement Learning

ICML 2026poster

While large language models (LLMs) have demonstrated strong performance on factoid question answering, they are still prone to hallucination and untruthful responses, particularly when tasks demand information outside their parametric knowledge. Indeed, truthfulness requires more than accuracy---mod…

Cited by 0SourceScholar
2025

AdaDecode: Accelerating LLM Decoding with Adaptive Layer Parallelism

ICML 2025poster

Large language models (LLMs) are increasingly used for long-content generation (e.g., long Chain-of-Thought reasoning) where decoding efficiency becomes a critical bottleneck: Autoregressive decoding is inherently limited by its sequential token generation process, where each token must be generated…

2025

InstructRAG: Instructing Retrieval-Augmented Generation via Self-Synthesized Rationales

ICLR 2025poster

Retrieval-augmented generation (RAG) has shown promising potential to enhance the accuracy and factuality of language models (LMs). However, imperfect retrievers or noisy corpora can introduce misleading or even erroneous information to the retrieved contents, posing a significant challenge to the g…

2025

Separate the Wheat from the Chaff: Winnowing Down Divergent Views in Retrieval Augmented Generation

EMNLP 2025

Retrieval-augmented generation (RAG) addresses the limitation of large language models (LLMs) in achieving up-to-date information by integrating external knowledge sources, but it is hindered by noisy or irrelevant retrieved data, leading to reduced accuracy. Additionally, most RAG methods rely on t

Cited by 0SourcePDFScholar
2025

The Surprising Effectiveness of Negative Reinforcement in LLM Reasoning

NeurIPS 2025poster

Reinforcement learning with verifiable rewards (RLVR) is a promising approach for training language models (LMs) on reasoning tasks that elicit emergent long chains of thought (CoTs). Unlike supervised learning, it updates the model using both correct and incorrect samples via policy gradients. To b…

Cited by 0SourcecodeScholar
2025

WebAgent-R1: Training Web Agents via End-to-End Multi-Turn Reinforcement Learning

EMNLP 2025

While reinforcement learning (RL) has demonstrated remarkable success in enhancing large language models (LLMs), it has primarily focused on single-turn tasks such as solving math problems. Training effective web agents for multi-turn interactions remains challenging due to the complexity of long-ho

2024

Incentivized Truthful Communication for Federated Bandits

ICLR 2024poster

To enhance the efficiency and practicality of federated bandit learning, recent advances have introduced incentives to motivate communication among clients, where a client participates only when the incentive offered by the server outweighs its participation cost. However, existing incentive mechani…

Cited by 1SourcePDFScholar
2022

AttExplainer: Explain Transformer via Attention by Reinforcement Learning

IJCAI 2022poster

Transformer and its variants, built based on attention mechanisms, have recently achieved remarkable performance in many NLP tasks. Most existing works on Transformer explanation tend to reveal and utilize the attention matrix with human subjective intuitions in a qualitative manner. However, the hu…

2022

Learning Semantic Textual Similarity via Topic-informed Discrete Latent Variables

EMNLP 2022main

Recently, discrete latent variable models have received a surge of interest in both Natural Language Processing (NLP) and Computer Vision (CV), attributed to their comparable performance to the continuous counterparts in representation learning, while being more interpretable in their predictions. I…