{"title":"基于梯度编码的分散学习与压缩通信","authors":"Chengxi Li;Mikael Skoglund","doi":"10.1109/TSP.2024.3467262","DOIUrl":null,"url":null,"abstract":"This paper considers the problem of \n<italic>decentralized learning (DEL)</i>\n with stragglers under the communication bottleneck. In the literature, various gradient coding techniques have been proposed for \n<italic>distributed learning</i>\n with stragglers by letting the devices transmit encoded gradients based on redundant training data. However, those techniques can not be directly applied to fully decentralized scenarios as considered in this paper due to the lack of a global model in DEL. To overcome this shortcoming, we first propose a new \n<underline>go</u>\nssip-based DEL method with gradient \n<underline>co</u>\nding (GOCO). In GOCO, to mitigate the negative impact of stragglers, the devices update the parameter vectors with encoded gradients based on stochastic gradient coding before averaging in a gossip-based manner. To further reduce the communication overhead associated with GOCO, we propose an enhanced version of GOCO, namely GOCO with compressed communication (2-GOCO), where the devices transmit compressed messages instead of the raw parameter vectors. The convergence of the proposed methods is analyzed for strongly convex loss functions. Simulation results demonstrate that the proposed methods outperform the baseline methods, which attain better learning performance under the same communication overhead.","PeriodicalId":13330,"journal":{"name":"IEEE Transactions on Signal Processing","volume":"72 ","pages":"4713-4729"},"PeriodicalIF":4.6000,"publicationDate":"2024-09-26","publicationTypes":"Journal Article","fieldsOfStudy":null,"isOpenAccess":false,"openAccessPdf":"","citationCount":"0","resultStr":"{\"title\":\"Decentralized Learning Based on Gradient Coding With Compressed Communication\",\"authors\":\"Chengxi Li;Mikael Skoglund\",\"doi\":\"10.1109/TSP.2024.3467262\",\"DOIUrl\":null,\"url\":null,\"abstract\":\"This paper considers the problem of \\n<italic>decentralized learning (DEL)</i>\\n with stragglers under the communication bottleneck. In the literature, various gradient coding techniques have been proposed for \\n<italic>distributed learning</i>\\n with stragglers by letting the devices transmit encoded gradients based on redundant training data. However, those techniques can not be directly applied to fully decentralized scenarios as considered in this paper due to the lack of a global model in DEL. To overcome this shortcoming, we first propose a new \\n<underline>go</u>\\nssip-based DEL method with gradient \\n<underline>co</u>\\nding (GOCO). In GOCO, to mitigate the negative impact of stragglers, the devices update the parameter vectors with encoded gradients based on stochastic gradient coding before averaging in a gossip-based manner. To further reduce the communication overhead associated with GOCO, we propose an enhanced version of GOCO, namely GOCO with compressed communication (2-GOCO), where the devices transmit compressed messages instead of the raw parameter vectors. The convergence of the proposed methods is analyzed for strongly convex loss functions. Simulation results demonstrate that the proposed methods outperform the baseline methods, which attain better learning performance under the same communication overhead.\",\"PeriodicalId\":13330,\"journal\":{\"name\":\"IEEE Transactions on Signal Processing\",\"volume\":\"72 \",\"pages\":\"4713-4729\"},\"PeriodicalIF\":4.6000,\"publicationDate\":\"2024-09-26\",\"publicationTypes\":\"Journal Article\",\"fieldsOfStudy\":null,\"isOpenAccess\":false,\"openAccessPdf\":\"\",\"citationCount\":\"0\",\"resultStr\":null,\"platform\":\"Semanticscholar\",\"paperid\":null,\"PeriodicalName\":\"IEEE Transactions on Signal Processing\",\"FirstCategoryId\":\"5\",\"ListUrlMain\":\"https://ieeexplore.ieee.org/document/10695150/\",\"RegionNum\":2,\"RegionCategory\":\"工程技术\",\"ArticlePicture\":[],\"TitleCN\":null,\"AbstractTextCN\":null,\"PMCID\":null,\"EPubDate\":\"\",\"PubModel\":\"\",\"JCR\":\"Q1\",\"JCRName\":\"ENGINEERING, ELECTRICAL & ELECTRONIC\",\"Score\":null,\"Total\":0}","platform":"Semanticscholar","paperid":null,"PeriodicalName":"IEEE Transactions on Signal Processing","FirstCategoryId":"5","ListUrlMain":"https://ieeexplore.ieee.org/document/10695150/","RegionNum":2,"RegionCategory":"工程技术","ArticlePicture":[],"TitleCN":null,"AbstractTextCN":null,"PMCID":null,"EPubDate":"","PubModel":"","JCR":"Q1","JCRName":"ENGINEERING, ELECTRICAL & ELECTRONIC","Score":null,"Total":0}
引用次数: 0
摘要
本文探讨了在通信瓶颈条件下有散兵的分散学习(DEL)问题。文献中已经提出了多种梯度编码技术,通过让设备根据冗余训练数据传输编码梯度,来实现有散兵的分布式学习。然而,由于 DEL 中缺乏全局模型,这些技术无法直接应用于本文所考虑的完全分散场景。为了克服这一缺陷,我们首先提出了一种新的基于梯度编码的流言式 DEL 方法(GOCO)。在 GOCO 中,为了减轻散兵游勇的负面影响,各设备在以流言方式平均之前,先用基于随机梯度编码的编码梯度更新参数向量。为了进一步减少与 GOCO 相关的通信开销,我们提出了 GOCO 的增强版本,即压缩通信 GOCO(2-GOCO),在该版本中,设备传输的是压缩信息,而不是原始参数矢量。针对强凸损失函数,分析了所提方法的收敛性。仿真结果表明,所提出的方法优于基线方法,后者在相同的通信开销下获得了更好的学习性能。
Decentralized Learning Based on Gradient Coding With Compressed Communication
This paper considers the problem of
decentralized learning (DEL)
with stragglers under the communication bottleneck. In the literature, various gradient coding techniques have been proposed for
distributed learning
with stragglers by letting the devices transmit encoded gradients based on redundant training data. However, those techniques can not be directly applied to fully decentralized scenarios as considered in this paper due to the lack of a global model in DEL. To overcome this shortcoming, we first propose a new
go
ssip-based DEL method with gradient
co
ding (GOCO). In GOCO, to mitigate the negative impact of stragglers, the devices update the parameter vectors with encoded gradients based on stochastic gradient coding before averaging in a gossip-based manner. To further reduce the communication overhead associated with GOCO, we propose an enhanced version of GOCO, namely GOCO with compressed communication (2-GOCO), where the devices transmit compressed messages instead of the raw parameter vectors. The convergence of the proposed methods is analyzed for strongly convex loss functions. Simulation results demonstrate that the proposed methods outperform the baseline methods, which attain better learning performance under the same communication overhead.
期刊介绍:
The IEEE Transactions on Signal Processing covers novel theory, algorithms, performance analyses and applications of techniques for the processing, understanding, learning, retrieval, mining, and extraction of information from signals. The term “signal” includes, among others, audio, video, speech, image, communication, geophysical, sonar, radar, medical and musical signals. Examples of topics of interest include, but are not limited to, information processing and the theory and application of filtering, coding, transmitting, estimating, detecting, analyzing, recognizing, synthesizing, recording, and reproducing signals.