Mitigating Sample-Level Imbalance via Probabilistic Separation for Adaptive Multimodal Fusion

arXiv:2510.21797v4 Announce Type: replace-cross Abstract: Multimodal learning faces modality imbalance, where dominant modalities suppress weaker ones due to inconsistent convergence rates. Existing static or heuristic methods overlook sample-level variations in prediction bias and fail to isolate low-quality outlier samples. To address this, we propose a novel framework to quantitatively diagnose and dynamically mitigate modality imbalance at the sample level. We first introduce a Modality Gap metric to quantify prediction discrepancies between unimodal branches. Empirical analysis reveals a distinct bimodal distribution, reflecting the natural coexistence of balanced and imbalanced sample subgroups. We then employ a Gaussian Mixture Model (GMM) to model this gap distribution, leveraging Bayesian posterior probabilities for probabilistic soft separation of subgroups. Next, we construct a two-stage training framework comprising a Warm-up stage and an Adaptive Training stage. In the Adaptive Training stage, a GMM-guided Adaptive Loss dynamically reallocates optimization priorities, imposing stronger modality alignment penalties on imbalanced samples while prioritizing multimodal fusion for balanced ones. Experimental results demonstrate that our method significantly outperforms current state-of-the-art baselines. Furthermore, fine-tuning on a high-quality balanced subset filtered by the GMM serves as an effective data purification strategy.

Abstract:Multimodal learning faces modality imbalance, where dominant modalities suppress weaker ones due to inconsistent convergence rates. Existing static or heuristic methods overlook sample-level variations in prediction bias and fail to isolate low-quality outlier samples. To address this, we propose a novel framework to quantitatively diagnose and dynamically mitigate modality imbalance at the sample level. We first introduce a Modality Gap metric to quantify prediction discrepancies between unimodal branches. Empirical analysis reveals a distinct bimodal distribution, reflecting the natural coexistence of balanced and imbalanced sample subgroups. We then employ a Gaussian Mixture Model (GMM) to model this gap distribution, leveraging Bayesian posterior probabilities for probabilistic soft separation of subgroups. Next, we construct a two-stage training framework comprising a Warm-up stage and an Adaptive Training stage. In the Adaptive Training stage, a GMM-guided Adaptive Loss dynamically reallocates optimization priorities, imposing stronger modality alignment penalties on imbalanced samples while prioritizing multimodal fusion for balanced ones. Experimental results demonstrate that our method significantly outperforms current state-of-the-art baselines. Furthermore, fine-tuning on a high-quality balanced subset filtered by the GMM serves as an effective data purification strategy.

Machine Learning (cs.LG); Artificial Intelligence (cs.AI); Sound (cs.SD); Audio and Speech Processing (eess.AS)

arXiv:2510.21797 [cs.LG]

(or arXiv:2510.21797v4 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2510.21797 Focus to learn more arXiv-issued DOI via DataCite

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.

aiqbench analysis

Mitigating Sample-Level Imbalance via Probabilistic Separation for Adaptive Multimodal Fusion

Related coverage