Stochastic Meta-Unlearning: Bridging Language Backbone and Multimodal Unlearning
Summary
Stochastic Meta-Unlearning (SMU) is a novel bilevel framework designed to address the complexities of machine unlearning in vision-language models (VLMs). Unlike single-modality models, VLMs present a unique challenge where targets forgotten by a language backbone can still be recovered when visual information is provided to the full VLM, indicating the inadequacy of text-only unlearning. SMU tackles this by learning an unlearning-ready initialization using VLM-level feedback. Its inner loop applies unlearning steps to the language backbone with text data, while the outer loop recomposes the updated backbone with the frozen VLM to evaluate forgetting and utility at the multimodal level. This approach ensures unlearning updates consider final multimodal behavior, localized to the language backbone. Experiments on two VLMs, two multimodal meme datasets, and three baselines demonstrate SMU's superior forget-retain trade-off, reducing average Forget accuracy by 10.52 points and improving average Retain and Test accuracy by 20.10 and 17.01 points, respectively. SMU also exhibits transferability to new forgetting targets and different meta-test unlearning methods.
Key takeaway
For Machine Learning Engineers developing unlearning solutions for vision-language models, you should integrate VLM-level feedback into your unlearning processes. Relying solely on text-only feedback for language backbones is insufficient, as forgotten information can still be recovered multimodally. Adopting a bilevel framework like Stochastic Meta-Unlearning can significantly improve forget-retain trade-offs and enhance transferability to new unlearning targets, ensuring more robust and reliable model unlearning.
Key insights
VLM-level feedback makes language-backbone unlearning more reliable and transferable.
Principles
- Text-only feedback is insufficient for VLM unlearning.
- Unlearning updates should be aware of multimodal behavior.
- Bilevel optimization can achieve better forget-retain trade-offs.
Method
SMU uses a bilevel framework: inner loop applies unlearning to language backbone with text; outer loop evaluates VLM-level forgetting and utility.
In practice
- Evaluate unlearning at the full VLM level, not just backbone.
- Consider bilevel optimization for complex multimodal unlearning.
- Focus on transferability to new forgetting targets.
Topics
- Machine Unlearning
- Vision-Language Models
- Stochastic Meta-Unlearning
- Language Backbones
- Multimodal AI
- Bilevel Optimization
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, AI Security Engineer
Related on AIssential
See Counsel's argued verdicts on the open AI decisions leaders are weighing →
Editorial summary, takeaway, and curation by AIssential. Original article published by Computation and Language.