← Search

Fan Lai

10 accepted papers

2026

CORRECT: COndensed eRror RECognition via knowledge Transfer in multi-agent systems

ICML 2026poster

Multi-agent systems (MAS) are increasingly capable of tackling complex real-world tasks, yet their reliance on inter-agent coordination, tool use, and long-horizon reasoning makes error recognition particularly challenging. Minor errors can propagate across agents, escalating into task failures whil…

Cited by 0SourceScholar
2026

XRPO: Pushing the Limits of GRPO with Targeted Exploration and Exploitation

ICML 2026poster

Reinforcement learning algorithms such as GRPO have driven recent advances in large language model (LLM) reasoning. While scaling the number of rollouts stabilizes training, existing approaches suffer from limited exploration on challenging prompts and leave informative feedback signals underexploit…

Cited by 0SourceScholar
2025

Act Only When It Pays: Efficient Reinforcement Learning for LLM Reasoning via Selective Rollouts

NeurIPS 2025poster

Reinforcement learning, such as PPO and GRPO, has powered recent breakthroughs in LLM reasoning. Scaling rollout to sample more prompts enables models to selectively use higher-quality data for training, which can stabilize RL training and improve model performance, but at the cost of significant co…

Cited by 0SourcecodeScholar
2025

Inv-Entropy: A Fully Probabilistic Framework for Uncertainty Quantification in Language Models

NeurIPS 2025poster

Large language models (LLMs) have transformed natural language processing, but their reliable deployment requires effective uncertainty quantification (UQ). Existing UQ methods are often heuristic and lack a fully probabilistic foundation. This paper begins by providing a theoretical justification f…

Cited by 0SourcecodeScholar
2024

Learn To be Efficient: Build Structured Sparsity in Large Language Models

NeurIPS 2024spotlight

Large Language Models (LLMs) have achieved remarkable success with their billion-level parameters, yet they incur high inference overheads. The emergence of activation sparsity in LLMs provides a natural approach to reduce this cost by involving only parts of the parameters for inference. However, e…

Cited by 14SourcePDFScholar
2023

Coverage-centric Coreset Selection for High Pruning Rates

ICLR 2023poster

One-shot coreset selection aims to select a representative subset of the training data, given a pruning rate, that can later be used to train future models while retaining high accuracy. State-of-the-art coreset selection methods pick the highest importance examples based on an importance metric and…

2022

FedScale: Benchmarking Model and System Performance of Federated Learning at Scale

ICML 2022spotlight

We present FedScale, a federated learning (FL) benchmarking suite with realistic datasets and a scalable runtime to enable reproducible FL research. FedScale datasets encompass a wide range of critical FL tasks, ranging from image classification and object detection to language modeling and speech r…