An open source project from Data to AI Lab at MIT.
Fibber is a library about text paraphrase methods. Text paraphrasing has sevaral applications such as adversarial attack and style transfer.
- Documentation: https://DAI-Lab.github.io/fibber
- GitHub: https://github.com/DAI-Lab/fibber
- Adversarial Attack: In Fibber, we provide built-in datasets, attack methods, defense method, evaluation metrics and a benchmark pipeline.
- Text Style Transfer: In Fibber, we provide built-in datasets, style transfer methods, evaluation metrics and a benchmark pipeline.
No matter how much experience you have on natural language processing and adversarial attack, we encourage you to try the demo. Our demo is running on colab, so you can try it without install!
This colab will automatically download a sentiment classifier, and all required resources. When resources are downloaded, you can type in your own sentences, and use Fibber to rewrite it. You can read the rewritten sentences, and metric evaluation of rewritten sentence. You will see some rewritten sentences have the same meaning as your input but get misclassified by the classifier.
fibber has been developed and tested on Python 3.6, 3.7 and 3.8
Also, although it is not strictly required, the usage of conda is highly recommended to avoid interfering with other software installed in the system in which fibber is run.
These are the minimum commands needed to create a conda environment using python3.6 for fibber:
# First you should install conda.
conda create -n fibber_env python=3.6
Afterward, you have to execute this command to activate the environment:
conda activate fibber_env
Then you should install tensorflow and pytorch. Please follow the instructions for tensorflow and pytorch. Fibber requires tensorflow>=2.0.0
and pytorch>=1.5.0
. Please choose a proper version of tensorflow and pytorch according to the CUDA version on your computer.
Remember to execute conda activate fibber_env
every time you start a new console to work on fibber!
Install Java Please install a Java runtime environment on your computer.
After creating the conda environment and activating it, we recommend using pip in order to install fibber:
pip install fibber
This will pull and install the latest stable release from PyPI.
If you are using this project for research purpose and want to make changes to the code, you can install all requirements by
git clone git@github.com:DAI-Lab/fibber.git
cd fibber
pip install --requirement requirement.txt
Then you can use fibber by
python -m fibber.datasets.download_datasets
python -m fibber.benchmark.benchmark
In this case, any changes you made on the code will take effect immediately.
With your conda environment activated, you can clone the repository and install it from
source by running make install
on the stable
branch:
git clone git@github.com:DAI-Lab/fibber.git
cd fibber
git checkout stable
make install
Fibber is a library to evaluate different strategies to paraphrase natural language. In this library, we have several built-in paraphrasing strategies. We also have a benchmark framework to evaluate the quality of paraphrase. In particular, we use the BERT language model to measure how meaningful is the paraphrased text. We use a Universal Sentence Encoder to evaluate the semantic similarity between original and paraphrased text. We also train a BERT classifier on the original dataset, and check of paraphrased sentences can break the text classifier.
In this short tutorial, we will guide you through a series of steps that will help you getting started with fibber.
(1) Install Fibber
(2) Get a demo dataset and resources.
from fibber.datasets import get_demo_dataset
trainset, testset = get_demo_dataset()
from fibber.resources import download_all
# resources are downloaded to ~/.fibber
download_all()
(3) Create a Fibber object.
from fibber.fibber import Fibber
# args starting with "asrs_" are hyperparameters for the ASRSStrategy.
arg_dict = {
"use_gpu_id": 0,
"gpt2_gpu_id": 0,
"transformer_clf_gpu_id": 0,
"strategy_gpu_id": 0,
"rr_enforcing_dist": "wpe",
"rr_wpe_threshold": 1.0,
"rr_wpe_weight": 5,
"rr_sim_threshold": 0.95,
"rr_sim_weight": 20,
"rr_ppl_weight": 5,
"rr_sampling_steps": 200,
"rr_clf_weight": 3,
"rr_window_size": 3,
"rr_accept_criteria": "joint_weighted_criteria",
"rr_lm_option": "finetune",
"rr_sim_metric": "USESimilarityMetric",
"rr_early_stop": "half",
}
# create a fibber object.
# This step may take a while (about 1 hour) on RTX TITAN, and requires 20G of
# GPU memory. If there's not enough GPU memory on your GPU, consider assign use
# gpt2, bert, and strategy to different GPUs.
#
fibber = Fibber(arg_dict, dataset_name="demo", strategy_name="ASRSStrategy",
trainset=trainset, testset=testset, output_dir="exp-demo")
(4) You can also ask fibber to paraphrase your sentence.
The following command can randomly paraphrase the sentence into 5 different ways.
# Try sentences you like.
# label 0 means negative, and 1 means positive.
fibber.paraphrase(
{"text0": ("The Avengers is a good movie. Although it is 3 hours long, every scene has something to watch."),
"label": 1},
field="text0",
n=5)
The output is a tuple of (str, list, list).
# Original Text
'The Avengers is a good movie. Although it is 3 hours long, every scene has something to watch.'
# 5 paraphrase_list
['the avengers is a good movie. even it is 2 hours long, there is not enough to watch.',
'the avengers is a good movie. while it is 3 hours long, it is still very watchable.',
'the avengers is a good movie and although it is 2 ¹⁄₂ hours long, it is never very interesting.',
'avengers is not a good movie. while it is three hours long, it is still something to watch.',
'the avengers is a bad movie. while it is three hours long, it is still something to watch.']
# Evaluation metrics of these 5 paraphrase_list.
{'EditingDistance': 8,
'USESimilarityMetric': 0.9523628950119019,
'GloVeSimilarityMetric': 0.9795315341042675,
'GPT2PerplexityMetric': 1.492070198059082,
'BertClassifier': 0},
{'EditingDistance': 9,
'USESimilarityMetric': 0.9372092485427856,
'GloVeSimilarityMetric': 0.9575780832312993,
'GPT2PerplexityMetric': 0.9813404679298401,
'BertClassifier': 1},
{'EditingDistance': 11,
'USESimilarityMetric': 0.9265919327735901,
'GloVeSimilarityMetric': 0.9710499628056698,
'GPT2PerplexityMetric': 1.325406551361084,
'BertClassifier': 0},
{'EditingDistance': 7,
'USESimilarityMetric': 0.8913971185684204,
'GloVeSimilarityMetric': 0.9800737898362042,
'GPT2PerplexityMetric': 1.2504483461380005,
'BertClassifier': 1},
{'EditingDistance': 8,
'USESimilarityMetric': 0.9124080538749695,
'GloVeSimilarityMetric': 0.9744155151490856,
'GPT2PerplexityMetric': 1.1626977920532227,
'BertClassifier': 0}]
(5) You can ask fibber to randomly pick a sentence from the dataset and paraphrase it.
fibber.paraphrase_a_random_sentence(n=5)
In this version, we implement three strategies
- IdentityStrategy:
- The identity strategy outputs the original text as its paraphrase.
- This strategy generates exactly 1 paraphrase for each original text regardless of
--num_paraphrases_per_text
flag.
- RandomStrategy:
- The random strategy outputs the random shuffle of words in the original text.
- TextAttackStrategy:
- We create a wrapper around TextAttack. To use TextAttack, run
pip install textattack
first.
- We create a wrapper around TextAttack. To use TextAttack, run
- RewriteRollbackStrategy:
- The implementation for R&R
If you use Fibber, please cite the following work:
- Lei Xu, Kalyan Veeramachaneni. Attacking Text Classifiers via Sentence Rewriting Sampler.
@article{xu2021attacking,
title={Attacking Text Classifiers via Sentence Rewriting Sampler},
author={Xu, Lei and Veeramachaneni, Kalyan},
journal={arXiv preprint arXiv:2104.08453},
year={2021}
}
For more details about fibber and all its possibilities and features, please check the documentation site.