yuanqinglee / erldk

multimodal emotion recognition model for conversational videos based on reinforcement learning and domain knowledge

ERLDK

multimodal emotion recognition model for conversational videos based on reinforcement learning and domain knowledge

This is the open access code for the paper: Real-Time Video Emotion Recognition based on Reinforcement Learning and Domain Knowledge, which has been published online on April 12, 2021. This paper can be downloaded from https://ieeexplore.ieee.org/document/9400391.

Here's the abstract.

Multimodal emotion recognition in conversational videos (ERC) develops rapidly in recent years. To fully extract the relative context from video clips, most studies build their models on the entire dialogues which make them lack of real-time ERC ability. Different from related researches, a novel multimodal emotion recognition model for conversational videos based on reinforcement learning and domain knowledge (ERLDK) is proposed in this paper. In ERLDK, the reinforcement learning algorithm is introduced to conduct real-time ERC with the occurrence of conversations. The collection of history utterances is composed as an emotion-pair which represents the multimodal context of the following utterance to be recognized. Dueling deep-Q-network (DDQN) based on gated recurrent unit (GRU) layers is designed to learn the correct action from the alternative emotion categories. Domain knowledge is extracted from public dataset based on the former information of emotion-pairs. The extracted domain knowledge is used to revise the results from the RL module and is transformed into other dataset to examine the rationality. The experimental results on datasets show that ERLDK achieves the state-of-the-art results on weighted average and most of the specific emotion categories.

How to use this repository:

#read.py: pre-process the original iemocap database to become the new database which can be used to extract the emotion domain knowledge. In this current file, the window size is set as 3.

#dataloader_1.py: load the original iemocap raw database.

#knowledge.py: use the new database from the read.py to extract the domain knowledge. In this current file, the window size is set as 5.

#dqn_env_iemocap.py: the train file

#dialogue_level_test.py: the dialogue level test file

#pair_datalodoader.py: load the data from the new database from the read.py

#dueling_dqn_model.py: establish the ddqn model.

erldk's Issues

问题

我想问下在训练代码中的用于测试和训练数据是怎么得到的，这代码跑不起来

request

Hello，I am trying to run your code now, can you tell me how to get trainset_pair_new_justtrain_4.pkl ？