论文标题

使用多任务深度学习,使用未知类别的高光谱图像分类

Few-Shot Hyperspectral Image Classification With Unknown Classes Using Multitask Deep Learning

论文作者

Liu, Shengjie, Shi, Qian, Zhang, Liangpei

论文摘要

当前的高光谱图像分类假设预定义的分类系统已封闭和完整,并且在看不见的数据中没有未知或新颖的类。但是,对于现实世界,这种假设可能太严格了。通常,当构建分类系统时,新颖的类被忽略。封闭的自然力量是分配标签的模型,并可能导致高估已知土地覆盖率(例如农作物区域)。为了解决这个问题,我们提出了一种多任务深度学习方法,该方法同时在开放世界(名为MDL4OW)中进行分类和重建可能存在,那里可能存在未知类别。将重建的数据与原始数据进行了比较;基于假设缺乏标签,那些未能重建的人被认为是未知的。需要定义一个阈值以将未知类别和已知类别分开;我们提出了基于极端价值理论的两种策略,以进行几次射击和许多镜头。提出的方法在现实世界中的高光谱图像上进行了测试。萨利纳斯数据的总体准确性提高了4.94%,取得了最先进的结果。通过考虑开放世界中未知类的存在,我们的方法实现了更准确的高光谱图像分类,尤其是在几个射击环境下。

Current hyperspectral image classification assumes that a predefined classification system is closed and complete, and there are no unknown or novel classes in the unseen data. However, this assumption may be too strict for the real world. Often, novel classes are overlooked when the classification system is constructed. The closed nature forces a model to assign a label given a new sample and may lead to overestimation of known land covers (e.g., crop area). To tackle this issue, we propose a multitask deep learning method that simultaneously conducts classification and reconstruction in the open world (named MDL4OW) where unknown classes may exist. The reconstructed data are compared with the original data; those failing to be reconstructed are considered unknown, based on the assumption that they are not well represented in the latent features due to the lack of labels. A threshold needs to be defined to separate the unknown and known classes; we propose two strategies based on the extreme value theory for few-shot and many-shot scenarios. The proposed method was tested on real-world hyperspectral images; state-of-the-art results were achieved, e.g., improving the overall accuracy by 4.94% for the Salinas data. By considering the existence of unknown classes in the open world, our method achieved more accurate hyperspectral image classification, especially under the few-shot context.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源