Robust distributed gradient aggregation using projections onto gradient manifolds
- Title
- Robust distributed gradient aggregation using projections onto gradient manifolds
- Authors
- KIM, KWANG IN
- Date Issued
- 2024-02-23
- Publisher
- AAAI Association for the Advancement of Artificial Intelligence
- Abstract
- We study the distributed gradient aggregation problem where individual clients contribute to learning a central model by sharing parameter gradients constructed from local losses. However, errors in some gradients, caused by low-quality data or adversaries, can degrade the learning process when naively combined. Existing robust gradient aggregation approaches assume that local data represent the global data-generating distribution, which may not always apply to heterogeneous (non-i.i.d.) client data. We propose a new algorithm that can robustly aggregate gradients from potentially heterogeneous clients. Our approach leverages the manifold structure inherent in heterogeneous client gradients and evaluates gradient anomaly degrees by projecting them onto this manifold. This algorithm is implemented as a simple and efficient method that accumulates random projections within the subspace defined by the nearest neighbors within a gradient cloud. Our experiments demonstrate consistent performance improvements over state-of-the-art robust aggregation algorithms.
- URI
- https://oasis.postech.ac.kr/handle/2014.oak/121861
- Article Type
- Conference
- Citation
- Annual AAAI Conference on Artificial Intelligence, 2024-02-23
- Files in This Item:
- There are no files associated with this item.
Items in DSpace are protected by copyright, with all rights reserved, unless otherwise indicated.