/GNN_ERC

This for GNN project team

Primary LanguagePython

Emotion Recognition In Conversation with
Transformer and Relational Graph Transformer

In this project, based on the DialogueGCN [1], we proposed new flows the have better performance than the original paper.

In the DialogGCN, their flow is as show in the below image

GCN drawio

In our methods, we changed the sequence context to Transformer, and RGCN to Relational Graph Transform, which better generalize and have global information.

RGT

In our methods, Relational Graph Tranformer having the same properties as Graph Transformer Architecture [2] but include the information of the relation after multiheader layer

image

For the coding, we use DGL library [3]

If you want to check the process of preprocessing data, extract the raw data in the folder and check change the path in sysconf and run the script in preprocessing folder

If you want to recheck the training without preprocessing steps, please download the full preprocessing data as the link above, extract and change the path in sysconf folder. After that enter our_method folder or base_line folder and try to run the code.

Reference

[1] Deepanway Ghosal, Navonil Majumder, Soujanya Poria, Niyati Chhaya, and Alexander Gelbukh. Dialoguegcn: A graph convolutional neural network for emotion recognition in conversation. In Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP), pages 154–164, 2019.

[2] @article{dwivedi2021generalization, title={A Generalization of Transformer Networks to Graphs}, author={Dwivedi, Vijay Prakash and Bresson, Xavier}, journal={AAAI Workshop on Deep Learning on Graphs: Methods and Applications}, year={2021} }
[3] https://www.dgl.ai/