Skip to content

Repository files navigation

SynPG

Code for our EACL-2021 paper "Generating Syntactically Controlled Paraphrases without Using Annotated Parallel Pairs".

If you find that the code is useful in your research, please consider citing our paper.

@inproceedings{Huang2021synpg,
author = {Kuan-Hao Huang and Kai-Wei Chang},
title = {Generating Syntactically Controlled Paraphrases without Using Annotated Parallel Pairs},
booktitle = {Proceedings of the Conference of the European Chapter of the Association for Computational Linguistics (EACL)},
year = {2021},
}

Setup

  • Python=3.7.10
$ pip install -r requirements.txt

Pretrained Models

Demo

python generate.py \
--synpg_model_path ./model/pretrained_synpg.pt \
--pg_model_path ./model/pretrained_parse_generator.pt \
--input_path ./demo/input.txt \
--output_path ./demo/output.txt \
--bpe_codes_path ./data/bpe.codes \
--bpe_vocab_path ./data/vocab.txt \
--bpe_vocab_thresh 50 \
--dictionary_path ./data/dictionary.pkl \
--max_sent_len 40 \
--max_tmpl_len 100 \
--max_synt_len 160 \
--temp 0.5 \
--seed 0

Training

  • Download data and put them under ./data/
  • Download glove.840B.300d.txt and put it under ./data/
  • Run scripts/train_synpg.sh or the following command to train SynPG
python train_synpg.py \
--model_dir ./model \
--output_dir ./output \
--bpe_codes_path ./data/bpe.codes \
--bpe_vocab_path ./data/vocab.txt \
--bpe_vocab_thresh 50 \
--dictionary_path ./data/dictionary.pkl \
--train_data_path ./data/train_data.h5 \
--valid_data_path ./data/valid_data.h5 \
--emb_path ./data/glove.840B.300d.txt \
--max_sent_len 40 \
--max_synt_len 160 \
--word_dropout 0.4 \
--n_epoch 5 \
--batch_size 64 \
--lr 1e-4 \
--weight_decay 1e-5 \
--log_interval 250 \
--gen_interval 5000 \
--save_interval 10000 \
--temp 0.5 \
--seed 0
  • Run scripts/train_parse_generator.sh or the following command to train the parse generator
python train_parse_generator.py \
--model_dir ./model \
--output_dir ./output_pg \
--dictionary_path ./data/dictionary.pkl \
--train_data_path ./data/train_data.h5 \
--valid_data_path ./data/valid_data.h5 \
--max_sent_len 40 \
--max_tmpl_len 100 \
--max_synt_len 160 \
--word_dropout 0.2 \
--n_epoch 5 \
--batch_size 32 \
--lr 1e-4 \
--weight_decay 1e-5 \
--log_interval 250 \
--gen_interval 5000 \
--save_interval 10000 \
--temp 0.5 \
--seed 0

Evaluating

  • Download testing data and put them under ./data/
  • Run scripts/eval.sh or the following command to evaluate SynPG
python eval_generate.py \
--test_data ./data/test_data_mrpc.h5 \
--dictionary_path ./data/dictionary.pkl \
--model_path ./model/pretrained_synpg.pt \
--output_dir ./eval/ \
--bpe_codes ./data/bpe.codes \
--bpe_vocab ./data/vocab.txt \
--bpe_vocab_thresh 50 \
--max_sent_len 40 \
--max_synt_len 160 \
--word_dropout 0.0 \
--batch_size 64 \
--temp 0.5 \
--seed 0 \
python eval_calculate_bleu.py --ref ./eval/target_sents.txt --input ./eval/outputs.txt

The BLEU scores should be similar to the following.

MRPCPANQuora
SynPG26.227.333.2
SynPG-Large36.227.134.7

Fine-Tuning

One main advantage of SynPG is that SynPG learns the paraphrase generation model without using any paraphrase pairs. Therefore, it is possible to fine-tune SynPG with the texts (without using the ground truth paraphrases) in the target domain when those texts are available. This fine-tuning step would significantly improve the quality of paraphrase generation in the target domain, as shown in our paper.

  • Download testing data and put them under ./data/
  • Run scripts/finetune_synpg.sh or the following command to finetune SynPG
python finetune_synpg.py \
--model_dir ./model_finetune \
--model_path ./model/pretrained_synpg.pt \
--output_dir ./output_finetune \
--bpe_codes_path ./data/bpe.codes \
--bpe_vocab_path ./data/vocab.txt \
--bpe_vocab_thresh 50 \
--dictionary_path ./data/dictionary.pkl \
--train_data_path ./data/test_data_mrpc.h5 \
--valid_data_path ./data/test_data_mrpc.h5 \
--max_sent_len 40 \
--max_synt_len 160 \
--word_dropout 0.4 \
--n_epoch 50 \
--batch_size 64 \
--lr 1e-4 \
--weight_decay 1e-5 \
--log_interval 250 \
--gen_interval 5000 \
--save_interval 10000 \
--temp 0.5 \
--seed 0

We can observe the significant improvement on BLEU scores from the table below.

MRPCPANQuora
SynPG26.227.333.2
SynPG-Large36.227.134.7
SynPG-Fine-Tune48.737.749.8

Author

Kuan-Hao Huang / @ej0cl6

About

Code for our EACL-2021 paper "Generating Syntactically Controlled Paraphrases without Using Annotated Parallel Pairs".

Topics

Resources

Stars

38 stars

Watchers

7 watching

Forks

Releases

Packages

Used by

Contributors

Languages