Universal and Efficient Adversarial Training Framework With Membership Inference Resistance
Adversarial training is an effective approach to enhance the robustness of machine learning models via adding adversarial examples into the training phase. However, existing adversarial training methods increase the advantage of membership inference attacks, which aim to determine from the model whether an example is in the training dataset. In this article, we propose an adversarial training framework that guarantees both robustness and membership privacy by introducing a tailor-made example called reverse-symmetry example. Moreover, our framework reduces the number of required adversarial examples compared with existing adversarial training methods. We implement our framework using four adversarial training methods on the FMNIST and CIFAR10 datasets and compare its performance with deep learning differential privacy. Our experimental findings demonstrate that our framework mitigates model overfitting and outperforms the original adversarial training with respect to the overall performance of accuracy, robustness, privacy, and runtime.
Authors
- Ran Yan (ORCID: https://orcid.org/0000-0002-7891-0624)
- Kun He (ORCID: https://orcid.org/0000-0003-3472-419X)
- Qiao Li (ORCID: https://orcid.org/0000-0002-3694-8237)
- Jing Chen (ORCID: https://orcid.org/0000-0002-7212-5297)
- Ruiying Du (ORCID: https://orcid.org/0000-0002-3634-3385)
- Cong Wu (ORCID: https://orcid.org/0000-0002-0930-0283)
Institutions
- Nanyang Technological University (SG)
- Wuhan University (CN)
Publication Details
- Journal
- IEEE Internet of Things Journal
- Published
- 2025-03-17
- DOI
- https://doi.org/10.1109/jiot.2025.3551762
- Citations
- 1
- Primary Topic
- Adversarial Robustness in Machine Learning
- Type
- article
- Field-Weighted Citation Impact
- 2.48
Funders
- National Natural Science Foundation of China
- National Key Research and Development Program of China