← Search

Cheems Wang

21 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

Latent Space Robust Optimization of Neural Processes with Aligned Stratified Order-Statistic Loss Reduction

ICML 2026poster

Importance-Weighted Neural Processes (IWNPs) provide a principled framework for probabilistic meta-learning by using multi-particle latent representations to approximate the marginal log-likelihood of task data tightly. However, this work reveals that the standard optimization of IWNPs suffers from …

Cited by 0SourceScholar
2026

LoRA in LoRA: Towards Parameter-Efficient Architecture Expansion for Continual Visual Instruction Tuning

AAAI 2026technical

Continual Visual Instruction Tuning (CVIT) enables Multimodal Large Language Models (MLLMs) to incrementally learn new tasks over time. However, this process is challenged by catastrophic forgetting, where performance on previously learned tasks deteriorates as the model adapts to new ones. A common

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

Stochastic Gradient Methods under Heavy-Tailed Noises in Weakly Convex Optimization

ICML 2026poster

Recently, many empirical work has shown that, in machine learning, the noise distribution of stochastic gradients often exhibits heavy tails when stochastic optimization methods are employed. Most existing theoretical analyses of heavy-tailed stochastic methods rely on various convexity and smoothne…

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

DynaPrompt: Dynamic Test-Time Prompt Tuning

ICLR 2025poster

Test-time prompt tuning enhances zero-shot generalization of vision-language models but tends to ignore the relatedness among test samples during inference. Online test-time prompt tuning provides a simple way to leverage the information in previous test samples, albeit with the risk of prompt colla…

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

Gains: Fine-grained Federated Domain Adaptation in Open Set

NeurIPS 2025poster

Conventional federated learning (FL) assumes a closed world with a fixed total number of clients. In contrast, new clients continuously join the FL process in real-world scenarios, introducing new knowledge. This raises two critical demands: detecting new knowledge, i.e., knowledge discovery, and in…

Cited by 0SourcecodeScholar
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…

2025

Selective Learning for Deep Time Series Forecasting

NeurIPS 2025poster

Benefiting from high capacity for capturing complex temporal patterns, deep learning (DL) has significantly advanced time series forecasting (TSF). However, deep models tend to suffer from severe overfitting due to the inherent vulnerability of time series to noise and anomalies. The prevailing DL p…

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

GO4Align: Group Optimization for Multi-Task Alignment

NeurIPS 2024poster

This paper proposes **GO4Align**, a multi-task optimization approach that tackles task imbalance by explicitly aligning the optimization across tasks. To achieve this, we design an adaptive group risk minimization strategy, comprising two techniques in implementation: (i) dynamical group assignment,…

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…

2024

Reducing Fine-Tuning Memory Overhead by Approximate and Memory-Sharing Backpropagation

ICML 2024poster

Fine-tuning pretrained large models to downstream tasks is an important problem, which however suffers from huge memory overhead due to large-scale parameters. This work strives to reduce memory overhead in fine-tuning from perspectives of activation function and layer normalization. To this end, we…

2024

Theoretical Investigations and Practical Enhancements on Tail Task Risk Minimization in Meta Learning

NeurIPS 2024poster

Meta learning is a promising paradigm in the era of large models and task distributional robustness has become an indispensable consideration in real-world scenarios. Recent advances have examined the effectiveness of tail task risk minimization in fast adaptation robustness improvement \citep{wang…

2023

A Simple Yet Effective Strategy to Robustify the Meta Learning Paradigm

NeurIPS 2023poster

Meta learning is a promising paradigm to enable skill transfer across tasks. Most previous methods employ the empirical risk minimization principle in optimization. However, the resulting worst fast adaptation to a subset of tasks can be catastrophic in risk-sensitive scenarios. To robustify fast ad…

Cited by 11SourcePDFScholar
2023

Episodic Multi-Task Learning with Heterogeneous Neural Processes

NeurIPS 2023spotlight

This paper focuses on the data-insufficiency problem in multi-task learning within an episodic training setup. Specifically, we explore the potential of heterogeneous information across tasks and meta-knowledge among episodes to effectively tackle each task with limited data. Existing meta-learning…