Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Forgetting Any Data at Any Time: A Theoretically Certified Unlearning Framework for Vertical Federated Learning

About

Privacy concerns in machine learning are heightened by regulations such as the GDPR, which enforces the "right to be forgotten" (RTBF), driving the emergence of machine unlearning as a critical research field. Vertical Federated Learning (VFL) enables collaborative model training by aggregating a sample's features across distributed parties while preserving data privacy at each source. This paradigm has seen widespread adoption in healthcare, finance, and other privacy-sensitive domains. However, existing VFL systems lack robust mechanisms to comply with RTBF requirements, as unlearning methodologies for VFL remain underexplored. In this work, we introduce the first VFL framework with theoretically guaranteed unlearning capabilities, enabling the removal of any data at any time. Unlike prior approaches -- which impose restrictive assumptions on model architectures or data types for removal -- our solution is model- and data-agnostic, offering universal compatibility. Moreover, our framework supports asynchronous unlearning, eliminating the need for all parties to be simultaneously online during the forgetting process. These advancements address critical gaps in current VFL systems, ensuring compliance with RTBF while maintaining operational flexibility.We make all our implementations publicly available at https://github.com/wangln19/vertical-federated-unlearning.

Linian Wang, Leye Wang• 2025

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (test)--
3518
Image ClassificationCIFAR-10 (test)--
3381
Image ClassificationTiny ImageNet (test)
Accuracy52.09
265
Class UnlearningCIFAR-10 (test)
Test Accuracy79.05
21
Binary ClassificationIncome (test)
Test Accuracy82.84
20
Class UnlearningTiny ImageNet (test)--
19
Class UnlearningCIFAR-100 (test)--
13
Label UnlearningMedMNIST PathMNIST (test)
Test Accuracy84.78
10
Image ClassificationMedMNIST PathMNIST (test)
Accuracy85.47
10
Federated UnlearningIncome
Running Time (s)1.18
5
Showing 10 of 14 rows

Other info

Follow for update