解决遥感图像多模态联邦学习难题,提升分类精度。
A Multi-Modal Federated Learning Framework for Remote Sensing Image Classification

- 通过多模态融合、特征白化和互信息最大化实现跨模态协作训练
- 在两个基准数据集上优于现有方法,显著提升多标签与像素级分类性能
- 适合处理异构多源遥感数据的分布式建模任务
联邦学习(FL)可在不共享客户端本地数据的前提下,实现跨分散数据源的深度神经网络协同训练。现有大多数FL方法假设所有客户端的数据属于同一模态,但实际遥感(RS)图像在不同客户端可能呈现多种数据模态。联合利用多模态遥感数据可显著提升分类性能。为此,本文提出一种新的多模态联邦学习框架,用于遥感图像分类。该框架包含三个模块:1)多模态融合(MF),采用迭代模型平均机制,在不访问客户端多模态数据的情况下促进学习;2)特征白化(FW),通过对齐各客户端数据分布缓解数据异构性问题;3)互信息最大化(MIM),通过增强不同模态间图像相似性来建模互信息。实验聚焦于遥感领域的多标签分类与像素级分类任务,在两个基准数据集上验证了该框架的有效性,结果优于文献中的先进算法。代码将发布于 https://git.tu-berlin.de/rsim/multi-modal-FL。
原文摘要 · Abstract (English)
Federated learning (FL) enables the collaborative training of deep neural networks across decentralized data archives (i.e., clients) without sharing the local data of the clients. Most of the existing FL methods assume that the data distributed across all clients is associated with the same data modality. However, remote sensing (RS) images present in different clients can be associated with diverse data modalities. The joint use of the multi-modal RS data can significantly enhance classification performance. To effectively exploit decentralized and unshared multi-modal RS data, our paper introduces a novel multi-modal FL framework for RS image classification problems. The proposed framework comprises three modules: 1) multi-modal fusion (MF); 2) feature whitening (FW); and 3) mutual information maximization (MIM). The MF module employs iterative model averaging to facilitate learning without accessing multi-modal training data on clients. The FW module aims to address the limitations of training data heterogeneity by aligning data distributions across clients. The MIM module aims to model mutual information by maximizing the similarity between images from different modalities. For the experimental analyses, we focus our attention on multi-label classification and pixel-based classification tasks in RS. The results obtained using two benchmark archives show the effectiveness of the proposed framework when compared to state-of-the-art algorithms in the literature. The code of the proposed framework will be available at https://git.tu-berlin.de/rsim/multi-modal-FL.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。