Robust Backdoor Removal by Reconstructing Trigger-Activated Changes in Latent Representation

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Iwahana, Kazuki, Yamasaki, Yusuke, Ito, Akira, Miura, Takayuki, Shibahara, Toshiki
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914153040445440
author Iwahana, Kazuki
Yamasaki, Yusuke
Ito, Akira
Miura, Takayuki
Shibahara, Toshiki
author_facet Iwahana, Kazuki
Yamasaki, Yusuke
Ito, Akira
Miura, Takayuki
Shibahara, Toshiki
contents Backdoor attacks pose a critical threat to machine learning models, causing them to behave normally on clean data but misclassify poisoned data into a poisoned class. Existing defenses often attempt to identify and remove backdoor neurons based on Trigger-Activated Changes (TAC) which is the activation differences between clean and poisoned data. These methods suffer from low precision in identifying true backdoor neurons due to inaccurate estimation of TAC values. In this work, we propose a novel backdoor removal method by accurately reconstructing TAC values in the latent representation. Specifically, we formulate the minimal perturbation that forces clean data to be classified into a specific class as a convex quadratic optimization problem, whose optimal solution serves as a surrogate for TAC. We then identify the poisoned class by detecting statistically small $L^2$ norms of perturbations and leverage the perturbation of the poisoned class in fine-tuning to remove backdoors. Experiments on CIFAR-10, GTSRB, and TinyImageNet demonstrated that our approach consistently achieves superior backdoor suppression with high clean accuracy across different attack types, datasets, and architectures, outperforming existing defense methods.
format Preprint
id arxiv_https___arxiv_org_abs_2511_08944
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Robust Backdoor Removal by Reconstructing Trigger-Activated Changes in Latent Representation
Iwahana, Kazuki
Yamasaki, Yusuke
Ito, Akira
Miura, Takayuki
Shibahara, Toshiki
Machine Learning
Cryptography and Security
Backdoor attacks pose a critical threat to machine learning models, causing them to behave normally on clean data but misclassify poisoned data into a poisoned class. Existing defenses often attempt to identify and remove backdoor neurons based on Trigger-Activated Changes (TAC) which is the activation differences between clean and poisoned data. These methods suffer from low precision in identifying true backdoor neurons due to inaccurate estimation of TAC values. In this work, we propose a novel backdoor removal method by accurately reconstructing TAC values in the latent representation. Specifically, we formulate the minimal perturbation that forces clean data to be classified into a specific class as a convex quadratic optimization problem, whose optimal solution serves as a surrogate for TAC. We then identify the poisoned class by detecting statistically small $L^2$ norms of perturbations and leverage the perturbation of the poisoned class in fine-tuning to remove backdoors. Experiments on CIFAR-10, GTSRB, and TinyImageNet demonstrated that our approach consistently achieves superior backdoor suppression with high clean accuracy across different attack types, datasets, and architectures, outperforming existing defense methods.
title Robust Backdoor Removal by Reconstructing Trigger-Activated Changes in Latent Representation
topic Machine Learning
Cryptography and Security
url https://arxiv.org/abs/2511.08944