← Search

Zijun Yao

24 accepted papers

2026

On Predictability of Reinforcement Learning Dynamics for Large Language Models

ICLR 2026poster

Recent advances in reasoning capabilities of large language models (LLMs) are largely driven by reinforcement learning (RL), yet the underlying parameter dynamics during RL training remain poorly understood. This work identifies two fundamental properties of RL-induced parameter updates in LLMs: (1)…

Cited by 0SourcecodeScholar
2026

R-Diverse: Mitigating Diversity Illusion in Self-Play LLM Training

ICML 2026poster

Self-play bootstraps LLM reasoning through an iterative Challenger–Solver loop: the Challenger is trained to generate questions that target the Solver's capabilities, and the Solver is optimized on the generated data to expand its reasoning skills. However, existing frameworks like R-Zero often exhi…

Cited by 0SourceScholar
2025

Agentic Reward Modeling: Integrating Human Preferences with Verifiable Correctness Signals for Reliable Reward Systems

ACL 2025long

Reward models (RMs) are crucial for the training and inference-time scaling up of large language models (LLMs). However, existing reward models primarily focus on human preferences, neglecting verifiable correctness signals which have shown strong potential in training LLMs. In this paper, we propos…

2025

How do Transformers Learn Implicit Reasoning?

NeurIPS 2025spotlight

Recent work suggests that large language models (LLMs) can perform multi-hop reasoning implicitly---producing correct answers without explicitly verbalizing intermediate steps---but the underlying mechanisms remain poorly understood. In this paper, we study how such implicit reasoning emerges by tra…

Cited by 0SourceScholar
2025

LinguaLens: Towards Interpreting Linguistic Mechanisms of Large Language Models via Sparse Auto-Encoder

EMNLP 2025

Large language models (LLMs) demonstrate exceptional performance on tasks requiring complex linguistic abilities, such as reference disambiguation and metaphor recognition/generation. Although LLMs possess impressive capabilities, their internal mechanisms for processing and representing linguistic

Cited by 0SourcePDFScholar
2025

Pre-training Distillation for Large Language Models: A Design Space Exploration

ACL 2025long

Knowledge distillation (KD) aims to transfer knowledge from a large teacher model to a smaller student model. Previous work applying KD in the field of large language models (LLMs) typically focused on the post-training phase, where the student LLM learns directly from instructions and corresponding…

Cited by 0SourcePDFScholar
2025

RM-Bench: Benchmarking Reward Models of Language Models with Subtlety and Style

ICLR 2025oral

Reward models are critical in techniques like Reinforcement Learning from Human Feedback (RLHF) and Inference Scaling Laws, where they guide language model alignment and select optimal responses. Despite their importance, existing reward model benchmarks often evaluate models by asking them to dist…

2025

SeaKR: Self-aware Knowledge Retrieval for Adaptive Retrieval Augmented Generation

ACL 2025long

Adaptive Retrieval-Augmented Generation (RAG) is an effective strategy to alleviate hallucination of large language models (LLMs). It dynamically determines whether LLMs need external knowledge for generation and invokes retrieval accordingly. This paper introduces Self-aware Knowledge Retrieval (Se…

2025

Steering LVLMs via Sparse Autoencoder for Hallucination Mitigation

EMNLP 2025

Large vision-language models (LVLMs) have achieved remarkable performance on multimodal tasks. However, they still suffer from hallucinations, generating text inconsistent with visual input, posing significant risks in real-world applications. Existing approaches to address this issue focus on incor

2025

T1: Advancing Language Model Reasoning through Reinforcement Learning and Inference Scaling

ICML 2025poster

Large language models (LLMs) have demonstrated remarkable capabilities in complex reasoning tasks. However, existing approaches mainly rely on imitation learning and struggle to achieve effective test-time scaling. While reinforcement learning (RL) holds promise for enabling self-exploration, recent…

2025

TableLLM: Enabling Tabular Data Manipulation by LLMs in Real Office Usage Scenarios

ACL 2025finding

We introduce TableLLM, a robust large language model (LLM) with 8 billion parameters, purpose-built for proficiently handling tabular data manipulation tasks, whether they are embedded within documents or spreadsheets, catering to real-world office scenarios. We propose a distant supervision method…

2025

Towards Understanding Safety Alignment: A Mechanistic Perspective from Safety Neurons

NeurIPS 2025poster

Large language models (LLMs) excel in various capabilities but pose safety risks such as generating harmful content and misinformation, even after safety alignment. In this paper, we explore the inner mechanisms of safety alignment through the lens of mechanistic interpretability, focusing on identi…

Cited by 0SourceScholar
2024

A Cause-Effect Look at Alleviating Hallucination of Knowledge-grounded Dialogue Generation

COLING 2024main

Empowered by the large-scale pretrained language models, existing dialogue systems have demonstrated impressive performance conducting fluent and natural-sounding conversations. However, they are still plagued by the <b>hallucination</b> problem, causing unpredictable factual errors in the generated…

2024

Evaluating Generative Language Models in Information Extraction as Subjective Question Correction

COLING 2024main

Modern Large Language Models (LLMs) have showcased remarkable prowess in various tasks necessitating sophisticated cognitive behaviors. Nevertheless, a paradoxical performance discrepancy is observed, where these models underperform in seemingly elementary tasks like relation extraction and event ex…

2024

KoLA: Carefully Benchmarking World Knowledge of Large Language Models

ICLR 2024poster

The unprecedented performance of large language models (LLMs) necessitates improvements in evaluations. Rather than merely exploring the breadth of LLM abilities, we believe meticulous and thoughtful designs are essential to thorough, unbiased, and applicable evaluations. Given the importance of wor…

2024

Transferable and Efficient Non-Factual Content Detection via Probe Training with Offline Consistency Checking

ACL 2024long

This paper proposes PiNose, which trains a probing model on offline self-consistency checking results, thereby circumventing the need for human-annotated data and achieving transferability across diverse data distributions. As the consistency check process is offline, PiNose reduces the computationa…

2024

Untangle the KNOT: Interweaving Conflicting Knowledge and Reasoning Skills in Large Language Models

COLING 2024main

Providing knowledge documents for large language models (LLMs) has emerged as a promising solution to update the static knowledge inherent in their parameters. However, knowledge in the document may conflict with the memory of LLMs due to outdated or incorrect knowledge in the LLMs’ parameters. This…

2023

FFAEval: Evaluating Dialogue System via Free-For-All Ranking

EMNLP 2023long findings

Evaluating open-domain dialogue systems is currently an open question. Automatic evaluation metrics have shown poor correlation with human assessment in dialogue generation tasks. Human evaluation, which involves annotators for multi-dimension scoring, is trustworthy but time-consuming. In this wor…

Cited by 0SourceScholar
2023

KoRC: Knowledge Oriented Reading Comprehension Benchmark for Deep Text Understanding

ACL 2023findings

Deep text understanding, which requires the connections between a given document and prior knowledge beyond its text, has been highlighted by many benchmarks in recent years. However, these benchmarks have encountered two major limitations. On the one hand, most of them require human annotation of k…

2023

Probabilistic Tree-of-thought Reasoning for Answering Knowledge-intensive Complex Questions

EMNLP 2023long findings

Large language models (LLMs) are capable of answering knowledge-intensive complex questions with chain-of-thought (CoT) reasoning. However, they tend to generate factually incorrect reasoning steps when the required knowledge is not available or up-to-date in models' parameters. Recent works turn to…

Cited by 0SourcecodeScholar
2022

Dependency Parsing via Sequence Generation

EMNLP 2022finding

Dependency parsing aims to extract syntactic dependency structure or semantic dependency structure for sentences.Existing methods for dependency parsing include transition-based method, graph-based method and sequence-to-sequence method.These methods obtain excellent performance and we notice them b…

2022

Invernet: An Inversion Attack Framework to Infer Fine-Tuning Datasets through Word Embeddings

EMNLP 2022finding

Word embedding aims to learn the dense representation of words and has become a regular input preparation in many NLP tasks. Due to the data and computation intensive nature of learning embeddings from scratch, a more affordable way is to borrow the pretrained embedding available in public and fine-…

2022

Program Transfer for Answering Complex Questions over Knowledge Bases

ACL 2022long

Program induction for answering complex questions over knowledge bases (KBs) aims to decompose a question into a multi-step program, whose execution against the KB produces the final answer. Learning to induce programs relies on a large number of parallel question-program pairs for the given KB. How…

2021

Interpretable and Low-Resource Entity Matching via Decoupling Feature Learning from Decision Making

ACL 2021long

Entity Matching (EM) aims at recognizing entity records that denote the same real-world object. Neural EM models learn vector representation of entity descriptions and match entities end-to-end. Though robust, these methods require many annotated resources for training, and lack of interpretability.…