Skip to content
Snippets Groups Projects

Addition of Choco sharing

Closed Milos Vujasinovic requested to merge (removed):main into main

This merge request adds implementation of ChocoSGD1 in form of a sharing module decentralizepy.sharing.Choco. As a compression of choice the implementation uses topk sparsification over all elements of a model.

This sharing module adds two new parameters:

  • step_size - Step size parameter from the formulation of ChocoSGD in the original paper1
  • alpha - Percentage of model parameters to keep during topk sparsification of a model

1 Koloskova, Anastasia, Sebastian Stich, and Martin Jaggi. "Decentralized stochastic optimization and gossip algorithms with compressed communication." International Conference on Machine Learning. PMLR, 2019.

Edited by Milos Vujasinovic

Merge request reports

Approval is optional

Closed by Rishi SharmaRishi Sharma 2 years ago (Oct 20, 2022 8:23pm UTC)

Merge details

  • The changes were not merged into main.

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
Please register or sign in to reply
Loading