FedAMB: Adaptive Modality Balancing for Dominance-Robust Multimodal Federated Distillation

摘要

Federated knowledge distillation (Fed-KD) exchanges distilled predictions on a shared proxy dataset, often reducing communication and accommodating heterogeneous client architectures. However, in multimodal federated learning (MFL), modality dominance can bias local optimization and contaminate the aggregated global teacher, degrading both multimodal accuracy and robustness to missing modalities, especially under non-IID heterogeneity. We propose Federated Adaptive-Modality Balancing (FedAMB). At the client side, Selective-Modality Regulation (SMR) models dominance as a state-dependent phenomenon and intervenes only when it destabilizes training, strengthening weak modalities without over-regularization. At the server side, Component-wise Modality Distillation (CMD) regulates how aggregated knowledge is transferred to each modality branch, preventing the propagation of fusion-biased teachers while directly improving unimodal representations. Experiments on CREMA-D, AVE, and UR-FUNNY show that FedAMB consistently improves multimodal accuracy and missing-modality robustness.