yyht/acl2019-odee

This is the code for our ACL 2019 paper "Open Domain Event Extraction Using Neural Latent Variable Models"

Open Domain Event Extraction Using Neural Latent Variable Models (ODEE)

This is the python3 code for the paper "Open Domain Event Extraction Using Neural Latent Variable Models" in ACL 2019.

Prepare ELMo model

Modify the Line 24 and 25 in cache_features.py.

The fine-tune process need 2 * GTX 1080Ti, if the fine-tune process is costly or somehow failed to complete, please use the initial parameters in allennlp.

Please note that it is optional to finetune the ELMo model if you just want to complete the whole procedure or use the model in somewhere else.

Prepare Data and Train Model

The data is HERE.

1. run the dataprocessor
2. run sudo chown [YOUR_UERS] [PROCESSED_DIR] and specify the directories in setting.yaml manually
3. run pip install -r requirements.txt to install required packages
4. run python cache_features.py
5. run python train_avitm.py
6. run python generate_slot_topN.py
7. run python decode.py
8. run cd slotcoherence && ./run-oc.sh
9. run visualize_test.ipynb

Produced Data

1. *.json.pt: cached features of ODEE input
2. *.json.answer: decoded full results of a news group
3. *.json.template: decoded template of a news group
4. *.json.events.topN: decoded top-N events of a news group
5. *.json.labeled: labeled events of test split
6. slotcoherence/slot_head_words.txt: generated topN head words for each slot

Cite

Please cite our ACL 2019 paper:

@inproceedings{DBLP:conf/acl/LiuHZ19,
  author    = {Xiao Liu and
               Heyan Huang and
               Yue Zhang},
  title     = {Open Domain Event Extraction Using Neural Latent Variable Models},
  booktitle = {Proceedings of the 57th Conference of the Association for Computational
               Linguistics, {ACL} 2019, Florence, Italy, July 28- August 2, 2019,
               Volume 1: Long Papers},
  pages     = {2860--2871},
  year      = {2019},
  crossref  = {DBLP:conf/acl/2019-1},
  url       = {https://www.aclweb.org/anthology/P19-1276/},
  timestamp = {Wed, 31 Jul 2019 17:03:52 +0200},
  biburl    = {https://dblp.org/rec/bib/conf/acl/LiuHZ19},
  bibsource = {dblp computer science bibliography, https://dblp.org}
}