arXiv reaDer
ハイパースペクトル画像分類のための堅牢な自己アンサンブルネットワーク
Robust Self-Ensembling Network for Hyperspectral Image Classification
最近の研究では、ハイパースペクトル画像(HSI)分類タスクにおける深層学習アルゴリズムの大きな可能性が示されています。それにもかかわらず、これらのモデルのトレーニングには通常、大量のラベル付きデータが必要です。 HSIのピクセルレベルの注釈の収集は手間と時間がかかるため、サンプルサイズが小さい状況で優れたパフォーマンスを実現できるアルゴリズムを開発することは非常に重要です。この研究では、この問題に対処するための堅牢な自己アンサンブルネットワーク(RSEN)を提案します。提案されたRSENは、ベースネットワークとアンサンブルネットワークを含む2つのサブネットワークで構成されています。ラベル付きデータからの教師なし損失とラベルなしデータからの教師なし損失の両方の制約により、ベースネットワークとアンサンブルネットワークは相互に学習でき、自己アンサンブルメカニズムを実現します。私たちの知る限り、提案された方法は、自己アンサンブル手法をHSI分類タスクに導入する最初の試みであり、ネットワークトレーニングを支援するためにHSIのラベルなしデータを利用する方法について異なる見解を提供します。さらに、自己アンサンブル学習のロバスト性を高めるための新しい整合性フィルターを提案します。 3つのベンチマークHSIデータセットでの広範な実験は、提案されたアルゴリズムが最先端の方法と比較して競争力のあるパフォーマンスを生み出すことができることを示しています。
Recent research has shown the great potential of deep learning algorithms in the hyperspectral image (HSI) classification task. Nevertheless, training these models usually requires a large amount of labeled data. Since the collection of pixel-level annotations for HSI is laborious and time-consuming, developing algorithms that can yield good performance in the small sample size situation is of great significance. In this study, we propose a robust self-ensembling network (RSEN) to address this problem. The proposed RSEN consists of two subnetworks including a base network and an ensemble network. With the constraint of both the supervised loss from the labeled data and the unsupervised loss from the unlabeled data, the base network and the ensemble network can learn from each other, achieving the self-ensembling mechanism. To the best of our knowledge, the proposed method is the first attempt to introduce the self-ensembling technique into the HSI classification task, which provides a different view on how to utilize the unlabeled data in HSI to assist the network training. We further propose a novel consistency filter to increase the robustness of self-ensembling learning. Extensive experiments on three benchmark HSI datasets demonstrate that the proposed algorithm can yield competitive performance compared with the state-of-the-art methods.
updated: Thu Apr 08 2021 13:33:14 GMT+0000 (UTC)
published: Thu Apr 08 2021 13:33:14 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト