arXiv reaDer
限られたデータのハイパースペクトルリモートセンシング画像分類のためのベイズ畳み込みニューラルネットワーク
Bayesian Convolutional Neural Networks for Limited Data Hyperspectral Remote Sensing Image Classification
ハイパースペクトルリモートセンシング(HSRS)画像分類にディープニューラルネットワークを採用することは、困難な作業です。 HSRS画像は、高次元で多数のチャネルを持ち、チャネル間にかなりの冗長性があります。さらに、HSRS画像を分類するためのトレーニングデータは限られており、利用可能なトレーニングデータの量は他の分類タスクと比較してはるかに少なくなっています。これらの要因により、多くのパラメータを持つディープニューラルネットワークのトレーニングプロセスが複雑になり、従来のモデルと比較してもパフォーマンスが低下します。さらに、畳み込みニューラルネットワークは自信過剰な予測を生成します。これは、前述の問題を考慮すると非常に望ましくありません。この作業では、HSRS画像を分類するために、特別なクラスのディープニューラルネットワーク、つまりベイズニューラルネットワークを使用します。私たちの知る限り、このクラスのニューラルネットワークがHSRS画像分類で使用されるのはこれが初めてです。ベイジアンニューラルネットワークは、不確実性を測定するための固有のツールを提供します。ベイジアンネットワークが、同様に構築された非ベイジアン畳み込みニューラルネットワーク(CNN)および既製のランダムフォレスト(RF)よりも優れていることを示します。さらに、Pavia Centre、Salinas、およびBotswanaデータセットの実験結果は、ベイジアンネットワークがモデルの剪定に対してより安定していて堅牢であることを示しています。さらに、ベイズモデルの予測の不確実性を分析し、予測の不確実性メトリックがモデルの予測に関する情報を提供でき、予測誤差と正の相関があることを示します。
Employing deep neural networks for Hyper-spectral remote sensing (HSRS) image classification is a challenging task. HSRS images have high dimensionality and a large number of channels with substantial redundancy between channels. In addition, the training data for classifying HSRS images is limited and the amount of available training data is much smaller compared to other classification tasks. These factors complicate the training process of deep neural networks with many parameters and cause them to not perform well even compared to conventional models. Moreover, convolutional neural networks produce over-confident predictions, which is highly undesirable considering the aforementioned problem. In this work, we use a special class of deep neural networks, namely Bayesian neural network, to classify HSRS images. To the extent of our knowledge, this is the first time that this class of neural networks has been used in HSRS image classification. Bayesian neural networks provide an inherent tool for measuring uncertainty. We show that a Bayesian network can outperform a similarly-constructed non-Bayesian convolutional neural network (CNN) and an off-the-shelf Random Forest (RF). Moreover, experimental results for the Pavia Centre, Salinas, and Botswana datasets show that the Bayesian network is more stable and robust to model pruning. Furthermore, we analyze the prediction uncertainty of the Bayesian model and show that the prediction uncertainty metric can provide information about the model predictions and has a positive correlation with the prediction error.
updated: Thu May 19 2022 00:02:16 GMT+0000 (UTC)
published: Thu May 19 2022 00:02:16 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アソシエイト