← Search

Wenhao yang

29 accepted papers

2026

Convergence Analysis of the Lion Optimizer in Centralized and Distributed Settings

ICML 2026poster

In this paper, we provide a comprehensive convergence analysis for the Lion optimizer. First, we establish that the original Lion achieves a convergence rate of $\mathcal{O}(d^{1/2}T^{-1/4})$, where $d$ denotes the problem dimension and $T$ is the iteration number. To improve this rate, we propose a…

Cited by 0SourceScholar
2026

Decentralized Online Convex Optimization with Efficient Communication: Improved Algorithm and Lower Bounds

ICML 2026poster

We investigate decentralized online convex optimization with compressed communication, where $n$ learners connected by a network collaboratively minimize a sequence of global loss functions using only local information and compressed data from neighbors. Prior work has established regret bounds of $…

Cited by 0SourceScholar
2025

All You Need is One: Capsule Prompt Tuning with a Single Vector

NeurIPS 2025poster

Prompt-based learning has emerged as a parameter-efficient finetuning (PEFT) approach to facilitate Large Language Model (LLM) adaptation to downstream tasks by conditioning generation with task-aware guidance. Despite its successes, current prompt-based learning methods heavily rely on laborious gr…

Cited by 0SourceScholar
2025

Exploring the Adversarial Vulnerabilities of Vision-Language-Action Models in Robotics

ICCV 2025poster

Recently in robotics, Vision-Language-Action (VLA) models have emerged as a transformative approach, enabling robots to execute complex tasks by integrating visual and linguistic inputs within an end-to-end learning framework. Despite their significant capabilities, VLA models introduce new attack s…

2025

Neuron based Personality Trait Induction in Large Language Models

ICLR 2025poster

Large language models (LLMs) have become increasingly proficient at simulating various personality traits, an important capability for supporting related applications (e.g., role-playing). To further improve this capacity, in this paper, we present a neuron based approach for personality trait induc…

2025

Revisiting Differentially Private Algorithms for Decentralized Online Learning

ICML 2025poster

Although the differential privacy (DP) of decentralized online learning has garnered considerable attention recently, existing algorithms are unsatisfactory due to their inability to achieve $(\epsilon, 0)$-DP over all $T$ rounds, recover the optimal regret in the non-private case, and maintain the…

Cited by 0SourcePDFScholar
2025

Towards Unbiased Information Extraction and Adaptation in Cross-Domain Recommendation

AAAI 2025technical

Cross-Domain Recommendation (CDR) leverages additional knowledge from auxiliary domains to address the long-standing data sparsity issue. However, existing methods typically acquire this knowledge by minimizing the average loss over all domains, overlooking the fact that different domains possess di…

Cited by 0SourcePDFScholar
2024

Distributionally Robust Optimization as a Scalable Framework to Characterize Extreme Value Distributions

UAI 2024poster

The goal of this paper is to develop distributionally robust optimization (DRO) estimators, specifically for multidimensional Extreme Value Theory (EVT) statistics. EVT supports using semi-parametric models called max-stable distributions built from spatial Poisson point processes. While powerful, t…

Cited by 2SourcePDFScholar
2024

Efficient Sign-Based Optimization: Accelerating Convergence via Variance Reduction

NeurIPS 2024poster

Sign stochastic gradient descent (signSGD) is a communication-efficient method that transmits only the sign of stochastic gradients for parameter updating. Existing literature has demonstrated that signSGD can achieve a convergence rate of $\mathcal{O}(d^{1/2}T^{-1/4})$, where $d$ represents the dim…

Cited by 4SourcePDFScholar
2024

Non-stationary Projection-Free Online Learning with Dynamic and Adaptive Regret Guarantees

AAAI 2024technical

Projection-free online learning has drawn increasing interest due to its efficiency in solving high-dimensional problems with complicated constraints. However, most existing projection-free online methods focus on minimizing the static regret, which unfortunately fails to capture the challenge of ch…

Cited by 12SourcePDFScholar
2024

Online Composite Optimization Between Stochastic and Adversarial Environments

NeurIPS 2024poster

We study online composite optimization under the Stochastically Extended Adversarial (SEA) model. Specifically, each loss function consists of two parts: a fixed non-smooth and convex regularizer, and a time-varying function which can be chosen either stochastically, adversarially, or in a manner th…

Cited by 4SourcePDFScholar
2024

Projection-Free Variance Reduction Methods for Stochastic Constrained Multi-Level Compositional Optimization

ICML 2024poster

This paper investigates projection-free algorithms for stochastic constrained multi-level optimization. In this context, the objective function is a nested composition of several smooth functions, and the decision set is closed and convex. Existing projection-free algorithms for solving this problem…

Cited by 2SourcePDFScholar
2024

Small-loss Adaptive Regret for Online Convex Optimization

ICML 2024poster

To deal with changing environments, adaptive regret has been proposed to minimize the regret over every interval. Previous studies have established a small-loss adaptive regret bound for general convex functions under the smoothness condition, offering the advantage of being much tighter than minima…

Cited by 3SourcePDFScholar
2024

Universal Online Convex Optimization with $1$ Projection per Round

NeurIPS 2024poster

To address the uncertainty in function types, recent progress in online convex optimization (OCO) has spurred the development of universal algorithms that simultaneously attain minimax rates for multiple types of convex functions. However, for a $T$-round online problem, state-of-the-art methods typ…

Cited by 4SourcePDFScholar
2023

A Statistical Analysis of Polyak-Ruppert Averaged Q-Learning

AISTATS 2023poster

We study Q-learning with Polyak-Ruppert averaging (a.k.a., averaged Q-learning) in a discounted markov decision process in synchronous and tabular settings. Under a Lipschitz condition, we establish a functional central limit theorem for the averaged iteration $\bar{\mathbf{Q}}_T$ and show that its…

2023

Regularization and Variance-Weighted Regression Achieves Minimax Optimality in Linear MDPs: Theory and Practice

ICML 2023poster

Mirror descent value iteration (MDVI), an abstraction of Kullback-Leibler (KL) and entropy-regularized reinforcement learning (RL), has served as the basis for recent high-performing practical RL algorithms. However, despite the use of function approximation in practice, the theoretical understandin…

2023

Semiparametrically Efficient Off-Policy Evaluation in Linear Markov Decision Processes

ICML 2023poster

We study semiparametrically efficient estimation in off-policy evaluation (OPE) where the underlying Markov decision process (MDP) is linear with a known feature map. We characterize the variance lower bound for regular estimators in the linear MDP setting and propose an efficient estimator whose va…

Cited by 6SourcePDFScholar
2022

Federated Reinforcement Learning with Environment Heterogeneity

AISTATS 2022poster

We study Federated Reinforcement Learning (FedRL) problem in which $n$ agents collaboratively learn a single policy without sharing the trajectories they collected during agent-environment interaction. In this paper, we stress the constraint of environment heterogeneity, which means $n$ environments…

2022

Pluralistic Image Completion with Gaussian Mixture Models

NeurIPS 2022accept

Pluralistic image completion focuses on generating both visually realistic and diverse results for image completion. Prior methods enjoy the empirical successes of this task. However, their used constraints for pluralistic image completion are argued to be not well interpretable and unsatisfactory f…