arXiv reaDer
正確かつ低複雑な SAR ATR のためのグラフ ニューラル ネットワーク
Graph Neural Network for Accurate and Low-complexity SAR ATR
合成開口レーダー (SAR) 自動目標認識 (ATR) は、リモート センシング画像認識の重要な技術です。最先端の研究では、SAR ATR にディープ畳み込みニューラル ネットワーク (CNN) が活用されており、計算コストが高くなります。これらのディープ CNN モデルは、リソースが限られたプラットフォームに展開するのには適していません。この研究では、正確で低遅延の SAR ATR を実現するグラフ ニューラル ネットワーク (GNN) モデルを提案します。入力された SAR 画像をグラフ表現に変換します。提案された GNN モデルは、入力グラフを操作してターゲットの分類を実行する GNN レイヤーのスタックで構成されています。大量の畳み込み演算を必要とする最先端の CNN とは異なり、提案された GNN モデルは計算の複雑さが低く、同等の高精度を実現します。 GNN ベースのアプローチにより、私たちが提案する入力プルーニング戦略が可能になります。入力グラフ内の無関係な頂点をフィルタリングして除外することで、計算の複雑さを軽減できます。さらに、計算の複雑さをさらに軽減するモデルの重み行列をスパース化するモデル枝刈り戦略を提案します。提案された GNN モデルを MSTAR データセットと船舶識別データセットで評価します。評価結果は、提案された GNN モデルが上記 2 つのデータセットに対してそれぞれ 99.38% と 99.7% の分類精度を達成することを示しています。提案された枝刈り戦略では、無視できる精度の損失で 98.6% の入力頂点と 97% の重みエントリを枝刈りできます。最先端の CNN と比較して、提案された GNN モデルの計算コストはわずか 1/3000、モデル サイズは 1/80 です。
Synthetic Aperture Radar (SAR) Automatic Target Recognition (ATR) is the key technique for remote sensing image recognition. The state-of-the-art works exploit the deep convolutional neural networks (CNNs) for SAR ATR, leading to high computation costs. These deep CNN models are unsuitable to be deployed on resource-limited platforms. In this work, we propose a graph neural network (GNN) model to achieve accurate and low-latency SAR ATR. We transform the input SAR image into the graph representation. The proposed GNN model consists of a stack of GNN layers that operates on the input graph to perform target classification. Unlike the state-of-the-art CNNs, which need heavy convolution operations, the proposed GNN model has low computation complexity and achieves comparable high accuracy. The GNN-based approach enables our proposed input pruning strategy. By filtering out the irrelevant vertices in the input graph, we can reduce the computation complexity. Moreover, we propose the model pruning strategy to sparsify the model weight matrices which further reduces the computation complexity. We evaluate the proposed GNN model on the MSTAR dataset and ship discrimination dataset. The evaluation results show that the proposed GNN model achieves 99.38% and 99.7% classification accuracy on the above two datasets, respectively. The proposed pruning strategies can prune 98.6% input vertices and 97% weight entries with negligible accuracy loss. Compared with the state-of-the-art CNNs, the proposed GNN model has only 1/3000 computation cost and 1/80 model size.
updated: Thu May 11 2023 20:17:41 GMT+0000 (UTC)
published: Thu May 11 2023 20:17:41 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アソシエイト