Safe foundation models
Developing methods for safer generation in diffusion and text-diffusion models, including safety-aware denoisers and negative guidance.
Assistant Professor · Department of Computer Science · University of British Columbia
Create machine learning tools for a better world.
I lead a machine learning research group working on privacy-preserving machine learning, differentially private synthetic data generation, Bayesian learning, and AI safety in diffusion foundation models.
Our lab develops privacy- and safety-preserving machine learning methods for sensitive, high-dimensional settings, with a probabilistic perspective grounded in Bayesian inference, kernel methods, uncertainty quantification, and generative modelling. We have made foundational contributions to differentially private synthetic data generation, including DP-MERF, DP-HP (Hermite-Polynomials), and DP-LDMs (latent diffusion models). We have also developed privacy-preserving inference and testing methods, including Variational Bayes in Private Settings, DP kernel two-sample testing, and interpretable & DP prediction. Our recent work extends this program to safe diffusion foundation models, including Training-Free Safe Denoisers, Safety-Guided Flow, and Safety-Aware Denoising for text diffusion. We are also interested in applying Bayesian perspectives to modern machine learning systems.
Developing methods for safer generation in diffusion and text-diffusion models, including safety-aware denoisers and negative guidance.
Designing differentially private algorithms for synthetic data generation and data distillation.
Studying how privacy-preserving learning changes model behavior, interpretability, fairness, and causal analysis.
Applying Bayesian methods to efficient inference, neural data analysis, neural network compression, and foundation-model prompting.