← Search

Wenbo Su

30 accepted papers

2026

Asymmetric Proximal Policy Optimization: mini-critics boost LLM reasoning

ICLR 2026poster

Reinforcement learning (RL) has become a central paradigm for post-training large language models (LLMs) to elicit stronger reasoning. Yet, most recent RL for LLMs (RL4LLM) methods avoid explicit critics, replacing them with average advantage baselines. This shift is largely pragmatic: conventional…

Cited by 0SourceScholar
2026

Attention Illuminates LLM Reasoning: The Uncovered Preplan-and-Anchor Rhythm Enables Fine-Grained Policy Optimization

ICML 2026poster

The reasoning patterns of large language models (LLMs) remain opaque, and Reinforcement learning (RL) typically assigns uniform credit across an entire generation, blurring the distinction between pivotal and routine steps. This work treats attention as a natural substrate for interpreting LLM reaso…

Cited by 0SourceScholar
2026

COMI: Coarse-to-fine Context Compression via Marginal Information Gain

ICLR 2026poster

Large Language Models (LLMs) have demonstrated exceptional capabilities across diverse tasks. However, their deployment in long context scenarios remains hindered by computational inefficiency and information redundancy. Context compression methods address these challenges by significantly reducing…

Cited by 0SourcecodeScholar
2026

DESIGNER: Design-Logic-Guided Multidisciplinary Data Synthesis for LLM Reasoning

ICLR 2026poster

Large language models (LLMs) perform strongly on many language tasks but still struggle with complex multi-step reasoning across disciplines. Existing reasoning datasets often lack disciplinary breadth, reasoning depth, and diversity, as well as guiding principles for question synthesis. We propose…

Cited by 0SourceScholar
2026

Expert Divergence Learning for MoE-based Language Models

ICLR 2026poster

The Mixture-of-Experts (MoE) architecture is a powerful technique for scaling language models, yet it often suffers from expert homogenization, where experts learn redundant functionalities, thereby limiting MoE's full potential. To address this, we introduce Expert Divergence Learning, a novel pre-…

Cited by 0SourceScholar
2026

How Does Reasoning Flow? Tracing Attention-Induced Information Flow for Targeted RL in LLMs

ICML 2026poster

Token-level credit assignment remains a key obstacle for reinforcement learning (RL) in large language models (LLMs), where RL recipes typically treat all tokens equally, failing to distinguish decisive reasoning steps from routine formatting or fluent filler. Recent attempts leverage model-internal…

Cited by 0SourceScholar
2026

MeSH: Memory-as-State-Highways for Recursive Transformers

ICLR 2026poster

Recursive transformers reuse parameters and iterate over hidden states multiple times, decoupling compute depth from parameter depth. However, under matched compute, recursive models with fewer parameters often lag behind non-recursive counterparts. By probing hidden states, we trace this performanc…

Cited by 0SourcecodeScholar
2026

Reasoning Palette: Modulating Reasoning via Latent Contextualization for Controllable Exploration for (V)LMs

CVPR 2026

Exploration capacity shapes both inference-time performance and reinforcement learning (RL) training for large (vision-) language models, as stochastic sampling often yields redundant reasoning paths with little high-level diversity. This paper proposes Reasoning Palette, a novel latent-modulation f

Cited by 0SourceScholar
2026

Reconstructing KV Caches with Cross-Layer Fusion for Enhanced Transformers

ICLR 2026poster

Transformer decoders have achieved strong results across tasks, but the memory required for the KV cache becomes prohibitive at long sequence lengths. Although Cross-layer KV Cache sharing (e.g., YOCO, CLA) offers a path to mitigate KV Cache bottleneck, it typically underperforms within-layer method…

Cited by 0SourcecodeScholar
2026

Think-J: Learning to Think for Generative LLM-as-a-Judge

AAAI 2026technical

LLM-as-a-Judge refers to the automatic modeling of preferences for responses generated by Large Language Models (LLMs), which is of significant importance for both LLM evaluation and reward modeling. Although generative LLMs have made substantial progress in various tasks, their performance as LLM-J

Cited by 0SourcePDFScholar
2026

Tricks or Traps? A Deep Dive into RL for LLM Reasoning

ICLR 2026poster

Reinforcement learning (RL) for LLM reasoning has rapidly emerged as a prominent research area, marked by a significant surge in related studies on both algorithmic innovations and practical applications. Despite this progress, several critical challenges remain, including the absence of standardize…

Cited by 0SourcecodeScholar
2025

2D-DPO: Scaling Direct Preference Optimization with 2-Dimensional Supervision

NAACL 2025findings

Recent advancements in Direct Preference Optimization (DPO) have significantly enhanced the alignment of Large Language Models (LLMs) with human preferences, owing to its simplicity and effectiveness. However, existing methods typically optimize a scalar score or ranking reward, thereby overlooking…

Cited by 2SourcePDFScholar
2025

AIR: Complex Instruction Generation via Automatic Iterative Refinement

EMNLP 2025

With the development of large language models, their ability to follow simple instructions has significantly improved. However, adhering to complex instructions remains a major challenge. Current approaches to generating complex instructions are often irrelevant to the current instruction requiremen

2025

Can Large Language Models Detect Errors in Long Chain-of-Thought Reasoning?

ACL 2025long

Recently, o1-like models have drawn significant attention, where these models produce the long Chain-of-Thought (CoT) reasoning steps to improve the reasoning abilities of existing Large Language Models (LLMs). In this paper, to understand the qualities of these long CoTs and measure the critique ab…

2025

Chinese SafetyQA: A Safety Short-form Factuality Benchmark for Large Language Models

ACL 2025long

With the rapid advancement of Large Language Models (LLMs), significant safety concerns have emerged. Fundamentally, the safety of large language models is closely linked to the accuracy, comprehensiveness, and clarity of their understanding of safety knowledge, particularly in domains such as law,…

Cited by 0SourcePDFScholar
2025

Chinese SimpleQA: A Chinese Factuality Evaluation for Large Language Models

ACL 2025long

New LLM benchmarks are important to align with the rapid development of Large Language Models (LLMs). In this work, we present Chinese SimpleQA, the first comprehensive Chinese benchmark to evaluate the factuality ability of LLMs to answer short questions, and Chinese SimpleQA mainly has five proper…

2025

How to inject knowledge efficiently? Knowledge Infusion Scaling Law for Pre-training Large Language Models

EMNLP 2025

Large language models (LLMs) have attracted significant attention due to their impressive general capabilities across diverse downstream tasks. However, without domain-specific optimization, they often underperform on specialized knowledge benchmarks and even produce hallucination. Recent studies sh

Cited by 0SourcePDFScholar
2025

M2RC-EVAL: Massively Multilingual Repository-level Code Completion Evaluation

ACL 2025long

Repository-level code completion has drawn great attention in software engineering, and several benchmarks have been introduced. However, existing repository-level code completion benchmarks usually focus on a limited number of languages (<5), which cannot evaluate the general code intelligence abil…

2025

MTU-Bench: A Multi-granularity Tool-Use Benchmark for Large Language Models

ICLR 2025poster

Large Language Models (LLMs) have displayed massive improvements in reason- ing and decision-making skills and can hold natural conversations with users. Recently, many tool-use benchmark datasets have been proposed. However, existing datasets have the following limitations: (1). Insufficient evalua…

2025

ProgCo: Program Helps Self-Correction of Large Language Models

ACL 2025short

Self-Correction aims to enable large language models (LLMs) to self-verify and self-refine their initial responses without external feedback. However, LLMs often fail to effectively self-verify and generate correct feedback, further misleading refinement and leading to the failure of self-correction…

2025

See the World, Discover Knowledge: A Chinese Factuality Evaluation for Large Vision Language Models

ACL 2025finding

The evaluation of factual accuracy in large vision language models (LVLMs) has lagged behind their rapid development, making it challenging to fully reflect these models’ knowledge capacity and reliability. In this paper, we introduce the first factuality-based visual question-answering benchmark in…

Cited by 0SourcePDFScholar
2025

SuperGPQA: Scaling LLM Evaluation across 285 Graduate Disciplines

NeurIPS 2025poster

Large language models (LLMs) have demonstrated remarkable proficiency in mainstream academic disciplines such as mathematics, physics, and computer science. However, human knowledge encompasses over 200 specialized disciplines, far exceeding the scope of existing benchmarks. The capabilities of LLMs…

Cited by 215SourceScholar
2025

Towards Visualization-of-Thought Jailbreak Attack against Large Visual Language Models

NeurIPS 2025poster

As Visual Language Models (VLMs) continue to evolve, they have demonstrated increasingly sophisticated logical reasoning capabilities and multimodal thought generation, opening doors to widespread applications. However, this advancement raises serious concerns about content security, particularly wh…

Cited by 0SourcecodeScholar
2024

ConceptMath: A Bilingual Concept-wise Benchmark for Measuring Mathematical Reasoning of Large Language Models

ACL 2024findings

This paper introduces ConceptMath, a bilingual (English and Chinese), fine-grained benchmark that evaluates concept-wise mathematical reasoning of Large Language Models (LLMs). Unlike traditional benchmarks that evaluate general mathematical reasoning with an average accuracy, ConceptMath systemical…

2024

D-CPT Law: Domain-specific Continual Pre-Training Scaling Law for Large Language Models

NeurIPS 2024poster

Continual Pre-Training (CPT) on Large Language Models (LLMs) has been widely used to expand the model’s fundamental understanding of specific downstream domains (e.g., math and code). For the CPT on domain-specific LLMs, one important question is how to choose the optimal mixture ratio between the g…

Cited by 18SourcePDFScholar
2024

DDK: Distilling Domain Knowledge for Efficient Large Language Models

NeurIPS 2024poster

Despite the advanced intelligence abilities of large language models (LLMs) in various applications, they still face significant computational and storage demands. Knowledge Distillation (KD) has emerged as an effective strategy to improve the performance of a smaller LLM (i.e., the student model)…

Cited by 10SourcePDFScholar
2024

E2-LLM: Efficient and Extreme Length Extension of Large Language Models

ACL 2024findings

Training Large Language Models (LLMs) to process extensive context lengths incurs prohibitive computational costs. Prevailing techniques for extending context capabilities in LLMs typically require not only additional training procedures but also access to datasets with long context (e.g., sequences…

Cited by 23SourcePDFScholar
2024

GraphReader: Building Graph-based Agent to Enhance Long-Context Abilities of Large Language Models

EMNLP 2024finding

Long-context capabilities are essential for large language models (LLMs) to tackle complex and long-input tasks. Despite numerous efforts made to optimize LLMs for long contexts, challenges persist in robustly processing long inputs. In this paper, we introduce GraphReader, a graph-based agent syste…

2024

MT-Bench-101: A Fine-Grained Benchmark for Evaluating Large Language Models in Multi-Turn Dialogues

ACL 2024long

The advent of Large Language Models (LLMs) has drastically enhanced dialogue systems. However, comprehensively evaluating the dialogue abilities of LLMs remains a challenge. Previous benchmarks have primarily focused on single-turn dialogues or provided coarse-grained and incomplete assessments of m…

2022

GBA: A Tuning-free Approach to Switch between Synchronous and Asynchronous Training for Recommendation Models

NeurIPS 2022accept

High-concurrency asynchronous training upon parameter server (PS) architecture and high-performance synchronous training upon all-reduce (AR) architecture are the most commonly deployed distributed training modes for recommendation models. Although synchronous AR training is designed to have higher…

Cited by 3SourcePDFScholar