PureGen

PureGen: Universal Data Purification for Train-Time Poison Defense via Generative Model Dynamics

PureGen purifies poisoned training data using iterative Langevin dynamics of Energy-Based Models and Denoising Diffusion Probabilistic Models, achieving state-of-the-art poison defense with minimal impact on classifier generalization.

June 2024 · Sunay Bhat, Jeffrey Jiang, Omead Pooladzandi, Alexander Branch, Gregory Pottie
Composable Latent Space Augmentations

Towards Composable Distributions of Latent Space Augmentations

A composable latent space augmentation framework using VAEs that allows augmentations to be combined through linear transformations, preserving specific augmentation variances and improving geometric interpretability over standard and Conditional VAEs.

March 2023 · Omead Pooladzandi, Jeffrey Jiang, Sunay Bhat, Gregory Pottie
De-Biasing Generative Models via Counterfactual Methods

De-Biasing Generative Models using Counterfactual Methods

CCGM combines a causal latent space VAE with modifications for causal fidelity to generate de-biased datasets from biased training data, offering fine-grained control over causal structure in both image and tabular data generation.

July 2022 · Sunay Bhat, Jeffrey Jiang, Omead Pooladzandi, Gregory Pottie