Graph Propagated Projection Unlearning: A Unified Framework for Vision and Audio Discriminative Models
Summary
Graph-Propagated Projection Unlearning (GPPU) is a new, unified, and scalable algorithm designed for class-level unlearning in deep neural networks, applicable to both vision and audio models. GPPU addresses the need for efficient removal of learned information due to privacy, regulatory compliance, and adaptive system design requirements. The method operates in two phases: first, it uses graph-based propagation to identify class-specific directions in the feature space, and second, it projects representations onto an orthogonal subspace, followed by targeted fine-tuning, to irreversibly remove target class information. Evaluations across six vision datasets and two large-scale audio benchmarks, using architectures like CNNs, Vision Transformers, and Audio Transformers, demonstrate that GPPU achieves 10-20x speedups over prior methods while preserving model utility on retained classes. It consistently reaches near-zero forget-class accuracy post-unlearning and maintains high utility on retained classes.
Key takeaway
For machine learning engineers and researchers building privacy-compliant or adaptable AI systems, GPPU offers a significantly more efficient and effective approach to class-level unlearning. You should consider integrating GPPU for scenarios requiring rapid and precise removal of specific class knowledge, especially in large-scale vision and audio models, as it drastically reduces computational overhead while maintaining high model utility on retained data. This framework enables robust continual unlearning and better adherence to data protection regulations like India's Digital Personal Data Protection Act, 2023.
Key insights
GPPU efficiently unlearns specific classes from deep networks by geometrically projecting features and fine-tuning.
Principles
- Graph propagation refines class boundaries.
- Orthogonal projection removes class-specific features.
- Targeted fine-tuning ensures intrinsic unlearning.
Method
GPPU identifies class-specific "forget directions" via graph-based propagation on features, then projects representations onto the orthogonal subspace and fine-tunes the model's final layers using projection and retention losses.
In practice
- Achieves 10-20x speedup over prior unlearning methods.
- Maintains >93% retention accuracy on non-target classes.
- Applicable to both vision and audio models.
Topics
- Graph-Propagated Projection Unlearning
- Machine Unlearning
- Class-Level Unlearning
- Vision Models
- Audio Models
Best for: Research Scientist, CTO, VP of Engineering/Data, AI Scientist, Machine Learning Engineer, AI Security Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.AI updates on arXiv.org.