← Search

John C.S. Lui

30 accepted papers

2026

A Multi-Agent Conversational Bandit Approach to Online Evaluation and Selection of User-Aligned LLM Responses

AAAI 2026technical

Prompt-based offline methods are commonly used to optimize large language model (LLM) responses, but evaluating these responses is computationally intensive and often fails to accommodate diverse response styles. This study introduces a novel online evaluation framework that employs a multi-agent co

Cited by 0SourcePDFScholar
2026

Online Multi-LLM Selection via Contextual Bandits Under Unstructured Context Evolution

AAAI 2026technical

Large language models (LLMs) exhibit diverse response behaviors, costs, and strengths, making it challenging to select the most suitable LLM for a given user query. We study the problem of adaptive multi-LLM selection in an online setting, where the learner interacts with users through multi-step qu

Cited by 0SourcePDFScholar
2025

Bandit Learning in Matching Markets with Indifference

ICLR 2025poster

A rich line of recent works studies how participants in matching markets learn their unknown preferences through iterative interactions with each other. The two sides of participants in the market can be respectively formulated as players and arms in the bandit problem. To ensure market stability, t…

Cited by 0SourcePDFScholar
2025

Demystifying Online Clustering of Bandits: Enhanced Exploration Under Stochastic and Smoothed Adversarial Contexts

ICLR 2025poster

The contextual multi-armed bandit (MAB) problem is crucial in sequential decision-making. A line of research, known as online clustering of bandits, extends contextual MAB by grouping similar users into clusters, utilizing shared features to improve learning efficiency. However, existing algorithms,…

Cited by 1SourcePDFScholar
2025

Federated In-Context Learning: Iterative Refinement for Improved Answer Quality

ICML 2025poster

For question-answering (QA) tasks, in-context learning (ICL) enables language models (LMs) to generate responses without modifying their parameters by leveraging examples provided in the input. However, the effectiveness of ICL heavily depends on the availability of high-quality examples, which are…

Cited by 0SourcePDFScholar
2025

Fusing Reward and Dueling Feedback in Stochastic Bandits

ICML 2025poster

This paper investigates the fusion of absolute (reward) and relative (dueling) feedback in stochastic bandits, where both feedback types are gathered in each decision round. We derive a regret lower bound, demonstrating that an efficient algorithm may incur only the smaller among the reward…

Cited by 0SourcePDFScholar
2025

Model-based RL as a Minimalist Approach to Horizon-Free and Second-Order Bounds

ICLR 2025poster

Learning a transition model via Maximum Likelihood Estimation (MLE) followed by planning inside the learned model is perhaps the most standard and simplest Model-based Reinforcement Learning (RL) framework. In this work, we show that such a simple Model-based RL scheme, when equipped with optimistic…

Cited by 5SourcePDFScholar
2025

Offline Learning for Combinatorial Multi-armed Bandits

ICML 2025poster

The combinatorial multi-armed bandit (CMAB) is a fundamental sequential decision-making framework, extensively studied over the past decade. However, existing work primarily focuses on the online setting, overlooking the substantial costs of online interactions and the readily available offline data…

Cited by 1SourcePDFScholar
2025

Online Clustering of Dueling Bandits

ICML 2025poster

The contextual multi-armed bandit (MAB) is a widely used framework for problems requiring sequential decision-making under uncertainty, such as recommendation systems. In applications involving a large number of users, the performance of contextual MAB can be significantly improved by facilitating c…

Cited by 0SourcePDFScholar
2025

Provable Zero-Shot Generalization in Offline Reinforcement Learning

ICML 2025poster

In this work, we study offline reinforcement learning (RL) with zero-shot generalization property (ZSG), where the agent has access to an offline dataset including experiences from different environments, and the goal of the agent is to train a policy over the training environments which performs we…

Cited by 0SourcePDFScholar
2025

Quantum Algorithms for Finite-horizon Markov Decision Processes

ICML 2025poster

In this work, we design quantum algorithms that are more efficient than classical algorithms to solve time-dependent and finite-horizon Markov Decision Processes (MDPs) in two distinct settings: (1) In the exact dynamics setting, where the agent has full knowledge of the environment's dynamics (i.e.…

Cited by 0SourcePDFScholar
2025

Quantum Best Arm Identification with Quantum Oracles

AAAI 2025technical

Best arm identification (BAI) is a key problem in stochastic multi-armed bandits, where K arms each has an associated reward distribution, and the objective is to minimize the number of queries needed to identify the best arm with high confidence. In this paper, we explore BAI using quantum oracles.…

Cited by 0SourcePDFScholar
2025

Quantum Speedups for Minimax Optimization and Beyond

NeurIPS 2025poster

This paper investigates convex-concave minimax optimization problems where only the function value access is allowed. We introduce a class of Hessian-aware quantum zeroth-order methods that can find the $\epsilon$-saddle point within $\tilde{\mathcal{O}}(d^{2/3}\epsilon^{-2/3})$ function value oracl…

Cited by 0SourceScholar
2025

Stochastic Bandits Robust to Adversarial Attacks

ICLR 2025poster

This paper investigates stochastic multi-armed bandit algorithms that are robust to adversarial attacks, where an attacker can first observe the learner's action and *then* alter their reward observation. We study two cases of this model, with or without the knowledge of an attack budget $C$, define…

Cited by 0SourcePDFScholar
2025

Variance-Dependent Regret Bounds for Nonstationary Linear Bandits

AISTATS 2025poster

We investigate the non-stationary stochastic linear bandit problem where the reward distribution evolves each round. Existing algorithms characterize the non-stationarity by the total variation budget $B_K$, which is the summation of the change of the consecutive feature vectors of the linear bandit…

Cited by 0SourceScholar
2024

Combinatorial Multivariant Multi-Armed Bandits with Applications to Episodic Reinforcement Learning and Beyond

ICML 2024poster

We introduce a novel framework of combinatorial multi-armed bandits (CMAB) with multivariant and probabilistically triggering arms (CMAB-MT), where the outcome of each arm is a $d$-dimensional multivariant random variable and the feedback follows a general arm triggering process. Compared with exist…

Cited by 4SourcePDFScholar
2024

D-LLM: A Token Adaptive Computing Resource Allocation Strategy for Large Language Models

NeurIPS 2024poster

Large language models have shown an impressive societal impact owing to their excellent understanding and logical reasoning skills. However, such strong ability relies on a huge amount of computing resources, which makes it difficult to deploy LLMs on computing resource-constrained platforms. Curren…

Cited by 3SourcePDFScholar
2024

Quantum Algorithm for Online Exp-concave Optimization

ICML 2024poster

We explore whether quantum advantages can be found for the zeroth-order feedback online exp-concave optimization problem, which is also known as bandit exp-concave optimization with multi-point feedback. We present quantum online quasi-Newton methods to tackle the problem and show that there exists…

Cited by 1SourcePDFScholar
2024

Quantum Algorithms for Non-smooth Non-convex Optimization

NeurIPS 2024poster

This paper considers the problem for finding the $(\delta,\epsilon)$-Goldstein stationary point of Lipschitz continuous objective, which is a rich function class to cover a great number of important applications. We construct a novel zeroth-order quantum estimator for the gradient of the smoothed…

Cited by 5SourcePDFScholar
2023

Achieving Near-Optimal Individual Regret & Low Communications in Multi-Agent Bandits

ICLR 2023poster

Cooperative multi-agent multi-armed bandits (CM2AB) study how distributed agents cooperatively play the same multi-armed bandit game. Most existing CM2AB works focused on maximizing the group performance of all agents---the accumulation of all agents' individual performance (i.e., individual reward)…

Cited by 13SourcePDFScholar
2023

Contextual Combinatorial Bandits with Probabilistically Triggered Arms

ICML 2023poster

We study contextual combinatorial bandits with probabilistically triggered arms (C$^2$MAB-T) under a variety of smoothness conditions that capture a wide range of applications, such as contextual cascading bandits and contextual influence maximization bandits. Under the triggering probability modula…

Cited by 21SourcePDFScholar
2023

Exploration for Free: How Does Reward Heterogeneity Improve Regret in Cooperative Multi-agent Bandits?

UAI 2023poster

This paper studies a cooperative multi-agent bandit scenario in which the rewards observed by agents are heterogeneous—one agent’s meat can be another agent’s poison. Specifically, the total reward observed by each agent is the sum of two values: an arm-specific reward, capturing the intrinsic value…

Cited by 2SourcePDFScholar
2023

Online Clustering of Bandits with Misspecified User Models

NeurIPS 2023poster

The contextual linear bandit is an important online learning problem where given arm features, a learning agent selects an arm at each round to maximize the cumulative rewards in the long run. A line of works, called the clustering of bandits (CB), utilize the collaborative effect over user preferen…

Cited by 13SourcePDFScholar
2023

Online Corrupted User Detection and Regret Minimization

NeurIPS 2023poster

In real-world online web systems, multiple users usually arrive sequentially into the system. For applications like click fraud and fake reviews, some users can maliciously perform corrupted (disrupted) behaviors to trick the system. Therefore, it is crucial to design efficient online learning algor…

Cited by 9SourcePDFScholar
2023

Uncertainty-Aware Instance Reweighting for Off-Policy Learning

NeurIPS 2023poster

Off-policy learning, referring to the procedure of policy optimization with access only to logged feedback data, has shown importance in various important real-world applications, such as search engines and recommender systems. While the ground-truth logging policy is usually unknown, previous work…

Cited by 7SourcePDFScholar