cs.LGApr 30, 2026

FedHarmony: Harmonizing Heterogeneous Label Correlations in Federated Multi-Label Learning

Authors: Zhiqiang KouJunxiang WuWenke HuangWenwen HeMing-Kun XieChangwei WangYuheng JiaDi Jiang+3 more

Organizations: School of Computer Science and Engineering, Southeast University, Nanjing, China · Key Laboratory of New Generation Artificial Intelligence Technology and Its Interdisciplinary Applications (Southeast University), Ministry of Education, China · Wuhan University, China · RIKEN Center for Advanced Intelligence Project, Japan · Qilu University of Technology (Shandong Academy of Sciences), China · Academy for Artificial Intelligence, Hong Kong Polytechnic University, Hong Kong, China

Abstract

Federated Multi-Label Learning is a distributed paradigm where multiple clients possess heterogeneous multi-label data and perform collaborative learning under privacy constraints without sharing raw data. However, modeling label correlations under heterogeneous distributions remains challenging. Due to client-specific label spaces and varying co-occurrence patterns, correlations learned by individual clients inevitably deviate from the global structure, a phenomenon we term label correlation drift. To address this, we propose FedHarmony, a framework that harmonizes heterogeneous label correlations across clients. It introduces consensus correlation, capturing agreement among other clients and serving as a global teacher to correct biased local estimates. During aggregation, FedHarmony evaluates each client by both data size and correlation quality, assigning weights accordingly. Moreover, we develop an accelerated optimization algorithm for FedHarmony and theoretically establish faster convergence without sacrificing accuracy. Experiments on real-world federated multi-label datasets show that FedHarmony consistently outperforms state-of-the-art methods.

Explore similar work

CardsList
  1. FedReLa: Imbalanced Federated Learning via Re-Labeling

    Jun 24, 2026Guangzheng Hu, Patricia Menéndez, Feng Liu +3Federated LearningClass Imbalance