← Search

Zhongxiang Dai

43 accepted papers

2026

ALSO: Adversarial Online Strategy Optimization for Social Agents

ICML 2026poster

Social simulation provides a compelling testbed for studying social intelligence, where agents interact through multi-turn dialogues under evolving contexts and strategically adapting opponents. Such environments are inherently non-stationary, requiring agents to dynamically adjust their strategies …

Cited by 0SourceScholar
2026

ActiveDPO: Active Direct Preference Optimization for Sample-Efficient Alignment

ICLR 2026poster

The recent success in using human preferences to align large language models (LLMs) has significantly improved their performance in various downstream tasks, such as question answering, mathematical reasoning, and code generation. However, achieving effective LLM alignment depends on high-quality hu…

Cited by 0SourceScholar
2026

DRIFT: Decoupled Rollouts and Importance-Weighted Fine-Tuning for Efficient Multi-Turn Optimization

ICML 2026poster

Large language models are increasingly deployed in multi-turn interactive settings where users or environments can iteratively provide lightweight feedback. Unfortunately, optimizing such behavior presents a sharp dilemma in practice: online reinforcement learning is able to effectively address mult…

Cited by 0SourceScholar
2026

Dual-Robust Cross-Domain Offline Reinforcement Learning Against Dynamics Shifts

ICLR 2026poster

Single-domain offline reinforcement learning (RL) often suffers from limited data coverage, while cross-domain offline RL handles this issue by leveraging additional data from other domains with dynamics shifts. However, existing studies primarily focus on train-time robustness (handling dynamics sh…

Cited by 0SourceScholar
2026

MASPOB: Bandit-Based Prompt Optimization for Multi-Agent Systems with Graph Neural Networks

ICML 2026spotlight

Large Language Models (LLMs) have achieved significant success across a wide range of tasks, serving as the cognitive backbone for Multi-Agent Systems (MAS) designed to orchestrate complex practical workflows. Given that MAS performance is highly sensitive to input prompts and many deployment scenar…

Cited by 0SourceScholar
2026

Real-Time Aligned Reward Model beyond Semantics

ICML 2026poster

Reinforcement Learning from Human Feedback (RLHF) is a pivotal technique for aligning large language models (LLMs) with human preferences, yet it is susceptible to reward overoptimization, in which policy models overfit to the reward model, exploit spurious reward patterns instead of faithfully capt…

Cited by 0SourceScholar
2026

T-POP: Test-Time Personalization with Online Preference Feedback

ICML 2026poster

Personalizing large language models (LLMs) to individual user preferences is a critical step beyond generating generically helpful responses. However, current personalization methods are ill-suited for new users, as they typically require either slow, resource-intensive fine-tuning or a substantial …

Cited by 0SourceScholar
2026

UniScale: Adaptive Unified Inference Scaling via Online Joint Optimization of Model Routing and Test-Time Scaling

ICML 2026poster

In real-world deployments of large language models (LLMs), balancing inference quality and computational cost has become a central challenge. Existing approaches tackle this trade-off along two largely independent dimensions: model routing, which switches among models of different scales to match re…

Cited by 0SourceScholar
2026

Words & Weights: Streamlining Multi-Turn Interactions via Co-Adaptation

ICML 2026poster

Test-time policy adaptation for multi-turn interactions (T$^2$PAM) is essential for aligning Large Language Models (LLMs) with dynamic user needs during inference time. However, existing paradigms commonly treat test-time adaptation as a single-axis problem, either purely refining instructions (Prom…

Cited by 0SourceScholar
2025

Adaptive Batch-Wise Sample Scheduling for Direct Preference Optimization

NeurIPS 2025poster

Direct Preference Optimization (DPO) has emerged as an effective approach for aligning large language models (LLMs) with human preferences. However, its performance is highly dependent on the quality of the underlying human preference data. To address this bottleneck, prior work has explored various…

Cited by 0SourceScholar
2025

Convergence Rates of Constrained Expected Improvement

NeurIPS 2025spotlight

Constrained Bayesian optimization (CBO) methods have seen significant success in black-box optimization with constraints. One of the most commonly used CBO methods is the constrained expected improvement (CEI) algorithm. CEI is a natural extension of expected improvement (EI) when constraints are i…

Cited by 0SourceScholar
2025

Neural Dueling Bandits: Preference-Based Optimization with Human Feedback

ICLR 2025poster

Contextual dueling bandit is used to model the bandit problems, where a learner's goal is to find the best arm for a given context using observed noisy human preference feedback over the selected arms for the past contexts. However, existing algorithms assume the reward function is linear, which can…

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

WASA: WAtermark-based Source Attribution for Large Language Model-Generated Data

ACL 2025finding

The impressive performances of Large Language Models (LLMs) and their immense potential for commercialization have given rise to serious concerns over the Intellectual Property (IP) of their training data. In particular, the synthetic texts generated by LLMs may infringe the IP of the data being use…

Cited by 19SourcePDFScholar
2024

Localized Zeroth-Order Prompt Optimization

NeurIPS 2024spotlight

The efficacy of large language models (LLMs) in understanding and generating natural language has aroused a wide interest in developing prompt-based methods to harness the power of black-box LLMs. Existing methodologies usually prioritize a global optimization for finding the global optimum, which h…

Cited by 14SourcePDFScholar
2024

Position Paper: Data-Centric AI in the Age of Large Language Models

EMNLP 2024finding

This position paper proposes a data-centric viewpoint of AI research, focusing on large language models (LLMs). We start by making a key observation that data is instrumental in the developmental (e.g., pretraining and fine-tuning) and inferential stages (e.g., in-context learning) of LLMs, and advo…

Cited by 1SourcePDFScholar
2024

Prompt Optimization with EASE? Efficient Ordering-aware Automated Selection of Exemplars

NeurIPS 2024poster

Large language models (LLMs) have shown impressive capabilities in real-world applications. The capability of *in-context learning* (ICL) allows us to adapt an LLM to downstream tasks by including input-label exemplars in the prompt without model fine-tuning. However, the quality of these exemplars…

2024

Robustifying and Boosting Training-Free Neural Architecture Search

ICLR 2024poster

Neural architecture search (NAS) has become a key component of AutoML and a standard tool to automate the design of deep neural networks. Recently, training-free NAS as an emerging paradigm has successfully reduced the search costs of standard training-based NAS by estimating the true architecture p…

2024

Use Your INSTINCT: INSTruction optimization for LLMs usIng Neural bandits Coupled with Transformers

ICML 2024poster

Large language models (LLMs) have shown remarkable instruction-following capabilities and achieved impressive performances in various applications. However, the performances of LLMs depend heavily on the instructions given to them, which are typically manually tuned with substantial human efforts. R…

2023

Batch Bayesian Optimization For Replicable Experimental Design

NeurIPS 2023poster

Many real-world experimental design problems (a) evaluate multiple experimental conditions in parallel and (b) replicate each condition multiple times due to large and heteroscedastic observation noise. Given a fixed total budget, this naturally induces a trade-off between evaluating more unique con…

Cited by 6SourcePDFScholar
2023

Exploiting Correlated Auxiliary Feedback in Parameterized Bandits

NeurIPS 2023poster

We study a novel variant of the parameterized bandits problem in which the learner can observe additional auxiliary feedback that is correlated with the observed reward. The auxiliary feedback is readily available in many real-life applications, e.g., an online platform that wants to recommend the b…

Cited by 0SourcePDFScholar
2023

Federated Neural Bandits

ICLR 2023poster

Recent works on neural contextual bandits have achieved compelling performances due to their ability to leverage the strong representation power of neural networks (NNs) for reward prediction. Many applications of contextual bandits involve multiple agents who collaborate without sharing raw observa…

2023

Quantum Bayesian Optimization

NeurIPS 2023poster

Kernelized bandits, also known as Bayesian optimization (BO), has been a prevalent method for optimizing complicated black-box reward functions. Various BO algorithms have been theoretically shown to enjoy upper bounds on their cumulative regret which are sub-linear in the number $T$ of iterations,…

Cited by 18SourcePDFScholar
2023

Training-Free Neural Active Learning with Initialization-Robustness Guarantees

ICML 2023poster

Existing neural active learning algorithms have aimed to optimize the predictive performance of neural networks (NNs) by selecting data for labelling. However, other than a good predictive performance, being robust against random parameter initializations is also a crucial requirement in safety-crit…

2023

Zeroth-Order Optimization with Trajectory-Informed Derivative Estimation

ICLR 2023poster

Zeroth-order (ZO) optimization, in which the derivative is unavailable, has recently succeeded in many important machine learning applications. Existing algorithms rely on finite difference (FD) methods for derivative estimation and gradient descent (GD)-based approaches for optimization. However, t…

Cited by 15SourcePDFScholar
2022

Bayesian Optimization under Stochastic Delayed Feedback

ICML 2022spotlight

Bayesian optimization (BO) is a widely-used sequential method for zeroth-order optimization of complex and expensive-to-compute black-box functions. The existing BO methods assume that the function evaluation (feedback) is available to the learner immediately or after a fixed delay. Such assumptions…

2022

NASI: Label- and Data-agnostic Neural Architecture Search at Initialization

ICLR 2022poster

Recent years have witnessed a surging interest in Neural Architecture Search (NAS). Various algorithms have been proposed to improve the search efficiency and effectiveness of NAS, i.e., to reduce the search cost and improve the generalization performance of the selected architectures, respectively.…

Cited by 55SourcePDFScholar
2022

On provably robust meta-Bayesian optimization

UAI 2022poster

Bayesian optimization (BO) has become popular for sequential optimization of black-box functions. When BO is used to optimize a target function, we often have access to previous evaluations of potentially related functions. This begs the question as to whether we can leverage these previous experien…

2022

Sample-Then-Optimize Batch Neural Thompson Sampling

NeurIPS 2022accept

Bayesian optimization (BO), which uses a Gaussian process (GP) as a surrogate to model its objective function, is popular for black-box optimization. However, due to the limitations of GPs, BO underperforms in some problems such as those with categorical, high-dimensional or image inputs. To this en…

2022

Unifying and Boosting Gradient-Based Training-Free Neural Architecture Search

NeurIPS 2022accept

Neural architecture search (NAS) has gained immense popularity owing to its ability to automate neural architecture design. A number of training-free metrics are recently proposed to realize NAS without training, hence making NAS more scalable. Despite their competitive empirical performances, a uni…

2021

Differentially Private Federated Bayesian Optimization with Distributed Exploration

NeurIPS 2021poster

Bayesian optimization (BO) has recently been extended to the federated learning (FL) setting by the federated Thompson sampling (FTS) algorithm, which has promising applications such as federated hyperparameter tuning. However, FTS is not equipped with a rigorous privacy guarantee which is an import…

Cited by 59SourcePDFScholar
2021

Fault-Tolerant Federated Reinforcement Learning with Theoretical Guarantee

NeurIPS 2021poster

The growing literature of Federated Learning (FL) has recently inspired Federated Reinforcement Learning (FRL) to encourage multiple agents to federatively build a better decision-making policy without sharing raw trajectories. Despite its promising applications, existing works on FRL fail to I) pro…

2021

Optimizing Conditional Value-At-Risk of Black-Box Functions

NeurIPS 2021poster

This paper presents two Bayesian optimization (BO) algorithms with theoretical performance guarantee to maximize the conditional value-at-risk (CVaR) of a black-box function: CV-UCB and CV-TS which are based on the well-established principle of optimism in the face of uncertainty and Thompson sampli…

2021

Value-at-Risk Optimization with Gaussian Processes

ICML 2021spotlight

Value-at-risk (VaR) is an established measure to assess risks in critical real-world applications with random environmental factors. This paper presents a novel VaR upper confidence bound (V-UCB) algorithm for maximizing the VaR of a black-box objective function with the first no-regret guarantee. T…

2020

Federated Bayesian Optimization via Thompson Sampling

NeurIPS 2020poster

Bayesian optimization (BO) is a prominent approach to optimizing expensive-to-evaluate black-box functions. The massive computational capability of edge devices such as mobile phones, coupled with privacy concerns, has led to a surging interest in federated learning (FL) which focuses on collaborati…

2020

R2-B2: Recursive Reasoning-Based Bayesian Optimization for No-Regret Learning in Games

ICML 2020poster

This paper presents a recursive reasoning formalism of Bayesian optimization (BO) to model the reasoning process in the interactions between boundedly rational, self-interested agents with unknown, complex, and costly-to-evaluate payoff functions in repeated games, which we call Recursive Reasoning-…

Cited by 37SourcePDFScholar
2019

Bayesian Optimization Meets Bayesian Optimal Stopping

ICML 2019oral

Bayesian optimization (BO) is a popular paradigm for optimizing the hyperparameters of machine learning (ML) models due to its sample efficiency. Many ML models require running an iterative training procedure (e.g., stochastic gradient descent). This motivates the question whether information availa…

Cited by 71SourcePDFScholar
2019

Implicit Posterior Variational Inference for Deep Gaussian Processes

NeurIPS 2019spotlight

A multi-layer deep Gaussian process (DGP) model is a hierarchical composition of GP models with a greater expressive power. Exact DGP inference is intractable, which has motivated the recent development of deterministic and stochastic approximation methods. Unfortunately, the deterministic approxima…