← Search

Shuibai Zhang

6 accepted papers

2026

ParallelBench: Understanding the Trade-offs of Parallel Decoding in Diffusion LLMs

ICLR 2026poster

While most autoregressive LLMs are constrained to one-by-one decoding, diffusion LLMs (dLLMs) have attracted growing interest for their potential to dramatically accelerate inference through parallel decoding. Despite this promise, the conditional independence assumption in dLLMs causes parallel dec…

Cited by 0SourcecodeScholar
2026

Planner Aware Path Learning in Diffusion Language Models Training

ICLR 2026oral

Diffusion language models have emerged as a powerful alternative to autoregressive models, enabling fast inference through more flexible and parallel generation paths. This flexibility of sampling is unlocked by new engineered sampling strategies, or *planners*, that select more favorable generation…

Cited by 13SourcecodeScholar
2026

ReJump: A Tree-Jump Representation for Analyzing and Improving LLM Reasoning

ICML 2026poster

Large Reasoning Models (LRMs) are Large Language Models (LLMs) explicitly trained to generate long-form Chain-of-Thoughts (CoTs), achieving impressive success on challenging tasks like math and programming. However, their underlying reasoning "algorithms" remain poorly understood. To investigate thi…

Cited by 0SourceScholar
2025

VersaPRM: Multi-Domain Process Reward Model via Synthetic Reasoning Data

ICML 2025oral

Process Reward Models (PRMs) have proven effective at enhancing mathematical reasoning for Large Language Models (LLMs) by leveraging increased inference-time computation. However, they are predominantly trained on mathematical data and their generalizability to non-mathematical domains has not been…

Cited by 4SourcePDFScholar
2024

Supervised Knowledge Makes Large Language Models Better In-context Learners

ICLR 2024poster

Large Language Models (LLMs) exhibit emerging in-context learning abilities through prompt engineering. The recent progress in large-scale generative models has further expanded their use in real-world language applications. However, the critical challenge of improving the generalizability and factu…

2023

GLUE-X: Evaluating Natural Language Understanding Models from an Out-of-Distribution Generalization Perspective

ACL 2023findings

Pre-trained language models (PLMs) are known to improve the generalization performance of natural language understanding models by leveraging large amounts of data during the pre-training phase. However, the out-of-distribution (OOD) generalization problem remains a challenge in many NLP tasks, limi…