Search papers, labs, and topics across Lattice.
This paper introduces a modified SISA (Sharded, Isolated, Sliced, and Aggregated) framework with a reinforced replay mechanism and gating network to achieve class-level unlearning in CNNs. The approach enables efficient removal of entire classes of data from trained models without full retraining, addressing growing data privacy concerns. Experiments on image datasets show the modified SISA framework effectively unlearns classes while maintaining model performance and reducing computational overhead.
Forget individual data points? Child's play. This work lets you surgically remove entire *classes* of data from CNNs without catastrophic forgetting.
The rapid proliferation of image generation models and other artificial intelligence (AI) systems has intensified concerns regarding data privacy and user consent. As the availability of public datasets declines, major technology companies increasingly rely on proprietary or private user data for model training, raising ethical and legal challenges when users request the deletion of their data after it has influenced a trained model. Machine unlearning seeks to address this issue by enabling the removal of specific data from models without complete retraining. This study investigates a modified SISA (Sharded, Isolated, Sliced, and Aggregated) framework designed to achieve class-level unlearning in Convolutional Neural Network (CNN) architectures. The proposed framework incorporates a reinforced replay mechanism and a gating network to enhance selective forgetting efficiency. Experimental evaluations across multiple image datasets and CNN configurations demonstrate that the modified SISA approach enables effective class unlearning while preserving model performance and reducing retraining overhead. The findings highlight the potential of SISA-based unlearning for deployment in privacy-sensitive AI applications. The implementation is publicly available at https://github.com/SiamFS/ sisa-class-unlearning.
