Discriminative Sentence Modeling for Story Ending Prediction

Authors

  • Yiming Cui Harbin Institute of Technology
  • Wanxiang Che Harbin Institute of Technology
  • Wei-Nan Zhang Harbin Institute of Technology
  • Ting Liu Harbin Institute of Technology
  • Shijin Wang iFLYTEK Co., Ltd.
  • Guoping Hu iFLYTEK Co., Ltd.

DOI:

https://doi.org/10.1609/aaai.v34i05.6260

Abstract

Story Ending Prediction is a task that needs to select an appropriate ending for the given story, which requires the machine to understand the story and sometimes needs commonsense knowledge. To tackle this task, we propose a new neural network called Diff-Net for better modeling the differences of each ending in this task. The proposed model could discriminate two endings in three semantic levels: contextual representation, story-aware representation, and discriminative representation. Experimental results on the Story Cloze Test dataset show that the proposed model siginificantly outperforms various systems by a large margin, and detailed ablation studies are given for better understanding our model. We also carefully examine the traditional and BERT-based models on both SCT v1.0 and v1.5 with interesting findings that may potentially help future studies.

Downloads

Published

2020-04-03

How to Cite

Cui, Y., Che, W., Zhang, W.-N., Liu, T., Wang, S., & Hu, G. (2020). Discriminative Sentence Modeling for Story Ending Prediction. Proceedings of the AAAI Conference on Artificial Intelligence, 34(05), 7602-7609. https://doi.org/10.1609/aaai.v34i05.6260

Issue

Section

AAAI Technical Track: Natural Language Processing