← Search

Hoi To Wai

13 accepted papers

2025

Clipped SGD Algorithms for Performative Prediction: Tight Bounds for Stochastic Bias and Remedies

ICML 2025poster

This paper studies the convergence of clipped stochastic gradient descent (SGD) algorithms with decision-dependent data distribution. Our setting is motivated by privacy preserving optimization algorithms that interact with performative data where the prediction models can influence future outcomes.…

Cited by 0SourcePDFScholar
2025

RoSTE: An Efficient Quantization-Aware Supervised Fine-Tuning Approach for Large Language Models

ICML 2025poster

Supervised fine-tuning is a standard method for adapting pre-trained large language models (LLMs) to downstream tasks. Quantization has been recently studied as a post-training technique for efficient LLM deployment. To obtain quantized fine-tuned LLMs, conventional pipelines would first fine-tune t…

2024

EMC$^2$: Efficient MCMC Negative Sampling for Contrastive Learning with Global Convergence

ICML 2024poster

A key challenge in contrastive learning is to generate negative samples from a large sample set to contrast with positive samples, for learning better encoding of the data. These negative samples often follow a softmax distribution which are dynamically updated during the training process. However,…

2024

Getting More Juice Out of the SFT Data: Reward Learning from Human Demonstration Improves SFT for LLM Alignment

NeurIPS 2024poster

Aligning human preference and value is an important requirement for contemporary foundation models. State-of-the-art techniques such as Reinforcement Learning from Human Feedback (RLHF) often consist of two stages: 1) supervised fine-tuning (SFT), where the model is fine-tuned by learning from human…

2024

Two-timescale Derivative Free Optimization for Performative Prediction with Markovian Data

ICML 2024poster

This paper studies the performative prediction problem where a learner aims to minimize the expected loss with a decision-dependent data distribution. Such setting is motivated when outcomes can be affected by the prediction model, e.g., in strategic classification. We consider a state-dependent set…

Cited by 3SourcePDFScholar
2022

Decentralized Learning for Overparameterized Problems: A Multi-Agent Kernel Approximation Approach

ICLR 2022poster

This work develops a novel framework for communication-efficient distributed learning where the models to be learned are overparameterized. We focus on a class of kernel learning problems (which includes the popular neural tangent kernel (NTK) learning as a special case) and propose a novel {\it mul…

Cited by 0SourcePDFScholar
2022

Distributed Optimization for Overparameterized Problems: Achieving Optimal Dimension Independent Communication Complexity

NeurIPS 2022accept

Decentralized optimization are playing an important role in applications such as training large machine learning models, among others. Despite its superior practical performance, there has been some lack of fundamental understanding about its theoretical properties. In this work, we address the foll…

Cited by 5SourcePDFScholar
2022

Inducing Equilibria via Incentives: Simultaneous Design-and-Play Ensures Global Convergence

NeurIPS 2022accept

To regulate a social system comprised of self-interested agents, economic incentives are often required to induce a desirable outcome. This incentive design problem naturally possesses a bilevel structure, in which a designer modifies the payoffs of the agents with incentives while anticipating the…

Cited by 19SourcePDFScholar
2022

Multi-agent Performative Prediction with Greedy Deployment and Consensus Seeking Agents

NeurIPS 2022accept

We consider a scenario where multiple agents are learning a common decision vector from data which can be influenced by the agents’ decisions. This leads to the problem of multi-agent performative prediction (Multi-PfD). In this paper, we formulate Multi-PfD as a decentralized optimization problem t…

Cited by 27SourcePDFScholar
2021

A Near-Optimal Algorithm for Stochastic Bilevel Optimization via Double-Momentum

NeurIPS 2021poster

This paper proposes a new algorithm -- the \underline{S}ingle-timescale Do\underline{u}ble-momentum \underline{St}ochastic \underline{A}pprox\underline{i}matio\underline{n} (SUSTAIN) -- for tackling stochastic unconstrained bilevel optimization problems. We focus on bilevel problems where the lower…

Cited by 147SourcePDFScholar
2021

Tight High Probability Bounds for Linear Stochastic Approximation with Fixed Stepsize

NeurIPS 2021poster

This paper provides a non-asymptotic analysis of linear stochastic approximation (LSA) algorithms with fixed stepsize. This family of methods arises in many machine learning tasks and is used to obtain approximate solutions of a linear system $\bar{A}\theta = \bar{b}$ for which $\bar{A}$ and $\bar{b…

Cited by 30SourcePDFScholar