Balancing Multimodal Learning through Label Space Reshaping
Not on the map. Why?
Abstract (arXiv)
Multimodal learning often suffers from modality imbalance, where modalities that converge faster dominate optimization while others remain undertrained. Existing approaches typically mitigate this issue by strengthening the weak modality or adjusting optimization gradients. However, such strategies mainly compensate for optimization rate discrepancies, often at the expense of the strong modality's optimization capacity, without analyzing how these discrepancies arise at the modality level. Based on theoretical insights and empirical observations, we argue that the discrepancy of learning pace arises from differences in the mapping difficulty between modality-specific feature space and the shared label space. To address this issue, we propose Balanced Multimodal Label Reshaping (BMLR), the first method that promotes multimodal balance from the label-side design. BMLR reshapes the cross-modal label space to equalize mapping difficulty across modalities, thereby facilitating modality interaction and injecting richer inter-class information into each modality. Extensive experiments across multiple architectures demonstrate that BMLR consistently improves multimodal performance and exhibits strong compatibility with diverse model designs. The source code will be released soon.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 12 of its 56 references that are in the map. First posted on arXiv 22 May 2026.
Builds on
12 of 56 references.
References · 12
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2017The Kinetics Human Action Video Datasetcited by 1,909
- 2017Regularizing Neural Networks by Penalizing Confident Output Distributionscited by 464
- 2024ReconBoost: Boosting Can Achieve Modality Reconcilementcited by 60
- 2021Is Label Smoothing Truly Incompatible with Knowledge Distillation: An Empirical Studycited by 51
- 2021Improving Multi-Modal Learning with Uni-Modal Teacherscited by 41
- 2024Classifier-guided Gradient Modulation for Enhanced Multimodal Learningcited by 35
- 2025ABKD: Pursuing a Proper Allocation of the Probability Mass in Knowledge Distillation via α-β-Divergencecited by 23
- 2023Understanding Unimodal Bias in Multimodal Deep Linear Networkscited by 19
- 2024Improving Multimodal Learning with Multi-Loss Gradient Modulationcited by 18
- 2025Boosting Multimodal Learning via Disentangled Gradient Learningcited by 16
- 2025BalanceBenchmark: A Survey for Multimodal Imbalance Learningcited by 8
Built on by
Not cited yet.