A BERT based dual-channel explainable text emotion recognition system

被引:67
作者
Kumar, Puneet [1 ]
Raman, Balasubramanian [1 ]
机构
[1] Indian Inst Technol Roorkee, Dept Comp Sci & Engn, Roorkee, Uttar Pradesh, India
关键词
Emotion recognition; Natural language processing; Explainable AI; Deep neural network explainability; MODEL;
D O I
10.1016/j.neunet.2022.03.017
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
In this paper, a novel dual-channel system for multi-class text emotion recognition has been proposed, and a novel technique to explain its training & predictions has been developed. The architecture of the proposed system contains the embedding module, dual-channel module, emotion classification module, and explainability module. The embedding module extracts the textual features from the input sentences in the form of embedding vectors using the pre-trained Bidirectional Encoder Representations from Transformers (BERT) model. Then the embedding vectors are fed as the inputs to the dual-channel network containing two network channels made up of convolutional neural network (CNN) and bidirectional long short term memory (BiLSTM) network. The intuition behind using CNN and BiLSTM in both the channels was to harness the goodness of the convolutional layer for feature extraction and the BiLSTM layer to extract text's order and sequence-related information. The outputs of both channels are in the form of embedding vectors which are concatenated and fed to the emotion classification module. The proposed system's architecture has been determined by thorough ablation studies, and a framework has been developed to discuss its computational cost. The emotion classification module learns and projects the emotion embeddings on a hyperplane in the form of clusters. The proposed explainability technique explains the training and predictions of the proposed system by analyzing the inter & intra-cluster distances and the intersection of these clusters. The proposed approach's consistent accuracy, precision, recall, and F1 score results for ISEAR, Aman, AffectiveText, and EmotionLines datasets, ensure its applicability to diverse texts.(C)& nbsp;& nbsp;2022 Elsevier Ltd. All rights reserved.
引用
收藏
页码:392 / 407
页数:16
相关论文
共 75 条
[11]   Lexicon based feature extraction for emotion text classification [J].
Bandhakavi, Anil ;
Wiratunga, Nirmalie ;
Padmanabhan, Deepak ;
Massie, Stewart .
PATTERN RECOGNITION LETTERS, 2017, 93 :133-142
[12]   Mining Social Emotions from Affective Text [J].
Bao, Shenghua ;
Xu, Shengliang ;
Zhang, Li ;
Yan, Rong ;
Su, Zhong ;
Han, Dingyi ;
Yu, Yong .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2012, 24 (09) :1658-1670
[13]   Semantic-Emotion Neural Network for Emotion Recognition From Text [J].
Batbaatar, Erdenebileg ;
Li, Meijing ;
Ryu, Keun Ho .
IEEE ACCESS, 2019, 7 :111866-111878
[14]   Explainable Machine Learning in Deployment [J].
Bhatt, Umang ;
Xiang, Alice ;
Sharma, Shubham ;
Weller, Adrian ;
Taly, Ankur ;
Jia, Yunhan ;
Ghosh, Joydeep ;
Puri, Ruchir ;
Moura, Jose M. F. ;
Eckersley, Peter .
FAT* '20: PROCEEDINGS OF THE 2020 CONFERENCE ON FAIRNESS, ACCOUNTABILITY, AND TRANSPARENCY, 2020, :648-657
[15]  
Buechel S., 2018, ARXIV PREPRINT ARXIV
[16]  
Chaffar S, 2011, LECT NOTES ARTIF INT, V6657, P62, DOI 10.1007/978-3-642-21043-3_8
[17]  
Cornelius C., 2018, arXiv preprint, arXiv:1802.08379
[18]   Emotion recognition in human-computer interaction [J].
Cowie, R ;
Douglas-Cowie, E ;
Tsapatsoulis, N ;
Votsis, G ;
Kollias, S ;
Fellenz, W ;
Taylor, JG .
IEEE SIGNAL PROCESSING MAGAZINE, 2001, 18 (01) :32-80
[19]  
Danilevsky M, 2020, 1ST CONFERENCE OF THE ASIA-PACIFIC CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS AND THE 10TH INTERNATIONAL JOINT CONFERENCE ON NATURAL LANGUAGE PROCESSING (AACL-IJCNLP 2020), P447
[20]  
Devlin J, 2019, 2019 CONFERENCE OF THE NORTH AMERICAN CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS: HUMAN LANGUAGE TECHNOLOGIES (NAACL HLT 2019), VOL. 1, P4171