LMCleaner: Efficient and Certified Online Unlearning via Influence Propagation Truncation
Abstract
Lay Summary
Modern AI systems like ChatGPT learn from enormous amounts of text scraped from the internet. Some of that text turns out to be harmful, copyrighted, or contains private information. Laws like Europe's "right to be forgotten" require companies to remove such data from their models. But how do you make a trained AI truly forget something? Retraining from scratch costs millions of dollars. Existing shortcuts only act after training is finished, by which point the unwanted influence has already spread through the model. We developed LMCleaner, a method that removes the influence of unwanted data while the model is still being trained, the moment that data is detected. Our key insight is that this influence does not need to be tracked through the entire training history. Most of it concentrates within a short recent window of training steps. The small remainder can be neutralized with carefully calibrated random noise. We mathematically prove that the cleaned model behaves as if it had never seen the unwanted data. In experiments, LMCleaner runs over 100 times faster than competing methods while preserving the model's overall abilities. This makes verifiable data deletion practical for today's largest AI systems, helping organizations comply with privacy laws and respond quickly when problematic training data is found.