FedHarmony: Harmonizing Heterogeneous Label Correlations in Federated Multi-Label Learning
Zhiqiang Kou, Junxiang Wu, Wenke Huang, Wenwen He, Ming-Kun Xie, Changwei Wang, Yuheng Jia, Di Jiang, Yang Liu, Xin Geng, Qiang Yang
Abstract
Multi-label representations encode higher-order label dependencies, yet in federated settings the local estimates of these dependencies are statistically inconsistent, causing structural drift across clients and rendering naive quantity-weighted aggregation suboptimal. We propose FedHarmony, a federated multi-label learning framework that harmonizes heterogeneous label correlations without sharing raw data. A Correlation Expert is formed by leave-one-out consolidation of clients’ label–label correlation statistics to provide a round-wise global consensus. Guided by this expert, each client performs consensus-guided correction that aligns its local correlation to the consensus within clusters of strongly related labels obtained via spectral clustering of the expert matrix. This block-wise alignment targets dense, high-signal subspaces. We establish two guarantees: (i) restricting alignment to in-cluster pairs strictly improves optimization curvature and linear convergence rate; (ii) ignoring cross-cluster entries incurs only a bounded, quantitatively small information loss when the consensus is near block-diagonal. Finally, a correlation-aware central aggregation combines data quantity with a dynamic measure of correlation learning quality, using a dynamic balance factor that transitions from quantity-driven weighting in early rounds to structure-driven weighting later. Extensive experiments under diverse non-IID regimes (varying label distributions, client heterogeneity, and client counts) show consistent gains over federated baselines in mAP/F1/Hamming Loss, with improved stability and communication efficiency.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext f5e52497-ad2b-4358-bdd5-84fd9edc9beeCited by top-tier papers12
- When Tabular Foundation Models Meet Strategic Tabular Data: A Prior Alignment ApproachXinpeng Lv, Yunxin Mao, Renzhe Xu, Chunyuan Zheng et al.ICML 2026 · 2 citations
- Beyond Rational Illusion: Behaviorally Realistic Strategic ClassificationXinpeng Lv, Yunxin Mao, Renzhe Xu, Chunyuan Zheng et al.ICML 2026 · 1 citation
- Beyond Single Embedding: Modeling User Preferences as Distribution in Federated RecommendationChunxu Zhang, Weipeng Zhang, Guodong Long, Zhiheng Xue et al.ICML 2026
- FedPAT: Federated Test-Time Adaptation via Prototype Affinity TopologyShunxin Guo, JIAQI LYU, Zhiqiang Kou, Shuxia Lin et al.ICML 2026
- Adaptive Visual Autoregressive Acceleration via Dual-Linkage Entropy AnalysisYu Zhang, Jingyi Liu, Feng Liu, Duoqian Miao et al.ICML 2026
Builds on15
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- Deep Learning with Differential PrivacyMartín Abadi, Andy Chu, Ian J. Goodfellow, H. Brendan McMahan et al.CCS 2016 · 7,620 citations
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi et al.ICML 2020 · 3,875 citations
- Tackling the Objective Inconsistency Problem in Heterogeneous Federated OptimizationJianyu Wang, Qinghua Liu, Hao Liang, Gauri Joshi et al.NeurIPS 2020 · 2,231 citations
- Ensemble Distillation for Robust Model Fusion in Federated LearningTao Lin, Lingjing Kong, Sebastian U. Stich, Martin JaggiNeurIPS 2020 · 1,615 citations
Related papers
- FedDisco: Federated Learning with Discrepancy-Aware CollaborationRui Ye, Mingkai Xu, Jianyu Wang, Chenxin Xu et al.ICML 2023 · 136 citations
- FedSC: Federated Learning with Semantic-Aware CollaborationHuan Wang, Haoran Li, Huaming Chen, Jun Yan et al.KDD 2025 · 1 citation
- HarmoFL: Harmonizing Local and Global Drifts in Federated Learning on Heterogeneous Medical ImagesMeirui Jiang, Zirui Wang, Qi DouAAAI 2022 · 187 citations
- FedCDWA: Decoupled Federated Prototype Distillation with Hierarchical Wasserstein AggregationZhenshen Liu, Kai Fan, Wenjie Li, Kuan Zhang et al.ICML 2026
- FedCorr: Multi-Stage Federated Learning for Label Noise CorrectionJingyi Xu, Zihan Chen, Tony Q. S. Quek, Kai Fong Ernest ChongCVPR 2022 · 101 citations
