← Search

Jiaxuan Gao

15 accepted papers

2026

AReaL-DTA: Dynamic Tree Attention for Efficient Reinforcement Learning of Large Language Models

ICML 2026poster

Reinforcement learning (RL) based post-training for large language models (LLMs) is computationally expensive, as it generates many rollout sequences that could frequently share long token prefixes. Existing RL frameworks usually process these sequences independently, repeatedly recomputing identica…

Cited by 0SourceScholar
2026

QuestA: Expanding Reasoning Capacity in LLMs via Question Augmentation

ICLR 2026poster

Reinforcement learning (RL) has emerged as a central paradigm for training large language models (LLMs) in reasoning tasks. Yet recent studies question RL’s ability to incentivize reasoning capacity beyond the base model. This raises a key challenge: how can RL be adapted to solve harder reasoning p…

Cited by 0SourcecodeScholar
2026

Unlocking Long-Horizon Agentic Search with Large-Scale End-to-End RL

ICLR 2026poster

Recent advancements in LLM-based agents have demonstrated remarkable capabilities in handling knowledge-intensive tasks using external tools. One representative example is search agent. Existing open-source search agents heavily rely on advanced commercial LLMs: they either collect trajectories from…

Cited by 0SourcecodeScholar
2025

AREAL: A Large-Scale Asynchronous Reinforcement Learning System for Language Reasoning

NeurIPS 2025poster

Reinforcement learning (RL) has become a trending paradigm for training large language models (LLMs), particularly for reasoning tasks. Effective RL for LLMs requires massive parallelization and poses an urgent need for efficient training systems. Most existing large-scale RL systems for LLMs are sy…

Cited by 0SourcecodeScholar
2025

How Far Are We from Optimal Reasoning Efficiency?

NeurIPS 2025poster

Large Reasoning Models (LRMs) demonstrate remarkable problem-solving capabilities through extended Chain-of-Thought (CoT) reasoning but often produce excessively verbose and redundant reasoning traces. This inefficiency incurs high inference costs and limits practical deployment. While existing fine…

Cited by 0SourcecodeScholar
2025

Industrial-Grade Sensor Simulation via Gaussian Splatting: A Modular Framework for Scalable Editing and Full-Stack Validation

IROS 2025

Sensor simulation is pivotal for scalable validation of autonomous driving systems, yet existing Neural Radiance Fields (NeRF) based methods face applicability and efficiency challenges in industrial workflows. This paper introduces a Gaussian Splatting (GS) based system to address these challenges:

Cited by 3SourceScholar
2024

Is DPO Superior to PPO for LLM Alignment? A Comprehensive Study

ICML 2024oral

Reinforcement Learning from Human Feedback (RLHF) is currently the most widely used method to align large language models (LLMs) with human preferences. Existing RLHF methods can be roughly categorized as either reward-based or reward-free. Novel applications such as ChatGPT and Claude leverage rewa…

2024

Robot Generating Data for Learning Generalizable Visual Robotic Manipulation

IROS 2024poster

It has been a popular trend in AI to pretrain foundation models on massive data. However, collecting sufficient offline training trajectories for robot learning is particularly expensive since valid control actions are required. Therefore, most existing robotic datasets are collected from human expe…

Cited by 0SourceScholar
2024

SRL: Scaling Distributed Reinforcement Learning to Over Ten Thousand Cores

ICLR 2024poster

The ever-growing complexity of reinforcement learning (RL) tasks demands a distributed system to efficiently generate and process a massive amount of data. However, existing open-source libraries suffer from various limitations, which impede their practical use in challenging scenarios where large-s…

Cited by 6SourcePDFScholar
2023

Learning Zero-Shot Cooperation with Humans, Assuming Humans Are Biased

ICLR 2023poster

There is a recent trend of applying multi-agent reinforcement learning (MARL) to train an agent that can cooperate with humans in a zero-shot fashion without using any human data. The typical workflow is to first repeatedly run self-play (SP) to build a policy pool and then train the final adaptive…

2023

SpeedyZero: Mastering Atari with Limited Data and Time

ICLR 2023poster

Many recent breakthroughs of deep reinforcement learning (RL) are mainly built upon large-scale distributed training of model-free methods using millions to billions of samples. On the other hand, state-of-the-art model-based RL methods can achieve human-level sample efficiency but often take a much…

Cited by 5SourcePDFScholar
2022

Learning Efficient Multi-agent Cooperative Visual Exploration

ECCV 2022poster

"We tackle the problem of cooperative visual exploration where multiple agents need to jointly explore unseen regions as fast as possible based on visual signals. Classical planning-based methods often suffer from expensive computation overhead at each step and a limited expressiveness of complex co…

Cited by 37SourcePDFScholar
2022

The Surprising Effectiveness of PPO in Cooperative Multi-Agent Games

NeurIPS 2022accept

Proximal Policy Optimization (PPO) is a ubiquitous on-policy reinforcement learning algorithm but is significantly less utilized than off-policy learning algorithms in multi-agent settings. This is often due to the belief that PPO is significantly less sample efficient than off-policy methods in mul…