Deep-Relative-Trust-Based Diffusion for Decentralized Deep Learning

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Li, Muyun, Fainman, Aaron, Vlaski, Stefan
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915117292060672
author Li, Muyun
Fainman, Aaron
Vlaski, Stefan
author_facet Li, Muyun
Fainman, Aaron
Vlaski, Stefan
contents Decentralized learning strategies allow a collection of agents to learn efficiently from local data sets without the need for central aggregation or orchestration. Current decentralized learning paradigms typically rely on an averaging mechanism to encourage agreement in the parameter space. We argue that in the context of deep neural networks, which are often over-parameterized, encouraging consensus of the neural network outputs, as opposed to their parameters can be more appropriate. This motivates the development of a new decentralized learning algorithm, termed DRT diffusion, based on deep relative trust (DRT), a recently introduced similarity measure for neural networks. We provide convergence analysis for the proposed strategy, and numerically establish its benefit to generalization, especially with sparse topologies, in an image classification task.
format Preprint
id arxiv_https___arxiv_org_abs_2501_03162
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Deep-Relative-Trust-Based Diffusion for Decentralized Deep Learning
Li, Muyun
Fainman, Aaron
Vlaski, Stefan
Machine Learning
Signal Processing
Decentralized learning strategies allow a collection of agents to learn efficiently from local data sets without the need for central aggregation or orchestration. Current decentralized learning paradigms typically rely on an averaging mechanism to encourage agreement in the parameter space. We argue that in the context of deep neural networks, which are often over-parameterized, encouraging consensus of the neural network outputs, as opposed to their parameters can be more appropriate. This motivates the development of a new decentralized learning algorithm, termed DRT diffusion, based on deep relative trust (DRT), a recently introduced similarity measure for neural networks. We provide convergence analysis for the proposed strategy, and numerically establish its benefit to generalization, especially with sparse topologies, in an image classification task.
title Deep-Relative-Trust-Based Diffusion for Decentralized Deep Learning
topic Machine Learning
Signal Processing
url https://arxiv.org/abs/2501.03162