paper-with-me

홈 › Papers

On the Effectiveness of Distillation in Mitigating Backdoors in Pre-trained Encoder

2024-03-06 · Tingxu Han, Shenghan Huang, Ziqi Ding, Weisong Sun, Yebo Feng, Chunrong Fang, Jun Li, Hanwei Qian, Cong Wu, Quanjun Zhang, Yang Liu, Zhenyu Chen

In this paper, we study a defense against poisoned encoders in SSL called distillation, which is a defense used in supervised learning originally. Distillation aims to distill knowledge from a given model (a.k.a the teacher net) and transfer it to another (a.k.a the student net). Now, we use it to distill benign knowledge from poisoned pre-trained encoders and transfer it to a new encoder, resulting in a clean pre-trained encoder. In particular, we conduct an empirical study on the effectiveness and performance of distillation against poisoned encoders. Using two state-of-the-art backdoor attacks against pre-trained image encoders and four commonly used image classification datasets, our experimental results show that distillation can reduce attack success rate from 80.87% to 27.51% while suffering a 6.35% loss in accuracy. Moreover, we investigate the impact of three core components of distillation on performance: teacher net, student net, and distillation loss. By comparing 4 different teacher nets, 3 student nets, and 6 distillation losses, we find that fine-tuned teacher nets, warm-up-training-based student nets, and attention-based distillation loss perform best, respectively.

📄 PDF Abstract BibTeX arXiv:2403.03846

Code (1)

wssun/sslbackdoormitigation 공식 구현 pytorch

Tasks

image-classificationImage Classification

Similar Papers 제목 키워드 기반

Revisiting Data-Free Knowledge Distillation with Poisoned Teachers

2023-06-04 · Junyuan Hong, Yi Zeng, Shuyang Yu, Lingjuan Lyu 외

Data-free knowledge distillation (KD) helps transfer knowledge from a pre-trained model (known as the teacher model) to a smaller model (known as the student model) without access to the original training data used for t…

Backdoor Defense for Data-Free Distillation with Poisoned TeachersData-free Knowledge DistillationKnowledge Distillation

Mutual Information Guided Backdoor Mitigation for Pre-trained Encoders

2024-06-05 · Tingxu Han, Weisong Sun, Ziqi Ding, Chunrong Fang 외

Self-supervised learning (SSL) is increasingly attractive for pre-training encoders without requiring labeled data. Downstream tasks built on top of those pre-trained encoders can achieve nearly state-of-the-art performa…

Knowledge DistillationSelf-Supervised Learning

MBTSAD: Mitigating Backdoors in Language Models Based on Token Splitting and Attention Distillation

2025-01-06 · Yidong Ding, Jiafei Niu, Ping Yi

In recent years, attention-based models have excelled across various domains but remain vulnerable to backdoor attacks, often from downloading or fine-tuning on poisoned datasets. Many current methods to mitigate backdoo…

Pay Attention to the Triggers: Constructing Backdoors That Survive Distillation

2025-10-21 · Giovanni De Muri, Mark Vero, Robin Staab, Martin Vechev arxiv

LLMs are often used by downstream users as teacher models for knowledge distillation, compressing their capabilities into memory-efficient models. However, as these teacher models may stem from untrusted parties, distill…

Knowledge Distillation

Dark Distillation: Backdooring Distilled Datasets without Accessing Raw Data

2025-02-06 · Ziyuan Yang, Ming Yan, Yi Zhang, Joey Tianyi Zhou

Dataset distillation (DD) enhances training efficiency and reduces bandwidth by condensing large datasets into smaller synthetic ones. It enables models to achieve performance comparable to those trained on the raw full …

Dataset Distillation