arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2109.09246cs.LGcs.AI
收藏

不带客户端同步的分流学习:分析客户端分流网络部分大小与整体性能

Splitfed learning without client-side synchronization: Analyzing client-side split network portion size to overall performance

Munster Technological University(芒斯特理工大学)

Praveen Joshi, Chandra Thapa, Seyit Camtepe, Mohammed Hasanuzzamana, Ted Scully, Haithem Afli

摘要:

联邦学习(FL)、分裂学习(SL)和分裂美联储学习(SFL)是分布式机器学习的三个最新发展,由于其保护原始数据隐私的能力而受到关注。因此,它们广泛适用于数据敏感的各种领域,例如大规模医疗图像分类、医疗物联网和跨组织钓鱼电子邮件检测。SFL是在FL和SL的交汇点上开发的。它通过提供来自FL范式的并行客户端机器学习模型更新,以及通过在来自SL的客户端和服务器之间拆分模型来提供更高级别的模型隐私(在训练时),从而实现了FL和SL的最佳效果。然而,由于客户端模型同步的要求,SFL在客户端具有通信和计算开销。对于资源受限的客户端侧,需要去除这样的要求以获得学习的效率。在这方面,本文研究了SFL没有客户端模型同步。由此产生的架构被称为多头分裂学习。我们的实证研究,考虑在分布式客户端之间的IID数据分布的MNIST数据的ResNet 18模型发现,多头分裂学习是可行的。其性能与SFL相当。此外,SFL在MNIST测试集上的准确率仅比多头分裂学习高1%-2%。为了进一步加强我们的结果,我们研究了具有各种客户端模型部分的多头分裂学习及其对整体性能的影响。为此,我们的结果发现对模型的整体性能的影响最小。

英文摘要:

Federated Learning (FL), Split Learning (SL), and SplitFed Learning (SFL) are three recent developments in distributed machine learning that are gaining attention due to their ability to preserve the privacy of raw data. Thus, they are widely applicable in various domains where data is sensitive, such as large-scale medical image classification, internet-of-medical-things, and cross-organization phishing email detection. SFL is developed on the confluence point of FL and SL. It brings the best of FL and SL by providing parallel client-side machine learning model updates from the FL paradigm and a higher level of model privacy (while training) by splitting the model between the clients and server coming from SL. However, SFL has communication and computation overhead at the client-side due to the requirement of client-side model synchronization. For the resource-constrained client-side, removal of such requirements is required to gain efficiency in the learning. In this regard, this paper studies SFL without client-side model synchronization. The resulting architecture is known as Multi-head Split Learning. Our empirical studies considering the ResNet18 model on MNIST data under IID data distribution among distributed clients find that Multi-head Split Learning is feasible. Its performance is comparable to the SFL. Moreover, SFL provides only 1%-2% better accuracy than Multi-head Split Learning on the MNIST test set. To further strengthen our results, we study the Multi-head Split Learning with various client-side model portions and its impact on the overall performance. To this end, our results find a minimal impact on the overall performance of the model.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑