DRIVE: One-bit Distributed Mean Estimation

  • Shay Vargaftik
  • , Ran Ben Basat
  • , Amit Portnoy
  • , Gal Mendelson
  • , Yaniv Ben-Itzhak
  • , Michael Mitzenmacher

    Research output: Chapter in Book/Report/Conference proceedingConference contributionpeer-review

    32 Scopus citations

    Abstract

    We consider the problem where n clients transmit d-dimensional real-valued vectors using d(1 + o(1)) bits each, in a manner that allows the receiver to approximately reconstruct their mean. Such compression problems naturally arise in distributed and federated learning. We provide novel mathematical results and derive computationally efficient algorithms that are more accurate than previous compression techniques. We evaluate our methods on a collection of distributed and federated learning tasks, using a variety of datasets, and show a consistent improvement over the state of the art.

    Original languageEnglish
    Title of host publicationAdvances in Neural Information Processing Systems 34 - 35th Conference on Neural Information Processing Systems, NeurIPS 2021
    EditorsMarc'Aurelio Ranzato, Alina Beygelzimer, Yann Dauphin, Percy S. Liang, Jenn Wortman Vaughan
    PublisherNeural information processing systems foundation
    Pages362-377
    Number of pages16
    ISBN (Electronic)9781713845393
    StatePublished - 1 Jan 2021
    Event35th Conference on Neural Information Processing Systems, NeurIPS 2021 - Virtual, Online
    Duration: 6 Dec 202114 Dec 2021

    Publication series

    NameAdvances in Neural Information Processing Systems
    Volume1
    ISSN (Print)1049-5258

    Conference

    Conference35th Conference on Neural Information Processing Systems, NeurIPS 2021
    CityVirtual, Online
    Period6/12/2114/12/21

    ASJC Scopus subject areas

    • Signal Processing
    • Information Systems
    • Computer Networks and Communications

    Fingerprint

    Dive into the research topics of 'DRIVE: One-bit Distributed Mean Estimation'. Together they form a unique fingerprint.

    Cite this