This repository provides the BMGF-RoBERTa model, which is used in the implicit discourse relation classification.
Model | PDTB-4 (F1) | PDTB-4 (Acc) | PDTB-11 (Acc) | CoNLL-Test (Acc) | CoNLL-Blind (Acc) |
---|---|---|---|---|---|
Previous SOTAs | 52.89 | 60.63 | 53.23 | 40.91 | 40.12 |
BMGF-RoBERTa | 63.39 | 69.06 | 58.13 | 57.26 | 55.19 |
Performance of multi-class classification on PDTB and CoNLL-2016 in terms of accuracy (Acc) (%) and macro-averaged F1 (F1) (%).
Model | Comp. | Cont. | Exp. | Temp. |
---|---|---|---|---|
Previous SOTAs | 47.85 | 58.96 | 72.47 | 45.61 |
BMGF-RoBERTa | 59.44 | 60.98 | 77.66 | 50.26 |
Performance of multiple binary classification on the top level classes in PDTB in terms of F1 (%).
- tqdm
- numpy
- pandas
- scipy
- torch >= 1.3.0
- transformers >= 2.1.1
- stanfordnlp >= 0.2.0
Due to the LDC policy, we cannot release the PDTB data and CoNLL test and blind data. If you have bought data from LDC, please put the PDTB .csv file from cgpotts/pdtb2 and CoNLL data in data/pdtb and data/conll respectively. Then run following scripts.
python preprocess_pdtb.py \
--csv_file_path ../data/pdtb/pdtb2.csv \
--types Implicit \
--encoder roberta \
--dataset_file_path ../data/pdtb/train_implicit_roberta.pkl \
--sections 2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18,19,20
python preprocess_pdtb.py \
--csv_file_path ../data/pdtb/pdtb2.csv \
--types Implicit \
--encoder roberta \
--dataset_file_path ../data/pdtb/valid_implicit_roberta.pkl \
--sections 0,1
python preprocess_pdtb.py \
--csv_file_path ../data/pdtb/pdtb2.csv \
--types Implicit \
--encoder roberta \
--dataset_file_path ../data/pdtb/test_implicit_roberta.pkl \
--sections 21,22
python preprocess_conll.py \
--json_file_path ../data/conll/2015-2016_conll_shared_task/data/conll16st-en-03-29-16-train/relations.json \
--types Implicit \
--encoder roberta \
--dataset_file_path ../data/conll/train_implicit_roberta.pkl
python preprocess_conll.py \
--json_file_path ../data/conll/2015-2016_conll_shared_task/data/conll16st-en-03-29-16-dev/relations.json \
--types Implicit \
--encoder roberta \
--dataset_file_path ../data/conll/valid_implicit_roberta.pkl
python preprocess_conll.py \
--json_file_path ../data/conll/2015-2016_conll_shared_task/data/conll16st-en-03-29-16-test/relations.json \
--types Implicit \
--encoder roberta \
--dataset_file_path ../data/conll/test_implicit_roberta.pkl
python preprocess_conll.py \
--json_file_path ../data/conll/2015-2016_conll_shared_task/data/conll15st-en-03-29-16-blind-test/relations.json \
--types Implicit \
--encoder roberta \
--dataset_file_path ../data/conll/blind_implicit_roberta.pkl
You can train models in different settings, including 4-way or 11-way classfications on PDTB or the binary classification for each relation. For the multi-label classification, please use the train.py; for the binary classification, please use train_binary.py.
To train a 4-way BMGF-RoBERTa, just set the num_rels as four, encoder as roberta, and finetune as type.
python train.py \
--train_dataset_path ../data/pdtb/train_implicit_roberta.pkl \
--valid_dataset_path ../data/pdtb/valid_implicit_roberta.pkl \
--test_dataset_path ../data/pdtb/test_implicit_roberta.pkl \
--save_model_dir ../data/dumps \
--num_rels 4 \
--gpu_ids 0,1 \
--batch_size 32 \
--lr 0.001 \
--weight_decay 0.0005 \
--max_grad_norm 2.0 \
--loss ce \
--encoder roberta \
--finetune type \
--hidden_dim 128 \
--num_perspectives 16 \
--num_filters 64 \
--activation leaky_relu \
--dropout 0.2
Macro-F1 and accuracy are used in discourse relation classification. Some instances were annotated as more than one label. Following previous work and the scorer at conll16st, a prediction is regarded as correct once it matches one of the ground-truth labels. You can check functions in evaluate.py.
The details of this architecture are described in the following paper. If you use this model in your work, please kindly cite it.
@inproceedings{liu2020bmgfroberta,
author = {Xin Liu, Jiefu Ou, Yangqiu Song, Xin Jiang},
title = {On the Importance of Word and Sentence Representation Learning in Implicit Discourse Relation Classification},
booktitle = {Proceedings of the Twenty-Ninth International Joint Conference on Artificial Intelligence, {IJCAI} 2020, July 11-17, 2020, Yokohama, Japan.}
}