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