Balancing Multimodal Domain Generalization via Gradient Modulation and Projection

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Li, Hongzhao, Shen, Guohao, Li, Shupan, Xu, Mingliang, Khan, Muhammad Haris
Format: Preprint
Publié: 2026
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866917342459461632
author Li, Hongzhao
Shen, Guohao
Li, Shupan
Xu, Mingliang
Khan, Muhammad Haris
author_facet Li, Hongzhao
Shen, Guohao
Li, Shupan
Xu, Mingliang
Khan, Muhammad Haris
contents Multimodal Domain Generalization (MMDG) leverages the complementary strengths of multiple modalities to enhance model generalization on unseen domains. A central challenge in multimodal learning is optimization imbalance, where modalities converge at different speeds during training. This imbalance leads to unequal gradient contributions, allowing some modalities to dominate the learning process while others lag behind. Existing balancing strategies typically regulate each modality's gradient contribution based on its classification performance on the source domain to alleviate this issue. However, relying solely on source-domain accuracy neglects a key insight in MMDG: modalities that excel on the source domain may generalize poorly to unseen domains, limiting cross-domain gains. To overcome this limitation, we propose Gradient Modulation Projection (GMP), a unified strategy that promotes balanced optimization in MMDG. GMP first decouples gradients associated with classification and domain-invariance objectives. It then modulates each modality's gradient based on semantic and domain confidence. Moreover, GMP dynamically adjusts gradient projections by tracking the relative strength of each task, mitigating conflicts between classification and domain-invariant learning within modality-specific encoders. Extensive experiments demonstrate that GMP achieves state-of-the-art performance and integrates flexibly with diverse MMDG methods, significantly improving generalization across multiple benchmarks.
format Preprint
id arxiv_https___arxiv_org_abs_2603_14175
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Balancing Multimodal Domain Generalization via Gradient Modulation and Projection
Li, Hongzhao
Shen, Guohao
Li, Shupan
Xu, Mingliang
Khan, Muhammad Haris
Machine Learning
Computer Vision and Pattern Recognition
Multimodal Domain Generalization (MMDG) leverages the complementary strengths of multiple modalities to enhance model generalization on unseen domains. A central challenge in multimodal learning is optimization imbalance, where modalities converge at different speeds during training. This imbalance leads to unequal gradient contributions, allowing some modalities to dominate the learning process while others lag behind. Existing balancing strategies typically regulate each modality's gradient contribution based on its classification performance on the source domain to alleviate this issue. However, relying solely on source-domain accuracy neglects a key insight in MMDG: modalities that excel on the source domain may generalize poorly to unseen domains, limiting cross-domain gains. To overcome this limitation, we propose Gradient Modulation Projection (GMP), a unified strategy that promotes balanced optimization in MMDG. GMP first decouples gradients associated with classification and domain-invariance objectives. It then modulates each modality's gradient based on semantic and domain confidence. Moreover, GMP dynamically adjusts gradient projections by tracking the relative strength of each task, mitigating conflicts between classification and domain-invariant learning within modality-specific encoders. Extensive experiments demonstrate that GMP achieves state-of-the-art performance and integrates flexibly with diverse MMDG methods, significantly improving generalization across multiple benchmarks.
title Balancing Multimodal Domain Generalization via Gradient Modulation and Projection
topic Machine Learning
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2603.14175