PriMD

<h1 align="center">[PriMD] Modality Disentangled Learning for Incomplete Multimodal Emotion Recognition: A Primitive Memory Distillation Perspective</h1> <p align="center"> <b>Jiaqi Zhang</b>, Zheng Pang, Mengting Li, Yiqi Wang, Guangyuan Dong, Chao Xue, Yusen Wu, Zihao Li, Huy Phan, Sicheng Zhao, Björn Schuller, Jiachen Luo </p> <div align="center"> <a href="https://arxiv.org/abs/2608.30563"><img src="https://img.shields.io/badge/Paper-arXiv.2608.30563-B31B1B?logo=arxiv&style=for-the-badge" alt="Paper" /></a> &nbsp;&nbsp; <a href="https://jiaqizhang-sengoku.github.io/PriMD/"><img src="https://img.shields.io/badge/Project_Page-Website-1081C1?logo=googlechrome&style=for-the-badge" alt="Project Page" /></a> </div>

📌 Abstract

Multimodal Emotion Recognition (MER) often suffers from missing modalities in real-world scenarios. Existing methods usually generate, align, or distill missing modalities as a whole, overlooking the heterogeneity of information components within each modality. Such holistic treatment mixes inferable shared semantics with uncertain modality-specific details, causing unstable representations and weakening robustness. To address this issue, we propose the Primitive Memory Distillation (PriMD) framework. Unlike existing methods, PriMD takes an intra-modal perspective and focuses on the differences among information components within each modality. PriMD first disentangles cross-modal shared semantics from modality-specific representations, and then discretizes the latter into learnable semantic prototypes to construct modality-specific memory banks. When modalities are missing, the student uses the shared semantics of visible modalities as queries to dynamically retrieve prototypes. It compensates for missing modality-specific information within a constrained memory space and aligns with the teacher. Extensive experiments on IEMOCAP, CMU-MOSI, and CMU-MOSEI demonstrate that PriMD achieves state-of-the-art robustness across a wide range of missing-modality settings, while mitigating the instability caused by holistic feature inference.

🎇 Method Overview

<p align="center"> <img width="900" alt="PriMD Architecture" src="./Figures/PriMD.png"> </p>

💡 Key Features

• We propose PriMD, a framework that starts from the internal information structure of modalities and formulates incomplete MER as a unified process of shared semantic estimation and constrained compensation of modality-specific information.
• PriMD includes SSSD for separating shared and specific features, as well as DPMC and DRAD for discretizing the feature space and addressing missing-modality compensation.
• Extensive experiments show that PriMD mitigates the instability caused by holistic representation and outperforms existing methods under various modality-missing scenarios.

🚀 Installation & Usage

This repository releases the PriMD source code. Datasets, pre-extracted features, and trained checkpoints are not included.

1. Environment

cd PriMD

# Optional: activate your own environment first.
# conda activate <your-env>

pip install -r requirements.txt

2. Dataset Layout

PriMD uses pre-extracted utterance-level multimodal features:

dataset/
|-- IEMOCAP/
|-- CMUMOSI/
`-- CMUMOSEI/

Default feature configuration:

3. Quick Start

Run the IEMOCAP four-class experiment:

sh run_PriMD_iemocap4.sh

Run CMU-MOSI or CMU-MOSEI:

sh run_PriMD_cmumosi.sh
sh run_PriMD_cmumosei.sh

Run a single missing-modality condition manually:

python -u PriMD/train_PriMD.py \
  --dataset=IEMOCAPFour \
  --audio-feature=wav2vec-large-c-UTT \
  --text-feature=deberta-large-4-UTT \
  --video-feature=manet_UTT \
  --seed=66 \
  --batch-size=16 \
  --epoch=200 \
  --lr=0.0001 \
  --hidden=256 \
  --depth=4 \
  --num_heads=2 \
  --drop_rate=0.5 \
  --attn_drop_rate=0.0 \
  --test_condition=a \
  --stage_epoch=100 \
  --gpu=0

Training Outputs

Training logs and checkpoints are saved under:

saved/
|-- log/main_result/
`-- model/main_result/

📝 References

If you find the code useful for your research, please consider citing:

@article{PriMD,
  author  = {Jiaqi Zhang and Zheng Pang and Mengting Li and Yiqi Wang and Guangyuan Dong and Chao Xue and Yusen Wu and Zihao Li and Huy Phan and Sicheng Zhao and Björn W. Schuller and Jiachen Luo},
  title   = {Modality Disentangled Learning for Incomplete Multimodal Emotion Recognition: A Primitive Memory Distillation Perspective},
  journal = {arXiv preprint arXiv:2608.30563},
  year    = {2026}
}

📢 LICENSE

The project is released under the MIT License.