atlas

Research map / Adversarial robustness and security / Backdoor and poisoning

Causality Based Front-door Defense Against Backdoor Attack on Language Models

ICML 2024 · Adversarial robustness and security › Backdoor and poisoning · approach: Backdoors in large multimodal models

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes FABE, a front-door adjustment defense against backdoor attacks on text classifiers, where a fine-tuned LLaMA2 defense model paraphrases inputs into semantically equivalent front-door variables whose predictions are aggregated causally.

Problem

Backdoor attacks on language models with diverse trigger types (token, sentence, syntactic) are hard to defend, because existing defenses rely on assumptions about trigger form and backdoor-adjustment requires observable attack features.

Approach

Treat the backdoor trigger as an unobservable confounder and apply Pearl's front-door adjustment, using an LLM to generate semantically equivalent paraphrases as front-door variables, scored by beam-search likelihood and trained with MLE plus a margin-based ranking loss so that paraphrases preserve the victim model's predictions.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use FABE instead of ONIONBackdoor defense on language models · SST-2, Offenseval, HSOL (average)ASR: 66.61 -> 15.12Helped
Use FABE instead of no defenseBackdoor defense on language models · SST-2, Offenseval, HSOL (average)ASR: 93.63 -> 15.12Helped
Use FABE instead of no defenseBackdoor defense on language models · SST-2, Offenseval, HSOL (average)CA: 87.47 -> 85.52Hurt
Rewrite poisoned inputs with a pre-trained model (ablation)BERT victim model · SST-2, Offenseval, HSOLASRHelped
Fine-tune the rewriting model with MLE loss only (ablation)BERT victim model · SST-2, Offenseval, HSOLASRHurt
Add ranking loss to MLE fine-tuning (ablation)BERT victim model · SST-2, Offenseval, HSOLASRHelped
Apply front-door adjustment to the ranking-tuned defense model (ablation)BERT victim model · SST-2, Offenseval, HSOLASRHelped

Surprising findings

Limitations

Experimental setup

Datasets: SST-2 Offenseval HSOL

Benchmarks: SST-2 Offenseval HSOL

Metrics: Attack Success Rate Clean Accuracy

Baselines: ONION RAP STRIP

Builds on: front-door adjustment (Pearl) OpenBackdoor instruction tuning

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy