IMDer icon indicating copy to clipboard operation
IMDer copied to clipboard

An official implementation of "Incomplete Multimodality-Diffused Emotion Recognition" in PyTorch. (NeurIPS 2023)

Incomplete Multimodality-Diffused Emotion Recognition, NeurIPS 2023.

We propose the Incomplete Multimodality-Diffused emotion recognition (IMDer) method that maps input random noise to the distribution space of missing modalities and recovers missing data in accordance with their original distributions.

The Framework of IMDer.

(1) IMDer maps input random noise to the distribution space of missing modalities and recovers missing data in accordance with their original distributions. (2) To minimize the semantic ambiguity between the missing and recovered modalities, IMDer utilize the available modalities as prior conditions to guide and refine the recovering process. Please refer to our paper for details.

Usage

Prerequisites

  • Python 3.8
  • PyTorch 1.9.0
  • CUDA 11.4

Datasets

Data files can be downloaded from here, and you only need to download the aligned data. You can put the downloaded datasets into dataset/ directory.

Pretrained weights

Before running missing cases, you should download the weights pretrained by complete multimodal data (i.e., MR=0.0). You can put the downloaded weights into pt/ directory.

Run the Codes

Running the following command:

python train.py

Citation

If you find the code helpful in your research or work, please cite the following paper.

@inproceedings{wang2023incomplete,
  title={Incomplete Multimodality-Diffused Emotion Recognition},
  author={Wang, Yuanzhi and Li, Yong and Cui, Zhen},
  booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
  year={2023}
}