Mitigating Privacy Risk in Membership Inference by Convex-Concave Loss

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Liu, Zhenlong, Feng, Lei, Zhuang, Huiping, Cao, Xiaofeng, Wei, Hongxin
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917696362250240
author Liu, Zhenlong
Feng, Lei
Zhuang, Huiping
Cao, Xiaofeng
Wei, Hongxin
author_facet Liu, Zhenlong
Feng, Lei
Zhuang, Huiping
Cao, Xiaofeng
Wei, Hongxin
contents Machine learning models are susceptible to membership inference attacks (MIAs), which aim to infer whether a sample is in the training set. Existing work utilizes gradient ascent to enlarge the loss variance of training data, alleviating the privacy risk. However, optimizing toward a reverse direction may cause the model parameters to oscillate near local minima, leading to instability and suboptimal performance. In this work, we propose a novel method -- Convex-Concave Loss, which enables a high variance of training loss distribution by gradient descent. Our method is motivated by the theoretical analysis that convex losses tend to decrease the loss variance during training. Thus, our key idea behind CCL is to reduce the convexity of loss functions with a concave term. Trained with CCL, neural networks produce losses with high variance for training data, reinforcing the defense against MIAs. Extensive experiments demonstrate the superiority of CCL, achieving state-of-the-art balance in the privacy-utility trade-off.
format Preprint
id arxiv_https___arxiv_org_abs_2402_05453
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Mitigating Privacy Risk in Membership Inference by Convex-Concave Loss
Liu, Zhenlong
Feng, Lei
Zhuang, Huiping
Cao, Xiaofeng
Wei, Hongxin
Machine Learning
Cryptography and Security
Machine learning models are susceptible to membership inference attacks (MIAs), which aim to infer whether a sample is in the training set. Existing work utilizes gradient ascent to enlarge the loss variance of training data, alleviating the privacy risk. However, optimizing toward a reverse direction may cause the model parameters to oscillate near local minima, leading to instability and suboptimal performance. In this work, we propose a novel method -- Convex-Concave Loss, which enables a high variance of training loss distribution by gradient descent. Our method is motivated by the theoretical analysis that convex losses tend to decrease the loss variance during training. Thus, our key idea behind CCL is to reduce the convexity of loss functions with a concave term. Trained with CCL, neural networks produce losses with high variance for training data, reinforcing the defense against MIAs. Extensive experiments demonstrate the superiority of CCL, achieving state-of-the-art balance in the privacy-utility trade-off.
title Mitigating Privacy Risk in Membership Inference by Convex-Concave Loss
topic Machine Learning
Cryptography and Security
url https://arxiv.org/abs/2402.05453