English

Explainability-Guided Defense: Attribution-Aware Model Refinement Against Adversarial Data Attacks

Machine Learning 2026-01-06 v1

Abstract

The growing reliance on deep learning models in safety-critical domains such as healthcare and autonomous navigation underscores the need for defenses that are both robust to adversarial perturbations and transparent in their decision-making. In this paper, we identify a connection between interpretability and robustness that can be directly leveraged during training. Specifically, we observe that spurious, unstable, or semantically irrelevant features identified through Local Interpretable Model-Agnostic Explanations (LIME) contribute disproportionately to adversarial vulnerability. Building on this insight, we introduce an attribution-guided refinement framework that transforms LIME from a passive diagnostic into an active training signal. Our method systematically suppresses spurious features using feature masking, sensitivity-aware regularization, and adversarial augmentation in a closed-loop refinement pipeline. This approach does not require additional datasets or model architectures and integrates seamlessly into standard adversarial training. Theoretically, we derive an attribution-aware lower bound on adversarial distortion that formalizes the link between explanation alignment and robustness. Empirical evaluations on CIFAR-10, CIFAR-10-C, and CIFAR-100 demonstrate substantial improvements in adversarial robustness and out-of-distribution generalization.

Keywords

Cite

@article{arxiv.2601.00968,
  title  = {Explainability-Guided Defense: Attribution-Aware Model Refinement Against Adversarial Data Attacks},
  author = {Longwei Wang and Mohammad Navid Nayyem and Abdullah Al Rakin and KC Santosh and Chaowei Zhang and Yang Zhou},
  journal= {arXiv preprint arXiv:2601.00968},
  year   = {2026}
}

Comments

8pages,4 figures

R2 v1 2026-07-01T08:48:59.631Z