I2MoE: Interpretable Multimodal Interaction-aware Mixture-of-Experts

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Xin, Jiayi, Yun, Sukwon, Peng, Jie, Choi, Inyoung, Ballard, Jenna L., Chen, Tianlong, Long, Qi
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866910967541006336
author Xin, Jiayi
Yun, Sukwon
Peng, Jie
Choi, Inyoung
Ballard, Jenna L.
Chen, Tianlong
Long, Qi
author_facet Xin, Jiayi
Yun, Sukwon
Peng, Jie
Choi, Inyoung
Ballard, Jenna L.
Chen, Tianlong
Long, Qi
contents Modality fusion is a cornerstone of multimodal learning, enabling information integration from diverse data sources. However, vanilla fusion methods are limited by (1) inability to account for heterogeneous interactions between modalities and (2) lack of interpretability in uncovering the multimodal interactions inherent in the data. To this end, we propose I2MoE (Interpretable Multimodal Interaction-aware Mixture of Experts), an end-to-end MoE framework designed to enhance modality fusion by explicitly modeling diverse multimodal interactions, as well as providing interpretation on a local and global level. First, I2MoE utilizes different interaction experts with weakly supervised interaction losses to learn multimodal interactions in a data-driven way. Second, I2MoE deploys a reweighting model that assigns importance scores for the output of each interaction expert, which offers sample-level and dataset-level interpretation. Extensive evaluation of medical and general multimodal datasets shows that I2MoE is flexible enough to be combined with different fusion techniques, consistently improves task performance, and provides interpretation across various real-world scenarios. Code is available at https://github.com/Raina-Xin/I2MoE.
format Preprint
id arxiv_https___arxiv_org_abs_2505_19190
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle I2MoE: Interpretable Multimodal Interaction-aware Mixture-of-Experts
Xin, Jiayi
Yun, Sukwon
Peng, Jie
Choi, Inyoung
Ballard, Jenna L.
Chen, Tianlong
Long, Qi
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Modality fusion is a cornerstone of multimodal learning, enabling information integration from diverse data sources. However, vanilla fusion methods are limited by (1) inability to account for heterogeneous interactions between modalities and (2) lack of interpretability in uncovering the multimodal interactions inherent in the data. To this end, we propose I2MoE (Interpretable Multimodal Interaction-aware Mixture of Experts), an end-to-end MoE framework designed to enhance modality fusion by explicitly modeling diverse multimodal interactions, as well as providing interpretation on a local and global level. First, I2MoE utilizes different interaction experts with weakly supervised interaction losses to learn multimodal interactions in a data-driven way. Second, I2MoE deploys a reweighting model that assigns importance scores for the output of each interaction expert, which offers sample-level and dataset-level interpretation. Extensive evaluation of medical and general multimodal datasets shows that I2MoE is flexible enough to be combined with different fusion techniques, consistently improves task performance, and provides interpretation across various real-world scenarios. Code is available at https://github.com/Raina-Xin/I2MoE.
title I2MoE: Interpretable Multimodal Interaction-aware Mixture-of-Experts
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2505.19190