轻量代理模型让边缘设备高效适配大模型,兼顾隐私与性能。
FedPromo: Federated Lightweight Proxy Models at the Edge Bring New Domains to Foundation Models
- 用轻量代理模型替代客户端直接训练大模型,降低算力负担。
- 在五个图像分类数据集上表现优于现有方法,提升适应能力。
- 适合资源受限设备,支持多领域个性化学习,保护用户隐私。
联邦学习(FL)是训练分布式数据上深度学习模型的成熟范式。然而,随着模型规模增大,传统联邦学习方法往往需要客户端设备具备大量计算资源,难以实现。本文提出 FedPromo,一种新框架,使存储于中心服务器的大规模基础模型能够高效适应仅由远程客户端遇到的新领域。不同于在客户端直接训练大模型,FedPromo 通过联邦学习优化轻量级代理模型,显著降低计算开销并保持隐私。该方法采用两阶段流程:首先,在服务器端进行知识蒸馏,将大规模基础模型(如 Transformer)的表征与紧凑模型(如 CNN)对齐;随后,将紧凑模型编码器部署至客户端,在本地学习可训练分类器,并将其聚合后无缝回传至基础模型,实现个性化适配而无需访问用户数据。通过创新正则化策略,本框架支持去中心化多领域学习,在性能、隐私与资源效率间取得平衡。在五个图像分类基准上的大量实验表明,FedPromo 在资源受限客户端条件下优于现有方法。
原文摘要 · Abstract (English)
Federated Learning (FL) is an established paradigm for training deep learning models on decentralized data. However, as the size of the models grows, conventional FL approaches often require significant computational resources on client devices, which may not be feasible. We introduce FedPromo, a novel framework that enables efficient adaptation of large-scale foundation models stored on a central server to new domains encountered only by remote clients. Instead of directly training the large model on client devices, FedPromo optimizes lightweight proxy models via FL, significantly reducing computational overhead while maintaining privacy. Our method follows a two-stage process: first, server-side knowledge distillation aligns the representations of a large-scale foundation model (e.g., a transformer) with those of a compact counterpart (e.g., a CNN). Then, the compact model encoder is deployed to client devices, where trainable classifiers are learned locally. These classifiers are subsequently aggregated and seamlessly transferred back to the foundation model, facilitating personalized adaptation without requiring direct access to user data. Through novel regularization strategies, our framework enables decentralized multi-domain learning, balancing performance, privacy, and resource efficiency. Extensive experiments on five image classification benchmarks demonstrate that FedPromo outperforms existing methods while assuming limited-resource clients.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。