Repository logo

Infoscience

  • English
  • French
Log In
Logo EPFL, École polytechnique fédérale de Lausanne

Infoscience

  • English
  • French
Log In
  1. Home
  2. Academic and Research Output
  3. Student works
  4. Privacy-preserving and Personalized Federated Machine Learning for Medical Data
 
semester or other student projects

Privacy-preserving and Personalized Federated Machine Learning for Medical Data

Grimberg, Felix  
June 18, 2020

The federated learning setting is prone to suffering from non-identically distributed data across participating agents. This gives rise to the task of model personalization, where agents collaborate to train several different machine learning models instead of training only one global model. The aim of model personalization is to minimize the sum of the generalization error incurred from training on small data sets, and the transfer error incurred from applying a globally-trained model to a specific local distribution. In this report, two novel approaches to personalized cross-silo federated learning are introduced and discussed from a theoretical perspective: the adapted Ndoye factor, and the Weight Erosion aggregation scheme. The latter is implemented and compared to two baseline aggregation schemes in two case studies: training a diagnostic model on real-world medical data, and predicting the survival of passengers on the publicly available Titanic data set. The models trained using the Weight Erosion aggregation scheme are compared to those trained using the baseline aggregation schemes, both in terms of their classification accuracy on the local test set and in terms of the learned parameters. We demonstrate that the novel Weight Erosion scheme can outperform both baseline aggregation schemes for some specific tasks.

  • Files
  • Details
  • Metrics
Loading...
Thumbnail Image
Name

Grimberg_personalizedFL.pdf

Access type

openaccess

License Condition

CC BY

Size

612.64 KB

Format

Adobe PDF

Checksum (MD5)

ad2f50dd7027426f52f458ddc5cef43d

Logo EPFL, École polytechnique fédérale de Lausanne
  • Contact
  • infoscience@epfl.ch

  • Follow us on Facebook
  • Follow us on Instagram
  • Follow us on LinkedIn
  • Follow us on X
  • Follow us on Youtube
AccessibilityLegal noticePrivacy policyCookie settingsEnd User AgreementGet helpFeedback

Infoscience is a service managed and provided by the Library and IT Services of EPFL. © EPFL, tous droits réservés