TextAttack Basic Functions

Welcome to the documentation for TextAttack!

What is TextAttack?

TextAttack is a Python framework for adversarial attacks, adversarial training, and data augmentation in NLP.

TextAttack makes experimenting with the robustness of NLP models seamless, fast, and easy. It’s also useful for NLP model training, adversarial training, and data augmentation.

TextAttack provides components for common NLP tasks like sentence encoding, grammar-checking, and word replacement that can be used on their own.

Where should I start?

This is a great question, and one we get a lot. First of all, almost everything in TextAttack can be done in two ways: via the command-line or via the Python API. If you’re looking to integrate TextAttack into an existing project, the Python API is likely for you. If you’d prefer to use built-in functionality end-to-end (training a model, running an adversarial attack, augmenting a CSV) then you can just use the command-line API.

TextAttack does three things very well:

  1. Adversarial attacks (Python: textattack.Attack, Bash: textattack attack)

  2. Data augmentation (Python: textattack.augmentation.Augmenter, Bash: textattack augment)

  3. Model training (Python: textattack.Trainer, Bash: textattack train)

Adversarial training can be achieved as a combination of [1] and/or [2] with [3] (via textattack train --attack). To see all this in action, see the TextAttack End-to-End tutorial.

All of the other components: datasets, models & model wrappers, loggers, transformations, constraints, search methods, goal functions, etc., are developed to support one or more of these three functions. Feel free though to install textattack to include just one of those components! (For example, TextAttack provides a really easy Python interface for accessing and using word embeddings that will automatically download and save them on the first use.)

NLP Attacks

TextAttack provides a framework for constructing and thinking about generating inputs in NLP via perturbation attacks.

TextAttack builds attacks from four components:

  • Goal Functions: stipulate the goal of the attack, like to change the prediction score of a classification model, or to change all of the words in a translation output.

  • Constraints: determine if a potential perturbation is valid with respect to the original input.

  • Transformations: take a text input and transform it by inserting and deleting characters, words, and/or phrases.

  • Search Methods: explore the space of possible transformations within the defined constraints and attempt to find a successful perturbation which satisfies the goal function.

TextAttack provides a set of Attack Recipes that assemble attacks from the literature from these four components. Take a look at these recipes (or our paper on ArXiv) to get a feel for how the four components work together to create an adversarial attack.

Data Augmentation

Data augmentation is easy and extremely common in computer vision but harder and less common in NLP. We provide a Data Augmentation module using transformations and constraints.

Features

TextAttack has some other features that make it a pleasure to use: