首页 >  , Vol. , Issue () : -

摘要

全文摘要次数: 352 全文下载次数: 357
引用本文:

DOI:

10.11834/jrs.20211368

收稿日期:

2021-05-30

修改日期:

2021-10-27

PDF Free   EndNote   BibTeX
遥感图像小样本舰船识别跨域迁移学习算法
陈华杰, 吕丹妮, 周枭, 刘俊
杭州电子科技大学 通信信息传输与融合技术国防重点学科实验室
摘要:

跨域迁移学习旨在利用现有公开数据集,突破源域和目标域样本类别空间须一致的约束,提升目标域样本的识别精度。针对现有跨域迁移学习算法应用于遥感图像小样本舰船目标识别时存在的迁移类别受限和负迁移问题,本文提出一种基于源域样本相关性排序的跨域迁移学习算法:首先将目标域样本逆向加入源域分类任务中,根据加入前后各类别源域样本的识别精度变化情况,对源域样本进行相关性排序,将其划分为强/弱/负相关样本;然后采取自监督联合学习策略,在目标域分类网络中引入自监督角度预测辅助分支,筛选出的强相关源域样本仅参与该辅助分支的训练,不改变目标域主分类网络的结构。算法通过相关性排序去除了弱/负相关源域样本,有效避免了负迁移;引入自监督角度预测辅助分支,在保持主分类网络结构完整性的同时,充分利用了强相关源域样本的有效信息,学习到更具泛化能力的目标特征。实验结果显示:在遥感舰船小样本目标数据集上,提出的算法优于跨域迁移学习中广泛使用的Finetune (微调)算法;与仅使用主分类网络的目标域识别算法相比,遥感舰船目标识别精度提升了17.59%。

Cross-domain transfer learning algorithm for few-shot ship recognition in remote-sensing images
Abstract:

Cross-domain transfer learning aims to utilize the public datasets, as the source data, to improve the recognition accuracy of target data,breaking through the limitation that the category space between source data and target data must be consistent. As for the few-shot remote-sensing ship recognition task, the existing cross-domain transfer learning algorithms have the disadvantages of transfer category restriction and negative transfer effect. Therefore, a cross-domain transfer learning algorithm based on the source data correlation sorting was proposed to solve the above problems. Firstly, the target data was added reversely into the source domain recognition task, and according to the variation of the source data recognition accuracy before and after the target data was added, the various source data were classified into strong/weak/negative correlation samples, and only the strong correlation samples would be selected. Then, the self-supervised joint learning strategy was adopted to introduce the auxiliary self-supervised angle prediction branch into the classification network in the target domain. The selected strong correlation source samples were added and only added into the training of the self-supervised branch, avoiding changing the main classification network structure. The proposed algorithm has two main advantages: firstly, the weak/negative correlation source samples are eliminated by correlation sorting, which can avoid the occurrence of negative transfer effect; Secondly, by introducing the self-supervised angle prediction branch, the information of the strong correlation source samples was fully utilized and the features with more generalization ability were extracted, while maintaining the structural integrity of the main classification network. Randomly selecting sixty-five categories samples as the source data from miniImageNet, through the comparative experiments on few-shot ship target in remote-sensing images, the following conclusions can be obtained: 1) Choosing Resnet18 as the classification network, the performance of the proposed algorithm is better than Fine-tune algorithm which is widely used in cross-domain transfer learning, and compared with the recognition algorithm which only uses the main classification network, the proposed algorithm improves the recognition accuracy of target data from 78.89% to 96.48%; 2) Using different networks to sort correlation for the source data, the selected strong correlation source samples are not exactly the same and their categories coincidence degree is close to 60%, but they are all helpful to the classification task of the target domain. At the end of this paper, through visualizing the extracted target features, it is verified that the target features extracted by using the proposed algorithm, are more abundant and have higher generalization ability.

本文暂时没有被引用!

欢迎关注学报微信

遥感学报交流群 分享按钮