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. Conferences, Workshops, Symposiums, and Seminars
  4. Weight Erosion: An Update Aggregation Scheme for Personalized Collaborative Machine Learning
 
conference paper

Weight Erosion: An Update Aggregation Scheme for Personalized Collaborative Machine Learning

Grimberg, Felix
•
Hartley, Mary-Anne
•
Jaggi, Martin  
Show more
2020
DART 2020, DCL 2020: Domain Adaptation and Representation Transfer, and Distributed and Collaborative Learning
MICCAI Workshop on Distributed and Collaborative Learning

Background. In medicine and other applications, the copying and sharing of data is impractical for a range of well-considered reasons. With federated learning (FL) techniques, machine learning models can be trained on data spread across several locations without such copying and sharing. While good privacy guarantees can often be made, FL does not automatically incentivize participation and the resulting model can suffer if data is non-identically distributed (non-IID) across locations. Model personalization is a way of addressing these concerns. Methods. In this study, we introduce Weight Erosion: an SGD-based gradient aggregation scheme for personalized collaborative ML. We evaluate this scheme on a binary classification task in the Titanic data set. Findings. We demonstrate that the novel Weight Erosion scheme can outperform two baseline FL aggregation schemes on a classification task, and is more resistant to over-fitting and non-IID data sets.

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

Grimberg2020_Chapter_WeightErosionAnUpdateAggregati (2).pdf

Type

Publisher's Version

Version

http://purl.org/coar/version/c_970fb48d4fbd8a85

Access type

openaccess

License Condition

CC BY

Size

663.66 KB

Format

Adobe PDF

Checksum (MD5)

c85beed65e8e75de0d533a1ad47fc6c9

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