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

FedRSClip:利用视觉-语言模型进行遥感场景分类的联邦学习

FedRSClip: Federated Learning for Remote Sensing Scene Classification Using Vision-Language Models

China Academy of Electronics and Information Technology(中国电子科技集团公司电子科学研究院);Aerospace Information Research Institute, Chinese Academy of Sciences(中国科学院空天信息创新研究院);University of Chinese Academy of Sciences(中国科学院大学);New York University Abu Dhabi(纽约大学阿布扎比分校)

Hui Lin, Chao Zhang, Danfeng Hong, Kexin Dong, Congcong Wen

AI总结:

针对遥感数据隐私分散与 CLIP 等 VLM 参数传输成本高的问题,提出 FedRSCLIP 联邦提示学习框架,通过共享/私有双提示及两类对齐约束提升分类性能。

摘要:

遥感数据通常分布在多个机构中,并且由于隐私问题和数据共享限制,在集中式训练框架中利用大规模数据集具有挑战性。联邦学习通过使分布式数据源之间能够协同训练模型而无需集中数据,提供了一种有前景的解决方案。然而,当前的视觉-语言模型(VLM)通常包含数十亿个参数,这给基于模型参数更新的传统联邦学习方法带来了显著的通信挑战,因为它们会产生巨大的通信成本。本文提出 FedRSCLIP,这是首个专为基于 VLM(具体为 CLIP)的遥感图像分类设计的联邦学习框架。FedRSCLIP 通过引入 Prompt Learning(提示学习)来解决联邦环境中的数据异质性和大规模模型传输挑战,该方法仅优化一小组可调节参数。该框架引入了双提示机制,由用于全局知识共享的 Shared Prompts(共享提示)和用于客户端特定适配的 Private Prompts(私有提示)组成。为保持共享提示与私有提示之间的语义一致性,我们提出 Dual Prompt Alignment Constraint(双提示对齐约束),以在多样化客户端分布中平衡全局一致性和局部适应性。此外,为增强跨模态表示学习,我们引入 Cross-Modal Feature Alignment Constraint(跨模态特征对齐约束),用于对齐文本提示与图像提示之间的多模态特征。为验证所提模型的有效性,我们基于三个现有遥感图像分类数据集构建了 Fed-RSIC 数据集,专门用于模拟各种联邦学习配置。实验结果证明了 FedRSCLIP 在遥感图像分类中的有效性和优越性。

英文摘要:

Remote sensing data is often distributed across multiple institutions, and due to privacy concerns and data-sharing restrictions, leveraging large-scale datasets in a centralized training framework is challenging. Federated learning offers a promising solution by enabling collaborative model training across distributed data sources without requiring data centralization. However, current Vision-Language Models (VLMs), which typically contain billions of parameters, pose significant communication challenges for traditional federated learning approaches based on model parameter updates, as they would incur substantial communication costs. In this paper, we propose FedRSCLIP, the first federated learning framework designed for remote sensing image classification based on a VLM, specifically CLIP. FedRSCLIP addresses the challenges of data heterogeneity and large-scale model transmission in federated environments by introducing Prompt Learning, which optimizes only a small set of tunable parameters. The framework introduces a dual-prompt mechanism, comprising Shared Prompts for global knowledge sharing and Private Prompts for client-specific adaptation. To maintain semantic coherence between shared and private prompts, we propose the Dual Prompt Alignment Constraint to balance global consistency and local adaptability across diverse client distributions. Additionally, to enhance cross-modal representation learning, we introduce the Cross-Modal Feature Alignment Constraint to align multimodal features between text and image prompts. To validate the effectiveness of our proposed model, we construct a Fed-RSIC dataset based on three existing remote sensing image classification datasets, specifically designed to simulate various federated learning configurations. Experimental results demonstrate the effectiveness and superiority of FedRSCLIP in remote sensing image classification.

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

↑