paper-with-me

Papers

Flatness-aware Sequential Learning Generates Resilient Backdoors

2024-07-20 · Hoang Pham, The-Anh Ta, Anh Tran, Khoa D. Doan

Recently, backdoor attacks have become an emerging threat to the security of machine learning models. From the adversary's perspective, the implanted backdoors should be resistant to defensive algorithms, but some recently proposed fine-tuning defenses can remove these backdoors with notable efficacy. This is mainly due to the catastrophic forgetting (CF) property of deep neural networks. This paper counters CF of backdoors by leveraging continual learning (CL) techniques. We begin by investigating the connectivity between a backdoored and fine-tuned model in the loss landscape. Our analysis confirms that fine-tuning defenses, especially the more advanced ones, can easily push a poisoned model out of the backdoor regions, making it forget all about the backdoors. Based on this finding, we re-formulate backdoor training through the lens of CL and propose a novel framework, named Sequential Backdoor Learning (SBL), that can generate resilient backdoors. This framework separates the backdoor poisoning process into two tasks: the first task learns a backdoored model, while the second task, based on the CL principles, moves it to a backdoored region resistant to fine-tuning. We additionally propose to seek flatter backdoor regions via a sharpness-aware minimizer in the framework, further strengthening the durability of the implanted backdoor. Finally, we demonstrate the effectiveness of our method through extensive empirical experiments on several benchmark datasets in the backdoor domain. The source code is available at https://github.com/mail-research/SBL-resilient-backdoors

📄 PDF Abstract BibTeX arXiv:2407.14738

Code (1)

mail-research/sbl-resilient-backdoors 공식 구현 pytorch

Tasks

Continual Learning

Similar Papers 제목 키워드 기반

Data Augmented Flatness-aware Gradient Projection for Continual Learning

2023-01-01 · ICCV 2023 1 · Enneng Yang, Li Shen, Zhenyi Wang, Shiwei Liu 외

The goal of continual learning (CL) is to continuously learn new tasks without forgetting previously learned old tasks. To alleviate catastrophic forgetting, gradient projection based CL methods require that the grad…

Continual Learning

FAM: Relative Flatness Aware Minimization

2023-07-05 · Linara Adilova, Amr Abourayya, Jianning Li, Amin Dada 외

Flatness of the loss curve around a model at hand has been shown to empirically correlate with its generalization ability. Optimizing for flatness has been proposed as early as 1994 by Hochreiter and Schmidthuber, and wa…

Input-Aware Dynamic Backdoor Attack Against Quantum Neural Networks

2026-07-13 · Junrui Zhang, Zemin Chen, Lusi Li, Mohammad Ghasemigol 외 arxiv

Quantum Neural Networks (QNNs) are a promising framework for quantum machine learning on near-term quantum devices, but their security risks remain insufficiently understood. Studies have shown that QNNs are vulnerable t…

Quantum Machine LearningContrastive Learning

Flatness-Aware Minimization for Domain Generalization

2023-07-20 · ICCV 2023 1 · Xingxuan Zhang, Renzhe Xu, Han Yu, Yancheng Dong 외

Domain generalization (DG) seeks to learn robust models that generalize well under unknown distribution shifts. As a critical aspect of DG, optimizer selection has not been explored in depth. Currently, most DG methods f…

Domain GeneralizationFAD

Gradient Norm Aware Minimization Seeks First-Order Flatness and Improves Generalization

2023-03-03 · CVPR 2023 1 · Xingxuan Zhang, Renzhe Xu, Han Yu, Hao Zou 외

Recently, flat minima are proven to be effective for improving generalization and sharpness-aware minimization (SAM) achieves state-of-the-art performance. Yet the current definition of flatness discussed in SAM and its …