← Search

Weichao Mao

6 accepted papers

2025

Teaching Language Models to Critique via Reinforcement Learning

ICML 2025poster

Teaching large language models (LLMs) to critique and refine their outputs is crucial for building systems that can iteratively improve, yet it is fundamentally limited by the ability to provide *accurate judgments* and *actionable suggestions*. In this work, we study LLM critics for code generation…

Cited by 2SourcePDFScholar
2023

Multi-Agent Meta-Reinforcement Learning: Sharper Convergence Rates with Task Similarity

NeurIPS 2023poster

Multi-agent reinforcement learning (MARL) has primarily focused on solving a single task in isolation, while in practice the environment is often evolving, leaving many related tasks to be solved. In this paper, we investigate the benefits of meta-learning in solving multiple MARL tasks collectively…

Cited by 8SourcePDFScholar
2022

A Mean-Field Game Approach to Cloud Resource Management with Function Approximation

NeurIPS 2022accept

Reinforcement learning (RL) has gained increasing popularity for resource management in cloud services such as serverless computing. As self-interested users compete for shared resources in a cluster, the multi-tenancy nature of serverless platforms necessitates multi-agent reinforcement learning (M…

Cited by 26SourcePDFScholar
2021

Near-Optimal Model-Free Reinforcement Learning in Non-Stationary Episodic MDPs

ICML 2021spotlight

We consider model-free reinforcement learning (RL) in non-stationary Markov decision processes. Both the reward functions and the state transition functions are allowed to vary arbitrarily over time as long as their cumulative variations do not exceed certain variation budgets. We propose Restarted…

Cited by 49SourcePDFScholar
2020

POLY-HOOT: Monte-Carlo Planning in Continuous Space MDPs with Non-Asymptotic Analysis

NeurIPS 2020poster

Monte-Carlo planning, as exemplified by Monte-Carlo Tree Search (MCTS), has demonstrated remarkable performance in applications with finite spaces. In this paper, we consider Monte-Carlo planning in an environment with continuous state-action spaces, a much less understood problem with important app…

Cited by 24SourcePDFScholar