OTE-MTL

OTE-MTL - Multi-Task Learning for Opinion Triplet Extraction

• Code and preprocessed dataset for EMNLP 2020 Findings paper titled "A Multi-task Learning Framework for Opinion Triplet Extraction"
• Chen Zhang, Qiuchi Li, Dawei Song and Benyou Wang.

Updates

• Feb. 20th, 2021. As is pointed out in our paper, we have noted that datav1 used in https://arxiv.org/abs/1911.01616 is rather incomplete and have corrected their mistakes. That is, the data used for our experiments is similar to datav2. However, as is requested by some users and in case of any inconsistencies between our data and datav2, we decide to support the test of our model on datav2. You could just run our model on datav2 with just an additional argument --v2.

Requirements

• Python 3.6
• PyTorch 1.0.0
• numpy 1.15.4

Usage

• Download pretrained GloVe embeddings with this link and extract glove.840B.300d.txt into glove/.
• Train with command, optional arguments could be found in train.py, --v2 denotes whether test on datav2

python train.py --model mtl --dataset rest14 [--v2]

• Infer with infer.py

Task

An overview of the task opinion triplet extraction (OTE) is given below

OTE is solving the same task proposed in https://arxiv.org/abs/1911.01616. While our work focuses on extracting (aspect term, opinion term, sentiment) opinion triplets (OTs), they extract (aspect term-sentiment pair, opinion term)s. Owing to the minor difference lying in formulations, two drawbacks in the latter formulation are presented: (i) sentiments are determined without accessing opinion terms, (ii) conflicting opinions expressed towards an aspect cannot be predicted.

Citation

If you use the code in your paper, please kindly star this repo and cite our paper

@inproceedings{zhang-etal-2020-multi,
    title = "A Multi-task Learning Framework for Opinion Triplet Extraction",
    author = "Zhang, Chen  and
      Li, Qiuchi  and
      Song, Dawei  and
      Wang, Benyou",
    booktitle = "Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing: Findings",
    month = nov,
    year = "2020",
    address = "Online",
    publisher = "Association for Computational Linguistics",
    url = "https://www.aclweb.org/anthology/2020.findings-emnlp.72",
    pages = "819--828",
}

Contact

• For any issues or suggestions about this work, don't hesitate to create an issue or directly contact me via [email protected] !