少样本下发现高光谱图像中的未知类别,提升分类准确性。
Few-shot Unknown Class Discovery of Hyperspectral Images with Prototype Learning and Clustering
- 基于原型学习与聚类,从少量标注数据中推断未知类原型。
- 在四个基准数据集上优于现有方法,显著提升未知类识别率。
- 适合遥感、环境监测等需要识别新类别的实际应用场景。
开放集少样本高光谱图像(HSI)分类旨在仅用每类少量标注像素对图像像素进行分类,且待分类像素可能不属于已知类别。现有方法主要关注区分未知类与已知类并拒绝未知类以提高已知类识别准确率,但无法进一步发现或识别未知类别。本文提出一种原型学习与聚类方法,用于在少样本环境下发现高光谱图像中的未知类别。利用少量标注样本,该方法不仅学习已知类分类器,还具备推断未知类原型的能力。当未知类样本被已知类分类器拒绝后,可依据其与推断出的未知类原型之间的距离进行聚类,从而实现未知类的发现。在四个基准高光谱图像数据集上的大量实验表明,所提方法在开放集少样本分类任务中表现优异,性能超过现有先进方法。代码已公开于 https://github.com/KOBEN-ff/OpenFUCD-main。
原文摘要 · Abstract (English)
Open-set few-shot hyperspectral image (HSI) classification aims to classify image pixels by using few labeled pixels per class, where the pixels to be classified may be not all from the classes that have been seen. To address the open-set HSI classification challenge, current methods focus mainly on distinguishing the unknown class samples from the known class samples and rejecting them to increase the accuracy of identifying known class samples. They fails to further identify or discovery the unknow classes among the samples. This paper proposes a prototype learning and clustering method for discoverying unknown classes in HSIs under the few-shot environment. Using few labeled samples, it strives to develop the ability of infering the prototypes of unknown classes while distinguishing unknown classes from known classes. Once the unknown class samples are rejected by the learned known class classifier, the proposed method can further cluster the unknown class samples into different classes according to their distance to the inferred unknown class prototypes. Compared to existing state-of-the-art methods, extensive experiments on four benchmark HSI datasets demonstrate that our proposed method exhibits competitive performance in open-set few-shot HSI classification tasks. All the codes are available at \href{https://github.com/KOBEN-ff/OpenFUCD-main} {https://github.com/KOBEN-ff/OpenFUCD-main}
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。