MultiModal-InfoMax
This repository contains the official implementation code of the paper Improving Multimodal Fusion with Hierarchical Mutual Information Maximization for Multimodal Sentiment Analysis, accepted at EMNLP 2021.
Introduction
Multimodal-informax (MMIM) synthesizes fusion results from multi-modality input through a two-level mutual information (MI) maximization. We use BA (Barber-Agakov) lower bound and contrastive predictive coding as the target function to be maximized. To facilitate the computation, we design an entropy estimation module with associated history data memory to facilitate the computation of BA lower bound and the training process.
Usage
-
Download the CMU-MOSI and CMU-MOSEI dataset from Google Drive or Baidu Disk (extraction code: g3m2). Place them under the folder
Multimodal-Infomax/datasets
-
Set up the environment (need conda prerequisite)
conda env create -f environment.yml
conda activate MMIM
- Start training
python main.py --dataset mosi --contrast
Citation
Please cite our paper if you find our work useful for your research:
@article{han2021improving,
title={Improving Multimodal Fusion with Hierarchical Mutual Information Maximization for Multimodal Sentiment Analysis},
author={Han, Wei and Chen, Hui and Poria, Soujanya},
journal={Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP)},
year={2021}
}
Contact
Should you have any question, feel free to contact me through [email protected]