ScholarWorks@성균관대학교

초록

Machine unlearning has emerged as a significant research topic in response to the increasing demands for data privacy and compliance with privacy regulations. The main challenge is to eliminate the influence of a specific subset of training data from a pretrained model while preserving the model's performance on the retain set without retraining the model from scratch. In this paper, we propose a novel efficient unlearning framework based on Optimal Transport, which can effectively work on both class-wise and instance-wise unlearning tasks. By analyzing and comparing the feature spaces of the original and retrained models, we formulate the unlearning problem as a distribution alignment task between the forget set and the retain set. We guide the feature distribution of the forget set, which initially forms distinct and structured patterns, to align with that of the retain set. Extensive experiments on three public benchmark datasets demonstrate its superior effectiveness compared to previous state-of-the-art methods.

키워드