Saved in:
Bibliographic Details
Main Authors: Han, Mengde, Zhu, Tianqing, Zhang, Lefeng, Huo, Huan, Zhou, Wanlei
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2412.11476
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912158647844864
author Han, Mengde
Zhu, Tianqing
Zhang, Lefeng
Huo, Huan
Zhou, Wanlei
author_facet Han, Mengde
Zhu, Tianqing
Zhang, Lefeng
Huo, Huan
Zhou, Wanlei
contents Vertical Federated Learning (VFL) offers a novel paradigm in machine learning, enabling distinct entities to train models cooperatively while maintaining data privacy. This method is particularly pertinent when entities possess datasets with identical sample identifiers but diverse attributes. Recent privacy regulations emphasize an individual's \emph{right to be forgotten}, which necessitates the ability for models to unlearn specific training data. The primary challenge is to develop a mechanism to eliminate the influence of a specific client from a model without erasing all relevant data from other clients. Our research investigates the removal of a single client's contribution within the VFL framework. We introduce an innovative modification to traditional VFL by employing a mechanism that inverts the typical learning trajectory with the objective of extracting specific data contributions. This approach seeks to optimize model performance using gradient ascent, guided by a pre-defined constrained model. We also introduce a backdoor mechanism to verify the effectiveness of the unlearning procedure. Our method avoids fully accessing the initial training data and avoids storing parameter updates. Empirical evidence shows that the results align closely with those achieved by retraining from scratch. Utilizing gradient ascent, our unlearning approach addresses key challenges in VFL, laying the groundwork for future advancements in this domain. All the code and implementations related to this paper are publicly available at https://github.com/mengde-han/VFL-unlearn.
format Preprint
id arxiv_https___arxiv_org_abs_2412_11476
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Vertical Federated Unlearning via Backdoor Certification
Han, Mengde
Zhu, Tianqing
Zhang, Lefeng
Huo, Huan
Zhou, Wanlei
Machine Learning
Vertical Federated Learning (VFL) offers a novel paradigm in machine learning, enabling distinct entities to train models cooperatively while maintaining data privacy. This method is particularly pertinent when entities possess datasets with identical sample identifiers but diverse attributes. Recent privacy regulations emphasize an individual's \emph{right to be forgotten}, which necessitates the ability for models to unlearn specific training data. The primary challenge is to develop a mechanism to eliminate the influence of a specific client from a model without erasing all relevant data from other clients. Our research investigates the removal of a single client's contribution within the VFL framework. We introduce an innovative modification to traditional VFL by employing a mechanism that inverts the typical learning trajectory with the objective of extracting specific data contributions. This approach seeks to optimize model performance using gradient ascent, guided by a pre-defined constrained model. We also introduce a backdoor mechanism to verify the effectiveness of the unlearning procedure. Our method avoids fully accessing the initial training data and avoids storing parameter updates. Empirical evidence shows that the results align closely with those achieved by retraining from scratch. Utilizing gradient ascent, our unlearning approach addresses key challenges in VFL, laying the groundwork for future advancements in this domain. All the code and implementations related to this paper are publicly available at https://github.com/mengde-han/VFL-unlearn.
title Vertical Federated Unlearning via Backdoor Certification
topic Machine Learning
url https://arxiv.org/abs/2412.11476