Exploiting data diversity in multi-domain federated learning.
Mach. Learn.: Sci. Technol. 5:025041 (2024)
Federated learning (FL) is an evolving machine learning technique that allows collaborative model training without sharing the original data among participants. In real-world scenarios, data residing at multiple clients are often heterogeneous in terms of different resolutions, magnifications, scanners, or imaging protocols, and thus challenging for global FL model convergence in collaborative training. Most of the existing FL methods consider data heterogeneity within one domain by assuming same data variation in each client site. In this paper, we consider data heterogeneity in FL with different domains of heterogeneous data by raising the problems of domain-shift, class-imbalance, and missing data. We propose a method, multi-domain FL as a solution to heterogeneous training data from multiple domains by training robust vision transformer model. We use two loss functions, one for correctly predicting class labels and other for encouraging similarity and dissimilarity over latent features, to optimize the global FL model. We perform various experiments using different convolution-based networks and non-convolutional Transformer architectures on multi-domain datasets. We evaluate the proposed approach on benchmark datasets and compare with the existing FL methods. Our results show the superiority of the proposed approach which performs better in term of robust FL global model than the exiting methods.
Impact Factor
Scopus SNIP
Web of Science
Times Cited
Scopus
Cited By
Altmetric
Publication type
Article: Journal article
Document type
Scientific Article
Thesis type
Editors
Keywords
Class-imbalance ; Data Heterogeneity ; Domain-shift ; Federated Learning ; Multi-domain Data
Keywords plus
Language
english
Publication Year
2024
Prepublished in Year
0
HGF-reported in Year
2024
ISSN (print) / ISBN
2632-2153
e-ISSN
2632-2153
ISBN
Book Volume Title
Conference Title
Conference Date
Conference Location
Proceedings Title
Quellenangaben
Volume: 5,
Issue: 2,
Pages: ,
Article Number: 025041
Supplement: ,
Series
Publisher
Institute of Physics Publishing (IOP)
Publishing Place
Temple Circus, Temple Way, Bristol Bs1 6be, England
Day of Oral Examination
0000-00-00
Advisor
Referee
Examiner
Topic
University
University place
Faculty
Publication date
0000-00-00
Application date
0000-00-00
Patent owner
Further owners
Application country
Patent priority
Reviewing status
Peer reviewed
POF-Topic(s)
30205 - Bioengineering and Digital Health
Research field(s)
Enabling and Novel Technologies
PSP Element(s)
G-540007-001
Grants
Department Strategic Project of the University of Udine
Copyright
Erfassungsdatum
2024-07-09