arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2012.02275cs.LGcs.AIcs.CVcs.ITmath.IT

Detecting Trojaned DNNs Using Counterfactual Attributions

SRI International(SRI国际)

Karan Sikka, Indranil Sur, Susmit Jha, Anirban Roy, Ajay Divakaran

英文摘要:

We target the problem of detecting Trojans or backdoors in DNNs. Such models behave normally with typical inputs but produce specific incorrect predictions for inputs poisoned with a Trojan trigger. Our approach is based on a novel observation that the trigger behavior depends on a few ghost neurons that activate on trigger pattern and exhibit abnormally higher relative attribution for wrong decisions when activated. Further, these trigger neurons are also active on normal inputs of the target class. Thus, we use counterfactual attributions to localize these ghost neurons from clean inputs and then incrementally excite them to observe changes in the model's accuracy. We use this information for Trojan detection by using a deep set encoder that enables invariance to the number of model classes, architecture, etc. Our approach is implemented in the TrinityAI tool that exploits the synergies between trustworthiness, resilience, and interpretability challenges in deep learning. We evaluate our approach on benchmarks with high diversity in model architectures, triggers, etc. We show consistent gains (+10%) over state-of-the-art methods that rely on the susceptibility of the DNN to specific adversarial attacks, which in turn requires strong assumptions on the nature of the Trojan attack.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑