Skip to content
/ IMDer Public

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

License

Notifications You must be signed in to change notification settings

mdswyz/IMDer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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}
}

About

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

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages