用原型压缩与高效通信,实现快速自适应的去中心化学习
Towards efficient compression and communication for prototype-based decentralized learning
- 仅传输信息有用的原型更新,结合聚类压缩消息
- 通信量显著降低,收敛速度保持不变
- 适合物联网等动态非独立同分布数据场景
在基于原型的联邦学习中,客户端与主服务器间交换模型参数被替换为传输原型或数据样本的量化版本。本文研究无中心聚合器的去中心化原型学习系统,通过双重数据压缩策略解决原型冗余问题:利用Jensen-Shannon距离判断原型是否信息有用,仅发送有效更新;并采用聚类压缩共识协议中的更新消息。同时引入并行而非串行的八卦通信,并分析其信息年龄(AoI)。实验表明,该方法在不降低学习算法收敛速度的前提下,显著减少通信负载。
原文摘要 · Abstract (English)
In prototype-based federated learning, the exchange of model parameters between clients and the master server is replaced by transmission of prototypes or quantized versions of the data samples to the aggregation server. A fully decentralized deployment of prototype-based learning, without a central agregartor of prototypes, is more robust upon network failures and reacts faster to changes in the statistical distribution of the data, suggesting potential advantages and quick adaptation in dynamic learning tasks, e.g., when the data sources are IoT devices or when data is non-iid. In this paper, we consider the problem of designing a communication-efficient decentralized learning system based on prototypes. We address the challenge of prototype redundancy by leveraging on a twofold data compression technique, i.e., sending only update messages if the prototypes are informationtheoretically useful (via the Jensen-Shannon distance), and using clustering on the prototypes to compress the update messages used in the gossip protocol. We also use parallel instead of sequential gossiping, and present an analysis of its age-of-information (AoI). Our experimental results show that, with these improvements, the communications load can be substantially reduced without decreasing the convergence rate of the learning algorithm.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。