Jun, 2024

FLoCoRA:具有低秩适应性的联邦学习压缩

TL;DR在此研究中,我们展示了 Low-Rank Adaptation 方法在从头开始训练小尺寸视觉模型的联邦学习中的应用,通过提出一种聚合不可知方法 FLoCoRA,我们证明该方法能够减少通信成本 4.8 倍,并且对于具有 ResNet-8 模型的 CIFAR-10 分类任务的准确度损失小于 1%。同时,我们还展示了通过引入仿射量化方案扩展该方法,将通信成本缩减 18.6 倍,与标准方法相比,对于 ResNet-18 模型的测试仍然保持不到 1% 的准确度损失。我们的研究在消息大小减小方面提供了一个强有力的基准,甚至与传统的模型压缩方法相比,同时还减少了由于低秩适应所需的训练内存。