Skip to content

Latest commit

 

History

History
56 lines (37 loc) · 1.08 KB

README.md

File metadata and controls

56 lines (37 loc) · 1.08 KB
Raw

adv-def-text-dist

Requirements:

  • Python 3.6+
  • TensorFlow 1.3+
  • spacy
  • transformers
  • tensorflow_hub

Please refer to requirement.txt for detailed packages

Introduction

This is an implementation for the paper: Grey-box Adversarial Attack and Defence for Text. The paper is currently submitted to EMNLP'20.

Run training and test

1, Please download the Yelp review dataset from the official website link.

2, Download the GloVe embeddings and the counter-fitted embeddings using

./download.sh.

3, Run dataset preprecessing using

python yelp_preprocessing.py --data_dir YELP_DATASET_PATH --embed_file GLOVE_EMB_PATH

4, Train target models using the scripts

./train_cls.sh

5, Pre-train Auto-encoder for reconstruction using the scripts

./train_ae.sh.

6, Train adversarial attack/defence models using the scripts (multiple variants of our model are available and commented out in the script)

./train_adv.sh

7, Perform independent test for adversarial attack/defence using

./test_adv.sh