arXiv reaDer
ハイパースペクトル画像分類のためのロバストな自己集合ネットワーク
Robust Self-Ensembling Network for Hyperspectral Image Classification
最近の研究では、ハイパースペクトル画像 (HSI) 分類タスクにおけるディープ ラーニング アルゴリズムの大きな可能性が示されています。それにもかかわらず、これらのモデルのトレーニングには通常、大量のラベル付きデータが必要です。 HSI のピクセル レベルの注釈の収集は面倒で時間がかかるため、小さなサンプル サイズの状況で優れたパフォーマンスを発揮できるアルゴリズムを開発することは非常に重要です。この研究では、この問題に対処するための堅牢な自己集合ネットワーク (RSEN) を提案します。提案された RSEN は、ベース ネットワークとアンサンブル ネットワークを含む 2 つのサブネットワークで構成されます。ラベル付きデータからの教師あり損失とラベルなしデータからの教師なし損失の両方の制約により、ベース ネットワークとアンサンブル ネットワークは相互に学習し、自己アンサンブル メカニズムを実現できます。私たちの知る限り、提案された方法は、HSI分類タスクに自己アンサンブル手法を導入する最初の試みであり、ネットワークトレーニングを支援するためにHSIのラベルなしデータを利用する方法について異なるビューを提供します。さらに、自己集合学習のロバスト性を高めるための新しい一貫性フィルターを提案します。 3 つのベンチマーク HSI データセットでの広範な実験により、提案されたアルゴリズムが最先端の方法と比較して競争力のあるパフォーマンスを生み出すことができることが実証されています。コードはオンラインで入手できます (https://github.com/YonghaoXu/RSEN)。
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. Code is available online (https://github.com/YonghaoXu/RSEN).
updated: Wed Sep 07 2022 14:14:22 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アソシエイト