← Search

Lifeng Lai

28 accepted papers

2026

Differentially Private Two-Stage Gradient Descent for Instrumental Variable Regression

ICLR 2026poster

We study instrumental variable regression (IVaR) under differential privacy constraints. Classical IVaR methods (like two-stage least squares regression) rely on solving moment equations that directly use sensitive covariates and instruments, creating significant risks of privacy leakage and posing…

Cited by 0SourceScholar
2026

Learn to change the world: Multi-level reinforcement learning with model-changing actions

ICML 2026poster

Reinforcement learning usually assumes a given or sometimes even fixed environment in which an agent seeks an optimal policy to maximize its long-term discounted reward. In contrast, we consider agents that are not limited to passive adaptations: they instead have model-changing actions that activel…

Cited by 0SourceScholar
2025

Absorb and Converge: Provable Convergence Guarantee for Absorbing Discrete Diffusion Models

NeurIPS 2025poster

Discrete state space diffusion models have shown significant advantages in applications involving discrete data, such as text and image generation. It has also been observed that their performance is highly sensitive to the choice of rate matrices, particularly between uniform and absorbing rate mat…

Cited by 0SourceScholar
2025

Discrete Diffusion Models: Novel Analysis and New Sampler Guarantees

NeurIPS 2025poster

Discrete diffusion models have recently gained significant prominence in applications involving natural language and graph data. A key factor influencing their effectiveness is the efficiency of discretized samplers. Among these, $\tau$-leaping samplers have become particularly popular due to their…

Cited by 0SourceScholar
2025

Transformers Handle Endogeneity in In-Context Linear Regression

ICLR 2025poster

We explore the capability of transformers to address endogeneity in in-context linear regression. Our main finding is that transformers inherently possess a mechanism to handle endogeneity effectively using instrumental variables (IV). First, we demonstrate that the transformer architecture can emul…

Cited by 2SourcePDFScholar
2024

A Huber Loss Minimization Approach to Mean Estimation under User-level Differential Privacy

NeurIPS 2024poster

Privacy protection of users' entire contribution of samples is important in distributed systems. The most effective approach is the two-stage scheme, which finds a small interval first and then gets a refined estimate by clipping samples into the interval. However, the clipping operation induces bia…

Cited by 7SourcePDFScholar
2024

Tree Network Design for Faster Distributed Machine Learning Process with Distributed Dual Coordinate Ascent

ICASSP 2024accepted

This paper delves into the subject of designing a tree network, enabling the application of Distributed Dual Coordinate Ascent on a general tree network (DDCA-Tree) introduced in [1] – [3] for distributed Machine Learning (ML) process. We assume that a network is characterized by communication delay…

Cited by 0SourceScholar
2021

A Riemannian Block Coordinate Descent Method for Computing the Projection Robust Wasserstein Distance

ICML 2021spotlight

The Wasserstein distance has become increasingly important in machine learning and deep learning. Despite its popularity, the Wasserstein distance is hard to approximate because of the curse of dimensionality. A recently proposed approach to alleviate the curse of dimensionality is to project the sa…

Cited by 52SourcePDFScholar
2019

Generalized Distributed Dual Coordinate Ascent in a Tree Network for Machine Learning

ICASSP 2019accepted

With explosion of data size and limited storage space at a single location, data are often distributed at different locations. We thus face the challenge of performing large-scale machine learning from these distributed data through communication networks. In this paper, we generalize the distribute…

Cited by 0SourceScholar
2017

Rate-distortion trade-offs in acquisition of signal parameters

ICASSP 2017accepted

We consider problems where one wishes to represent a parameter associated with a signal source - subject to a certain rate and distortion - based on the observation of a number of realizations of the source signal. By reducing these indirect vector quantization problems to a standard vector quantiza…

Cited by 0SourceScholar