Skip to content

BioIntelligence-Lab/SurgicalAggregation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

NOTE: This repository is currently work in progress.

📢 Machine Learning for Health 2024

We are excited to announce that our paper has been accepted for proceedings track at 2024 Machine Learning for Health Symposium in Vancouver!


arXiv License: GPL v3

From Isolation to Collaboration: Federated Class-Heterogeneous Learning for Chest X-Ray Classification

Pranav Kulkarni, Adway kanhere, Paul H. Yi, Vishwa S. Parekh

concept figure

Federated learning (FL) is a promising paradigm to collaboratively train a global chest x-ray (CXR) classification model using distributed datasets while preserving patient privacy. A significant, yet relatively underexplored, challenge in FL is class-heterogeneity, where clients have different sets of classes. We propose surgical aggregation, a FL method that uses selective aggregation to collaboratively train a global model using distributed, class-heterogeneous datasets. Unlike other methods, our method does not rely on the assumption that clients share the same classes as other clients, know the classes of other clients, or have access to a fully annotated dataset. We evaluate surgical aggregation using class-heterogeneous CXR datasets across IID and non-IID settings. Our results show that our method outperforms current methods and has better generalizability.

Read the full paper here.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages