← Search

Wei Fu

13 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

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

Unveiling the Pruning Risks on Privacy Vulnerabilities of Deep Neural Networks

ICASSP 2025accepted

Large-scale deep neural networks (DNNs), such as large language models, have gained immense popularity due to their outstanding performance across various tasks. However, their application in resource-constrained scenarios faces significant challenges due to the high computational costs and memory u…

Cited by 0SourceScholar
2024

Hyper-opinion Evidential Deep Learning for Out-of-Distribution Detection

NeurIPS 2024poster

Evidential Deep Learning (EDL), grounded in Evidence Theory and Subjective Logic (SL), provides a robust framework to estimate uncertainty for out-of-distribution (OOD) detection alongside traditional classification probabilities.However, the EDL framework is constrained by its focus on evidence tha…

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

Iteratively Learn Diverse Strategies with State Distance Information

NeurIPS 2023poster

In complex reinforcement learning (RL) problems, policies with similar rewards may have substantially different behaviors. It remains a fundamental challenge to optimize rewards while also discovering as many *diverse* strategies as possible, which can be crucial in many practical applications. Our…

Cited by 4SourcePDFScholar
2022

Continuously Discovering Novel Strategies via Reward-Switching Policy Optimization

ICLR 2022poster

We present Reward-Switching Policy Optimization (RSPO), a paradigm to discover diverse strategies in complex RL environments by iteratively finding novel policies that are both locally optimal and sufficiently different from existing ones. To encourage the learning policy to consistently converge to…

Cited by 36SourcePDFScholar
2022

Revisiting Some Common Practices in Cooperative Multi-Agent Reinforcement Learning

ICML 2022spotlight

Many advances in cooperative multi-agent reinforcement learning (MARL) are based on two common design principles: value decomposition and parameter sharing. A typical MARL algorithm of this fashion decomposes a centralized Q-function into local Q-networks with parameters shared across agents. Such a…

Cited by 49SourcePDFScholar