← Search

Michael R. Zhang

6 accepted papers

2025

Fast Exact Unlearning for In-Context Learning Data for LLMs

ICML 2025poster

Modern machine learning models are expensive to train, and there is a growing concern about the challenge of retroactively removing specific training data. Achieving exact unlearning in deep learning pipelines—producing models as if certain data had never been included in training—remains an open pr…

Cited by 0SourcePDFScholar
2023

Multi-Rate VAE: Train Once, Get the Full Rate-Distortion Curve

ICLR 2023top-5%

Variational autoencoders (VAEs) are powerful tools for learning latent representations of data used in a wide range of applications. In practice, VAEs usually require multiple training rounds to choose the amount of information the latent variable should retain. This trade-off between the reconstruc…

Cited by 21SourcePDFScholar
2021

Autoregressive Dynamics Models for Offline Policy Evaluation and Optimization

ICLR 2021poster

Standard dynamics models for continuous control make use of feedforward computation to predict the conditional distribution of next state and reward given current state and action using a multivariate Gaussian with a diagonal covariance structure. This modeling choice assumes that different dimensio…

Cited by 53SourcePDFScholar
2021

Benchmarks for Deep Off-Policy Evaluation

ICLR 2021poster

Off-policy evaluation (OPE) holds the promise of being able to leverage large, offline datasets for both evaluating and selecting complex policies for decision making. The ability to learn offline is particularly important in many real-world domains, such as in healthcare, recommender systems, or ro…

2021

Learning Domain Invariant Representations in Goal-conditioned Block MDPs

NeurIPS 2021poster

Deep Reinforcement Learning (RL) is successful in solving many complex Markov Decision Processes (MDPs) problems. However, agents often face unanticipated environmental changes after deployment in the real world. These changes are often spurious and unrelated to the underlying problem, such as backg…

2021

On Monotonic Linear Interpolation of Neural Network Parameters

ICML 2021spotlight

Linear interpolation between initial neural network parameters and converged parameters after training with stochastic gradient descent (SGD) typically leads to a monotonic decrease in the training objective. This Monotonic Linear Interpolation (MLI) property, first observed by Goodfellow et al. 201…

Cited by 15SourcePDFScholar