← Search

Zhihui Xie

14 accepted papers

2026

Attention as a Compass: Efficient Exploration for Process-Supervised RL in Reasoning Models

ICLR 2026poster

Reinforcement Learning (RL) has shown remarkable success in enhancing the reasoning capabilities of Large Language Models (LLMs). Process-Supervised RL (PSRL) has emerged as a more effective paradigm compared to outcome-based RL. However, existing PSRL approaches suffer from limited exploration effi…

Cited by 0SourcecodeScholar
2026

Can Simple Denoising Improve Uniform State Diffusion Models?

ICML 2026poster

Recent Uniform-state Diffusion Models (USDMs), initialized from a uniform prior, offer the promise of fast text generation due to their inherent self-correction ability compared to masked diffusion models. However, they still rely on complex loss formulations with additional computational overhead, …

Cited by 0SourceScholar
2026

DreamOn: Diffusion Language Models For Code Infilling Beyond Fixed-size Canvas

ICLR 2026poster

Diffusion Language Models (DLMs) present a compelling alternative to autoregressive models, offering flexible, any-order infilling without specialized prompting design. However, their practical utility is blocked by a critical limitation: the requirement of a fixed-length masked sequence for generat…

Cited by 0SourcecodeScholar
2026

Self-Distilled Reasoner: On-Policy Self-Distillation for Large Language Models

ICML 2026poster

Knowledge distillation improves large language model (LLM) reasoning by compressing the knowledge of a teacher LLM to train smaller LLMs. On-policy distillation advances this approach by having the student sample its own trajectories while a teacher LLM provides dense token-level supervision, addres…

Cited by 0SourceScholar
2025

Jailbreaking as a Reward Misspecification Problem

ICLR 2025poster

The widespread adoption of large language models (LLMs) has raised concerns about their safety and reliability, particularly regarding their vulnerability to adversarial attacks. In this paper, we propose a new perspective that attributes this vulnerability to reward misspecification during the alig…

2025

Long Chain-of-Thought Fine-tuning via Understanding-to-Reasoning Transition

EMNLP 2025

Reasoning models have demonstrated remarkable performance on complex tasks by generating long reasoning traces prior to producing final answers. However, previous research on long-context scaling in language models has generally focused on managing lengthy input prompts instead of producing long out

2025

Teaching Language Models to Critique via Reinforcement Learning

ICML 2025poster

Teaching large language models (LLMs) to critique and refine their outputs is crucial for building systems that can iteratively improve, yet it is fundamentally limited by the ability to provide *accurate judgments* and *actionable suggestions*. In this work, we study LLM critics for code generation…

Cited by 2SourcePDFScholar
2025

VL-RewardBench: A Challenging Benchmark for Vision-Language Generative Reward Models

CVPR 2025highlight

Vision-language generative reward models (VL-GenRMs) play a crucial role in aligning and evaluating multimodal AI systems, yet their own evaluation remains under-explored. Current assessment methods primarily rely on AI-annotated preference labels from traditional VL tasks, which can introduce biase…

2024

Calibrating Reasoning in Language Models with Internal Consistency

NeurIPS 2024poster

Large language models (LLMs) have demonstrated impressive capabilities in various reasoning tasks, aided by techniques like chain-of-thought prompting that elicits verbalized reasoning. However, LLMs often generate text with obvious mistakes and contradictions, raising doubts about their ability to…

2024

Learning Versatile Skills with Curriculum Masking

NeurIPS 2024poster

Masked prediction has emerged as a promising pretraining paradigm in offline reinforcement learning (RL) due to its versatile masking schemes, enabling flexible inference across various downstream tasks with a unified model. Despite the versatility of masked prediction, it remains unclear how to bal…

2024

VLFeedback: A Large-Scale AI Feedback Dataset for Large Vision-Language Models Alignment

EMNLP 2024main

As large vision-language models (LVLMs) evolve rapidly, the demand for high-quality and diverse data to align these models becomes increasingly crucial. However, the creation of such data with human supervision proves costly and time-intensive. In this paper, we investigate the efficacy of AI feedba…

2023

Future-conditioned Unsupervised Pretraining for Decision Transformer

ICML 2023poster

Recent research in offline reinforcement learning (RL) has demonstrated that return-conditioned supervised learning is a powerful paradigm for decision-making problems. While promising, return conditioning is limited to training data labeled with rewards and therefore faces challenges in learning fr…

2022

Discovering Low-rank Subspaces for Language-agnostic Multilingual Representations

EMNLP 2022main

Large pretrained multilingual language models (ML-LMs) have shown remarkable capabilities of zero-shot cross-lingual transfer, without direct cross-lingual supervision. While these results are promising, follow-up works found that, within the multilingual embedding spaces, there exists strong langua…

2020

Layered Neighborhood Expansion for Incremental Multiple Graph Matching

ECCV 2020poster

Graph matching has been a fundamental problem in computer vision and pattern recognition, for its practical flexibility as well as NP hardness challenge. Though the matching between two graphs and among multiple graphs have been intensively studied in literature, the online setting for incremental m…