← Search

Yixiu Mao

13 accepted papers

2026

Dynamics-Predictive Sampling for Active RL Finetuning of Large Reasoning Models

ICLR 2026poster

Reinforcement learning (RL) finetuning has become a key technique for enhancing the reasoning abilities of large language models (LLMs). However, its effectiveness critically depends on the selection of training data. Recent advances underscore the importance of online prompt selection methods, whic…

Cited by 0SourcecodeScholar
2026

Enhancing Generative Auto-bidding with Offline Reward Evaluation and Policy Search

ICLR 2026oral

Auto-bidding serves as a critical tool for advertisers to improve their advertising performance. Recent progress has demonstrated that AI-Generated Bidding (AIGB), which learns a conditional generative planner from offline data, achieves superior performance compared to typical offline reinforcement…

Cited by 0SourceScholar
2026

Small Generalizable Prompt Predictive Models Can Steer Efficient RL Post-Training of Large Reasoning Models

ICML 2026poster

Reinforcement learning enhances the reasoning capabilities of large language models but often involves high computational costs due to rollout-intensive optimization. Online prompt selection presents a plausible solution by prioritizing informative prompts to improve training efficiency. However, cu…

Cited by 0SourceScholar
2026

Utility-Diversity Aware Online Batch Selection for LLM Supervised Fine-tuning

ICML 2026poster

Supervised fine-tuning (SFT) is a commonly used technique to adapt large language models (LLMs) to downstream tasks. In practice, SFT on a full dataset is computationally expensive and sometimes suffers from overfitting or bias amplification. This facilitates the rise of data curation in SFT, which …

Cited by 0SourceScholar
2025

Adaptive Neighborhood-Constrained Q Learning for Offline Reinforcement Learning

NeurIPS 2025spotlight

Offline reinforcement learning (RL) suffers from extrapolation errors induced by out-of-distribution (OOD) actions. To address this, offline RL algorithms typically impose constraints on action selection, which can be systematically categorized into density, support, and sample constraints. However,…

Cited by 0SourceScholar
2025

Fast and Robust: Task Sampling with Posterior and Diversity Synergies for Adaptive Decision-Makers in Randomized Environments

ICML 2025poster

Task robust adaptation is a long-standing pursuit in sequential decision-making. Some risk-averse strategies, e.g., the conditional value-at-risk principle, are incorporated in domain randomization or meta reinforcement learning to prioritize difficult tasks in optimization, which demand costly inte…

Cited by 0SourcePDFScholar
2025

Latent Reward: LLM-Empowered Credit Assignment in Episodic Reinforcement Learning

AAAI 2025technical

Reinforcement learning (RL) often encounters delayed and sparse feedback in real-world applications, even with only episodic rewards. Previous approaches have made some progress in reward redistribution for credit assignment but still face challenges, including training difficulties due to redundan…

2024

Doubly Mild Generalization for Offline Reinforcement Learning

NeurIPS 2024poster

Offline Reinforcement Learning (RL) suffers from the extrapolation error and value overestimation. From a generalization perspective, this issue can be attributed to the over-generalization of value functions or policies towards out-of-distribution (OOD) actions. Significant efforts have been devote…

2024

Offline Reinforcement Learning with OOD State Correction and OOD Action Suppression

NeurIPS 2024poster

In offline reinforcement learning (RL), addressing the out-of-distribution (OOD) action issue has been a focus, but we argue that there exists an OOD state issue that also impairs performance yet has been underexplored. Such an issue describes the scenario when the agent encounters states out of the…

2023

In-sample Actor Critic for Offline Reinforcement Learning

ICLR 2023poster

Offline reinforcement learning suffers from out-of-distribution issue and extrapolation error. Most methods penalize the out-of-distribution state-action pairs or regularize the trained policy towards the behavior policy but cannot guarantee to get rid of extrapolation error. We propose In-sample…

Cited by 13SourcePDFScholar
2023

Supported Trust Region Optimization for Offline Reinforcement Learning

ICML 2023poster

Offline reinforcement learning suffers from the out-of-distribution issue and extrapolation error. Most policy constraint methods regularize the density of the trained policy towards the behavior policy, which is too restrictive in most cases. We propose Supported Trust Region optimization (STR) whi…

Cited by 16SourcePDFScholar
2023

Supported Value Regularization for Offline Reinforcement Learning

NeurIPS 2023poster

Offline reinforcement learning suffers from the extrapolation error and value overestimation caused by out-of-distribution (OOD) actions. To mitigate this issue, value regularization approaches aim to penalize the learned value functions to assign lower values to OOD actions. However, existing value…

2021

A Hypergradient Approach to Robust Regression without Correspondence

ICLR 2021poster

We consider a regression problem, where the correspondence between the input and output data is not available. Such shuffled data are commonly observed in many real world problems. Take flow cytometry as an example: the measuring instruments are unable to preserve the correspondence between the samp…

Cited by 18SourcePDFScholar