Adversarial Training for Natural Language Understanding
./huggingface-transformers/respectively. We also integrated our implementations of vanilla PGD, FreeAT and YOPO in our fairseq version. FreeLB is an adversarial training approach for improving transformer-based language models on Natural Language Understanding tasks. It accumulates the gradient in the ascent steps and updates the parameters with the accumulated gradients, which is approximately equivalent to enlarging the batch size with diversified adversarial examples within different radiuses around the clean input. FreeLB improves the performance of BERT and RoBERTa on various Natural Language Understanding tasks including Question Answering, Natural Language Inference, and Sentiment Analysis.
For technical details and additional experimental results, please refer to our paper:
Chen Zhu, Yu Cheng, Zhe Gan, Siqi Sun, Tom Goldstein, and Jingjing Liu. FreeLB: Enhanced Adversarial Training for Language Understanding. In ICLR, 2020.
Feb 15, 2020: Initial release of FreeLB based on fairseq and HuggingFace's transformers. The first one contains our implementations of FreeLB, FreeAT, YOPO for RoBERTa, while the latter one is FreeLB for ALBERT.
May 16, 2020: Hyperparameters for ALBERT are now available at
The code is compatible with PyTorch 1.4.0. In addition, you need to execute the followings in order, to install the prerequisites for fairseq and HuggingFace's transformers: ```
git clone https://github.com/NVIDIA/apex cd apex pip install -v --no-cache-dir --global-option="--cppext" --global-option="--cudaext" ./
cd ../fairseq-RoBERTa pip install --editable .
wget https://gist.githubusercontent.com/W4ngatang/60c2bdb54d156a41194446737ce03e2e/raw/17b8dd0d724281ed7c3b2aeeda662b92809aadd5/downloadgluedata.py python downloadgluedata.py --datadir gluedata --tasks all source ./examples/roberta/preprocessGLUEtasks.sh glue_data ALL
cd ../huggingface-transformers pip install --editable . mkdir logs ```
The launch scripts are under
./huggingface-transformers/launch/, where we have included most of the running scripts for RoBERTa and ALBERT on GLUE dev sets. We will release more details in the future.