• Home
  • Search
  • Adversarial Training for Improving Model Robustness? Look at Both Prediction and Interpretation
  • Cite Icon22
  • https://doi.org/10.1609/aaai.v36i10.21289Copy DOI Icon

Adversarial Training for Improving Model Robustness? Look at Both Prediction and Interpretation

  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Neural language models show vulnerability to adversarial examples which are semantically similar to their original counterparts with a few words replaced by their synonyms. A common way to improve model robustness is adversarial training which follows two steps—collecting adversarial examples by attacking a target model, and fine-tuning the model on the augmented dataset with these adversarial examples. The objective of traditional adversarial training is to make a model produce the same correct predictions on an original/adversarial example pair. However, the consistency between model decision-makings on two similar texts is ignored. We argue that a robust model should behave consistently on original/adversarial example pairs, that is making the same predictions (what) based on the same reasons (how) which can be reflected by consistent interpretations. In this work, we propose a novel feature-level adversarial training method named FLAT. FLAT aims at improving model robustness in terms of both predictions and interpretations. FLAT incorporates variational word masks in neural networks to learn global word importance and play as a bottleneck teaching the model to make predictions based on important words. FLAT explicitly shoots at the vulnerability problem caused by the mismatch between model understandings on the replaced words and their synonyms in original/adversarial example pairs by regularizing the corresponding global word importance scores. Experiments show the effectiveness of FLAT in improving the robustness with respect to both predictions and interpretations of four neural network models (LSTM, CNN, BERT, and DeBERTa) to two adversarial attacks on four text classification tasks. The models trained via FLAT also show better robustness than baseline models on unforeseen adversarial examples across different attacks.

Similar Papers
  • PDF
  • Research Article
  • Citations26

On the Effectiveness of Adversarial Training in Defending against Adversarial Example Attacks for Image Classification

  • Nov 14, 2020
  • Applied Sciences
  • Sanglee Park +1
  • Conference Article
  • Citations304

Parametric Noise Injection: Trainable Randomness to Improve Deep Neural Network Robustness Against Adversarial Attack

  • Jun 01, 2019
  • Zhezhi He +2
  • Conference Article
  • Citations45

Adversarial Attacks on Deep Models for Financial Transaction Records

  • Aug 14, 2021
  • Ivan Fursov +9
  • Research Article
  • Citations5

Provable Unrestricted Adversarial Training Without Compromise With Generalizability.

  • Dec 01, 2024
  • IEEE transactions on pattern analysis and machine intelligence
  • Lilin Zhang +3
  • PDF
  • Research Article
  • Citations1

Improving Model Robustness through Hybrid Adversarial Training: Integrating FGSM and PGD Methods

  • Nov 26, 2024
  • Applied and Computational Engineering
  • Zeshan Zhong
  • Research Article
  • Citations11

Feature autoencoder for detecting adversarial examples

  • May 26, 2022
  • International Journal of Intelligent Systems
  • Hongwei Ye +1
  • Conference Article

Regional Saliency Map Attack for Medical Image Segmentation

  • Oct 16, 2022
  • Pei Yang +1
  • Research Article
  • Citations32

Joint Character-Level Word Embedding and Adversarial Stability Training to Defend Adversarial Text

  • Apr 03, 2020
  • Proceedings of the AAAI Conference on Artificial Intelligence
  • Hui Liu +4
  • Video Transcripts

A Closer Look into the Robustness of Neural Dependency Parsers Using Better Adversarial Examples

  • Aug 01, 2021
  • Underline Science Inc.
  • Zhilin Lei +5
  • Research Article
  • Citations21

Reconstruction-Assisted and Distance-Optimized Adversarial Training: A Defense Framework for Remote Sensing Scene Classification

  • Jan 01, 2023
  • IEEE Transactions on Geoscience and Remote Sensing
  • Yuru Su +5
  • Research Article

Adversarial Example Generation Method Based on Wavelet Transform

  • Feb 10, 2026
  • Information
  • Meng Bi +5
  • Conference Article
  • Citations2

Robust Cross-Modal Retrieval by Adversarial Training

  • Jul 18, 2022
  • Tao Zhang +2
  • Conference Article
  • Citations16

Adversarial Attacks and Defenses in Image Classification: A Practical Perspective

  • Jul 26, 2022
  • Yongkang Chen +3
  • Conference Article
  • Citations1

Exploiting Doubly Adversarial Examples for Improving Adversarial Robustness

  • Oct 16, 2022
  • Junyoung Byun +3
  • Research Article
  • Citations25

Adversarial training for prostate cancer classification using magnetic resonance imaging.

  • Jun 01, 2022
  • Quantitative Imaging in Medicine and Surgery
  • Lei Hu +5
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.