English

FedMLAC: Mutual Learning Driven Heterogeneous Federated Audio Classification

Sound 2025-08-05 v2 Distributed, Parallel, and Cluster Computing Audio and Speech Processing

Abstract

Federated Learning (FL) offers a privacy-preserving framework for training audio classification (AC) models across decentralized clients without sharing raw data. However, Federated Audio Classification (FedAC) faces three major challenges: data heterogeneity, model heterogeneity, and data poisoning, which degrade performance in real-world settings. While existing methods often address these issues separately, a unified and robust solution remains underexplored. We propose FedMLAC, a mutual learning-based FL framework that tackles all three challenges simultaneously. Each client maintains a personalized local AC model and a lightweight, globally shared Plug-in model. These models interact via bidirectional knowledge distillation, enabling global knowledge sharing while adapting to local data distributions, thus addressing both data and model heterogeneity. To counter data poisoning, we introduce a Layer-wise Pruning Aggregation (LPA) strategy that filters anomalous Plug-in updates based on parameter deviations during aggregation. Extensive experiments on four diverse audio classification benchmarks, including both speech and non-speech tasks, show that FedMLAC consistently outperforms state-of-the-art baselines in classification accuracy and robustness to noisy data.

Keywords

Cite

@article{arxiv.2506.10207,
  title  = {FedMLAC: Mutual Learning Driven Heterogeneous Federated Audio Classification},
  author = {Jun Bai and Rajib Rana and Di Wu and Youyang Qu and Xiaohui Tao and Ji Zhang and Carlos Busso and Shivakumara Palaiahnakote},
  journal= {arXiv preprint arXiv:2506.10207},
  year   = {2025}
}

Comments

updated version for the first submission

R2 v1 2026-07-01T03:12:13.279Z