Journal of Computer Applications ›› 2024, Vol. 44 ›› Issue (3): 732-736.DOI: 10.11772/j.issn.1001-9081.2023030366

• Artificial intelligence • Previous Articles     Next Articles

Remote sensing image classification based on sample incremental learning

Xue LI1,2, Guangle YAO1,2(), Honghui WANG1,2, Jun LI2, Haoran ZHOU3,4, Shaoze YE4   

  1. 1.State Key Laboratory of Geohazard Prevention and Geoenvironment Protection (Chengdu University of Technology),Chengdu Sichuan 610059,China
    2.College of Computer Science and Cyber Security,Chengdu University of Technology,Chengdu Sichuan 610059,China
    3.Key Laboratory of Natural Disaster Monitoring,Early Warning and Assessment of Jiangxi Province (Jiangxi Normal University),Nanchang Jiangxi 330022,China
    4.Shenzhen Sensing Data Technology Company Limited,Shenzhen Guangdong 518052,China
  • Received:2023-04-06 Revised:2023-05-15 Accepted:2023-05-18 Online:2023-06-05 Published:2024-03-10
  • Contact: Guangle YAO
  • About author:LI Xue, born in 1998, M. S. candidate. Her research interests include incremental learning, remote sensing image intelligent processing.
    WANG Honghui, born in 1985, Ph. D., professor. His research interests include deep learning, geological disaster monitoring and early warning.
    LI Jun, born in 1973, Ph. D., professor. His research interests include big data analysis.
    ZHOU Haoran, born in 1997. His research interests include information processing, intelligent mapping.
    YE Shaoze, born in 1993, M. S. His research interests include deep learning, intelligent mapping.
  • Supported by:
    Key Research and Development Project of Sichuan Province(2021YFG0298);Basic Research Project of Sichuan Province(2021YJ0086);Geomathematics Key laboratory of Sichuan Province Open Foundation(SCSXDZ2020YB04)

基于样本增量学习的遥感影像分类

李雪1,2, 姚光乐1,2(), 王洪辉1,2, 李军2, 周皓然3,4, 叶绍泽4   

  1. 1.地质灾害防治与地质环境保护国家重点实验室(成都理工大学), 成都 610059
    2.成都理工大学 计算机与网络安全学院, 成都 610059
    3.江西省自然灾害监测预警与评估重点实验室(江西师范大学), 南昌 330022
    4.深圳市森歌数据技术有限公司, 广东 深圳 518052
  • 通讯作者: 姚光乐
  • 作者简介:李雪(1998—),女,四川中江人,硕士研究生,主要研究方向:增量学习、遥感图像智能处理
    王洪辉(1985—),男,湖北孝感人,教授,博士,主要研究方向:深度学习、地质灾害监测预警
    李军(1973—),男,湖北潜江人,教授,博士,主要研究方向:大数据分析
    周皓然(1997—),男,广东广州人,主要研究方向:信息处理、智能测绘
    叶绍泽(1993—),男,广东深圳人,硕士,主要研究方向:深度学习、智能测绘。
  • 基金资助:
    四川省重点研发项目(2021YFG0298);四川省基础研究项目(2021YJ0086);数学地质四川省重点实验室开放基金资助项目(SCSXDZ2020YB04)

Abstract:

Deep learning models have achieved remarkable results in remote sensing image classification. With the continuous collection of new remote sensing images, when the remote sensing image classification models based on deep learning train new data to learn new knowledge, their recognition performance of old data will decline, that is, old knowledge forgetting. In order to help remote sensing image classification model consolidate old knowledge and learn new knowledge, a remote sensing image classification model based on sample incremental learning, namely ICLKM (Incremental Collaborative Learning Knowledge Model) was proposed. The model consisted of two knowledge networks. The first network mitigated knowledge forgetting by retaining the output of the old model through knowledge distillation. The second network took the output of new data as the learning objective of the first network and effectively learned new knowledge by maintaining the consistency of the dual network models. Finally, two networks learned together to generate more accurate model through knowledge collaboration strategy. Experimental results on two remote sensing datasets NWPU-RESISC45 and AID show that, ICLKM has the accuracy improved by 3.53 and 6.70 percentage points respectively compared with FT (Fine-Tuning) method. It can be seen that ICLKM can effectively solve the knowledge forgetting problem of remote sensing image classification and continuously improve the recognition accuracy of known remote sensing images.

Key words: remote sensing image classification, incremental learning, Knowledge Distillation (KD), collaborative learning, Fine-Tuning (FT)

摘要:

深度学习模型在遥感影像分类中取得了显著的成绩。随着新的遥感数据不断被采集,基于深度学习的遥感影像分类模型在训练新数据、学习新知识时,对旧数据的识别性能会下降,即旧知识遗忘。为帮助遥感影像分类模型巩固旧知识和学习新知识,提出一种基于样本增量学习的遥感影像分类模型——增量协同学习知识模型(ICLKM)。该模型由两个知识网络组成,第一个网络通过知识蒸馏保留旧模型的输出,缓解知识遗忘问题;第二个网络将新数据的输出作为第一个网络的学习目标,通过维护双网络模型的一致性有效地学习新知识。最后两个网络共同学习,通过知识协同策略生成更精确的模型。在两个遥感数据集NWPU-RESISC45和AID上的实验结果表明,相较于微调训练(FT)方法,ICLKM的准确率分别提升了3.53和6.70个百分点。可见ICLKM能够有效解决遥感影像分类的知识遗忘问题,不断提高对已知遥感影像的识别准确率。

关键词: 遥感影像分类, 增量学习, 知识蒸馏, 协同学习, 微调

CLC Number: