Track papers across labs, topics, and trends.
Search papers, labs, and topics across Lattice.
This study investigates machine unlearning strategies for 3D medical image segmentation, specifically addressing compliance with GDPR data erasure requirements. Utilizing a pre-trained 3D ResNet-50 model on the MRBrainS18 dataset, the authors evaluate various unlearning methods through metrics such as Dice similarity coefficient and mean absolute error. The Noisy Label strategy emerged as the most effective, achieving a 93% decrease in the forget set while retaining 84% accuracy for the retained set after 50 epochs, setting a performance baseline for future unlearning approaches.
The Noisy Label strategy achieves a remarkable balance, erasing 93% of forgotten data while preserving 84% accuracy in retained data, setting a new standard for machine unlearning in medical imaging.
With new data privacy laws such as the General Data Protection Regulation (GDPR) [1] that allow individuals to ask that any of their personal information be erased from trained machine learning models, there has been a push to investigate the unlearning of data from models as a way to comply with these laws. In this regard, based on four mechanics, we consider several approximate unlearning strategies applied to the MRBrainS18 dataset [2]. We use a 3D ResNet-50 [3] as a backbone architecture for segmentation that has been pre-trained with the Med3D framework [4]. Considering the pre-trained model as a baseline, we evaluate respective retention accuracy on 2 types of subjects, i.e., retain and forget. We assess these approaches through their Dice similarity coefficient and mean absolute error (MAE) values using two separate training horizons 20 and 50 epochs. The results show that the Noisy Label strategy had the best overall trade-off with a decrease of 93% in the forget set while maintaining 84% accuracy for the retained set after 50 epochs. All other strategies showed extreme levels of forgetting at higher epoch numbers while also demonstrating catastrophic degradation of their retain set performance. The results of this study provide a strict baseline of performance metrics for unlearning on a subject-specific level and provide practitioners with clear criteria for selecting the proper strategies.