xvshiting / ReCO

ReCO: A Large Scale Chinese Reading Comprehension Dataset on Opinion

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ReCO

ReCO: A Large Scale Chinese Reading Comprehension Dataset on Opinion

Data

Dataset is available at https://drive.google.com/drive/folders/1rOAoKcLhMhge9uVQFM2_D1EU0AjnpWFa?usp=sharing

download the data and put the json files to the data/ReCO directory

Stats

Train Dev Test-a Test-b
250,000 30,000 10,000 10,000

Requirenments

transformers
torch>=1.3.0
tqdm
joblib
apex(for mixed-precision training)

Train and Test

For BiDAF and other types of model, you can go to the BiDAF folder and run. But the result is somewhat low _

Pre-training methods finetuning:

For single node training:
python3 train.py --model_type=bert-base-chinese
for multiple nodes distributed training:
python3 -m torch.distributed.launch --nproc_per_node=8 train.py --model_type=bert-base-chinese

If you want to use the original doc as the context, you can set the clean(one['passage']) in prepare_data.py line 29 to clean(one['doc']).

model card

Model Name Model Type Model Size Paper
Bert-base bert-base-chinese 102m BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding
RoBerta-large clue/roberta_chinese_large 325m RoBERTa: A Robustly Optimized BERT Pretraining Approach
ALBERT-tiny voidful/albert_chinese_tiny 4.1m ALBERT: A Lite BERT for Self-supervised Learning of Language Representations
ALBERT-base voidful/albert_chinese_base 10.5m -
ALBERT-xxlarge voidful/albert_chinese_xxlarge 221m -

Test

python3 test.py --model_type=bert-base-chinese

Results

Doc level

Model Dev Test-a
BiDAF 55.8 56.4
Bert-Base 61.4 61.1
RoBerta-Large 65.7 65.3
Human -- 88.0

Evidence level

Model Dev Test-a
BiDAF 68.9 68.4
Bert-Base 76.3 77.1
RoBerta-Large 78.7 79.2
ALBert-tiny 70.9 70.4
ALBert-base 76.9 77.3
ALBert-xxLarge 80.8 81.2
Human -- 91.5

Citation

If you use ReCO in your research, please cite our work with the following BibTex Entry

@inproceedings{DBLP:conf/aaai/WangYZXW20,
  author    = {Bingning Wang and
               Ting Yao and
               Qi Zhang and
               Jingfang Xu and
               Xiaochuan Wang},
  title     = {ReCO: {A} Large Scale Chinese Reading Comprehension Dataset on Opinion},
  booktitle = {The Thirty-Fourth {AAAI} Conference on Artificial Intelligence, {AAAI}
               2020, The Thirty-Second Innovative Applications of Artificial Intelligence
               Conference, {IAAI} 2020, The Tenth {AAAI} Symposium on Educational
               Advances in Artificial Intelligence, {EAAI} 2020, New York, NY, USA,
               February 7-12, 2020},
  pages     = {9146--9153},
  publisher = {{AAAI} Press},
  year      = {2020},
  url       = {https://aaai.org/ojs/index.php/AAAI/article/view/6450},
  timestamp = {Thu, 04 Jun 2020 13:18:48 +0200},
  biburl    = {https://dblp.org/rec/conf/aaai/WangYZXW20.bib},
  bibsource = {dblp computer science bibliography, https://dblp.org}
}

About

ReCO: A Large Scale Chinese Reading Comprehension Dataset on Opinion


Languages

Language:Python 100.0%