mlsanityData to 7 Oct 2026

Label Granularity Skew in Federated Learning with Hierarchical Image Classification

arXiv 2608.09236 ↗First posted 10 Aug 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Federated learning enables privacy-preserving collaboration across distributed devices without centralizing local data. However, clients may differ not only in data distributions but also in domain knowledge and annotation capabilities. In this paper, we introduce label granularity skew, a new form of statistical heterogeneity in federated hierarchical classification, in which clients provide taxonomy-consistent labels at different levels of detail within a shared class hierarchy. To model this heterogeneity, we generate client-specific local label hierarchies using a probabilistic relational neighbor classifier and construct a WordNet-guided hierarchy via silhouette score-based coarsening. Our analysis shows that strongly coupled hierarchical models are sensitive to incomplete supervision, while the conditional softmax classifier is more robust. Based on this insight, we propose Branch-wise Decoupled Fine-Tuning (BDFT) and its federated version, FedBDFT, which fine-tune branch-wise classifiers and aggregate them through federated optimization. Experiments on CIFAR-100, TinyImageNet, and ImageNet show that FedBDFT substantially improves robustness under severe label granularity skew, with average gains of 27.9% and 56.4% at skewness levels of 0.6 and 0.9, respectively. Zero-shot results further indicate that FedBDFT better preserves hierarchical representations for unseen fine-grained classes. These findings demonstrate its effectiveness for federated hierarchical classification with heterogeneous label granularities.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 14 of its 33 references that are in the map. First posted on arXiv 10 Aug 2026.

Builds on

14 of 33 references.

References · 14

  1. 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
  2. 2016YOLO9000: Better, Faster, Strongercited by 1,429
  3. 2016SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model sizecited by 1,054
  4. 2023Heterogeneous Federated Learning: State-of-the-art and Research Challengescited by 160
  5. 2019Making Better Mistakes: Leveraging Class Hierarchies with Deep Networkscited by 112
  6. 2020Progressive Identification of True Labels for Partial-Label Learningcited by 102
  7. 2017Open Vocabulary Scene Parsingcited by 76
  8. 2020Hierarchical Image Classification using Entailment Cone Embeddingscited by 46
  9. 2023FedNoRo: Towards Noise-Robust Federated Learning by Addressing Class Imbalance and Label Noise Heterogeneitycited by 33
  10. 2022Label Relation Graphs Enhanced Hierarchical Residual Network for Hierarchical Multi-Granularity Classificationcited by 29
  11. 2018Integrating domain knowledge: using hierarchies to improve deep classifierscited by 25
  12. 2022Hierarchical classification at multiple operating pointscited by 14
2 more
  1. 2023Federated Learning with Instance-Dependent Noisy Labelcited by 9
  2. 2024FedMLP: Federated Multi-Label Medical Image Classification under Task Heterogeneitycited by 5

Built on by

Not cited yet.