KDk: A Defense Mechanism Against Label Inference Attacks in Vertical Federated Learning

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Arazzi, Marco, Nicolazzo, Serena, Nocera, Antonino
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866909464900141056
author Arazzi, Marco
Nicolazzo, Serena
Nocera, Antonino
author_facet Arazzi, Marco
Nicolazzo, Serena
Nocera, Antonino
contents Vertical Federated Learning (VFL) is a category of Federated Learning in which models are trained collaboratively among parties with vertically partitioned data. Typically, in a VFL scenario, the labels of the samples are kept private from all the parties except for the aggregating server, that is the label owner. Nevertheless, recent works discovered that by exploiting gradient information returned by the server to bottom models, with the knowledge of only a small set of auxiliary labels on a very limited subset of training data points, an adversary can infer the private labels. These attacks are known as label inference attacks in VFL. In our work, we propose a novel framework called KDk, that combines Knowledge Distillation and k-anonymity to provide a defense mechanism against potential label inference attacks in a VFL scenario. Through an exhaustive experimental campaign we demonstrate that by applying our approach, the performance of the analyzed label inference attacks decreases consistently, even by more than 60%, maintaining the accuracy of the whole VFL almost unaltered.
format Preprint
id arxiv_https___arxiv_org_abs_2404_12369
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle KDk: A Defense Mechanism Against Label Inference Attacks in Vertical Federated Learning
Arazzi, Marco
Nicolazzo, Serena
Nocera, Antonino
Machine Learning
Cryptography and Security
Vertical Federated Learning (VFL) is a category of Federated Learning in which models are trained collaboratively among parties with vertically partitioned data. Typically, in a VFL scenario, the labels of the samples are kept private from all the parties except for the aggregating server, that is the label owner. Nevertheless, recent works discovered that by exploiting gradient information returned by the server to bottom models, with the knowledge of only a small set of auxiliary labels on a very limited subset of training data points, an adversary can infer the private labels. These attacks are known as label inference attacks in VFL. In our work, we propose a novel framework called KDk, that combines Knowledge Distillation and k-anonymity to provide a defense mechanism against potential label inference attacks in a VFL scenario. Through an exhaustive experimental campaign we demonstrate that by applying our approach, the performance of the analyzed label inference attacks decreases consistently, even by more than 60%, maintaining the accuracy of the whole VFL almost unaltered.
title KDk: A Defense Mechanism Against Label Inference Attacks in Vertical Federated Learning
topic Machine Learning
Cryptography and Security
url https://arxiv.org/abs/2404.12369