Graph Propagated Projection Unlearning: A Unified Framework for Vision and Audio Discriminative Models

· Source: cs.AI updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Cybersecurity & Data Privacy · Depth: Expert, extended

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

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

Topics

Best for: Research Scientist, CTO, VP of Engineering/Data, AI Scientist, Machine Learning Engineer, AI Security Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.AI updates on arXiv.org.