← Search

Weixun Wang

22 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

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

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

Smarter Not Harder: Generative Process Evaluation with Intrinsic-Signal Driving and Ability‑Adaptive Reward Shaping

ICLR 2026poster

Large reasoning models (LRMs) have shown strong performance in complex mathematical reasoning when optimized via reinforcement learning (RL). However, conventional outcome-only reward provides sparse feedback, leading to inefficient optimization. In this work, we investigate whether generative proce…

Cited by 0SourceScholar
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

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 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

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…

2024

PORTAL: Automatic Curricula Generation for Multiagent Reinforcement Learning

AAAI 2024technical

Despite many breakthroughs in recent years, it is still hard for MultiAgent Reinforcement Learning (MARL) algorithms to directly solve complex tasks in MultiAgent Systems (MASs) from scratch. In this work, we study how to use Automatic Curriculum Learning (ACL) to reduce the number of environmental…

Cited by 8SourcePDFScholar
2023

Boosting Multiagent Reinforcement Learning via Permutation Invariant and Permutation Equivariant Networks

ICLR 2023poster

The state space in Multiagent Reinforcement Learning (MARL) grows exponentially with the agent number. Such a curse of dimensionality results in poor scalability and low sample efficiency, inhibiting MARL for decades. To break this curse, we propose a unified agent permutation framework that exploit…

Cited by 30SourcePDFScholar
2022

Individual Reward Assisted Multi-Agent Reinforcement Learning

ICML 2022spotlight

In many real-world multi-agent systems, the sparsity of team rewards often makes it difficult for an algorithm to successfully learn a cooperative team policy. At present, the common way for solving this problem is to design some dense individual rewards for the agents to guide the cooperation. Howe…

2022

Transformer-based Working Memory for Multiagent Reinforcement Learning with Action Parsing

NeurIPS 2022accept

Learning in real-world multiagent tasks is challenging due to the usual partial observability of each agent. Previous efforts alleviate the partial observability by historical hidden states with Recurrent Neural Networks, however, they do not consider the multiagent characters that either the multia…

Cited by 20SourcePDFScholar
2021

An Efficient Transfer Learning Framework for Multiagent Reinforcement Learning

NeurIPS 2021poster

Transfer Learning has shown great potential to enhance single-agent Reinforcement Learning (RL) efficiency. Similarly, Multiagent RL (MARL) can also be accelerated if agents can share knowledge with each other. However, it remains a problem of how an agent should learn from other agents. In this pap…

2020

Action Semantics Network: Considering the Effects of Actions in Multiagent Systems

ICLR 2020poster

In multiagent systems (MASs), each agent makes individual decisions but all of them contribute globally to the system evolution. Learning in MASs is difficult since each agent's selection of actions must take place in the presence of other co-learning agents. Moreover, the environmental stochasticit…

Cited by 48SourcecodeScholar
2020

Efficient Deep Reinforcement Learning via Adaptive Policy Transfer

IJCAI 2020poster

Transfer learning has shown great potential to accelerate Reinforcement Learning (RL) by leveraging prior knowledge from past learned policies of relevant tasks. Existing approaches either transfer previous knowledge by explicitly computing similarities between tasks or select appropriate source pol…

2020

KoGuN: Accelerating Deep Reinforcement Learning via Integrating Human Suboptimal Knowledge

IJCAI 2020poster

Reinforcement learning agents usually learn from scratch, which requires a large number of interactions with the environment. This is quite different from the learning process of human. When faced with a new task, human naturally have the common sense and use the prior knowledge to derive an initial…

Cited by 0SourcePDFScholar
2020

Learning to Accelerate Heuristic Searching for Large-Scale Maximum Weighted b-Matching Problems in Online Advertising

IJCAI 2020poster

Bipartite b-matching is fundamental in algorithm design, and has been widely applied into diverse applications, such as economic markets, labor markets, etc. These practical problems usually exhibit two distinct features: large-scale and dynamic, which requires the matching algorithm to be repeatedl…

Cited by 0SourcePDFScholar
2020

Learning to Utilize Shaping Rewards: A New Approach of Reward Shaping

NeurIPS 2020poster

Reward shaping is an effective technique for incorporating domain knowledge into reinforcement learning (RL). Existing approaches such as potential-based reward shaping normally make full use of a given shaping reward function. However, since the transformation of human knowledge into numeric reward…

Cited by 246SourcePDFScholar