Skip to content

wykstc/BIGFUSION

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 

Repository files navigation

BIG-FUSION

This is the official implementation for the paper "BIG-FUSION: Brain-Inspired Global-Local Context Fusion Framework for Multimodal Emotion Recognition in Conversations".

Dependencies

  • Python 3.9.19
  • PyTorch toolbox (2.1.0+cu121)
  • Torch-geometric 2.4.0

Please use the same CUDA version for training.

Dataset

We use the data from paper "A Transformer-based Model with Self-distillation for Multimodal Emotion Recognition in Conversations".

To preprocess the data, please extract multimodal features as described in the paper and use the following preprocessing command:

python preprocess.py --dataset="IEMOCAP"

Usage

Train the model

python train.py --dataset="iemocap" --modalities="atv" --from_begin --epochs=60

Evaluate the model

python eval.py --dataset="iemocap" --modalities="atv"

Citation

If this repository is useful for you, please cite as:

@inproceedings{Wang2025BIGFUSIONBG,
  title={BIG-FUSION: Brain-Inspired Global-Local Context Fusion Framework for Multimodal Emotion Recognition in Conversations},
  author={Yusong Wang and Xuanye Fang and Huifeng Yin and Dongyuan Li and Guoqi Li and Qi Xu and Yi Xu and Shuai Zhong and Mingkun Xu},
  booktitle={AAAI Conference on Artificial Intelligence},
  year={2025},
}

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages