arXiv reaDer
SCG-Net:セマンティックセグメンテーション用の自己構築型グラフニューラルネットワーク
SCG-Net: Self-Constructing Graph Neural Networks for Semantic Segmentation
長距離のピクセルピクセル依存関係を利用してグローバルコンテキスト表現をキャプチャすると、セマンティックセグメンテーションのパフォーマンスが向上することが示されています。ただし、注意スキームまたは非常に深いモデルを利用してモデルの視野を拡大する現在のアプローチでは、メモリ消費量が多い複雑なモデルになるため、これを効率的に行う方法は未解決の問題です。グラフニューラルネットワークに関する最近の研究に触発されて、画像から直接長距離依存グラフを学習し、それを使用してコンテキスト情報を効率的に伝播してセマンティックセグメンテーションを改善する自己構築グラフ(SCG)モジュールを提案します。このモジュールは、斬新な適応対角線強化法と、カスタマイズされたグラフ再構成項とカルバックライブラーダイバージェンス正則化項で構成される変分下限を介して最適化されています。ニューラルネットワーク(SCG-Net)に組み込むと、セマンティックセグメンテーションがエンドツーエンドの方法で実行され、一般に入手可能なISPRSポツダムとヴァイヒンゲンのデータセットで競合パフォーマンス(それぞれ、平均F1スコアが92.0%と89.8%)が得られます。関連する純粋な畳み込みニューラルネットワーク(CNN)ベースのモデルと比較して、はるかに少ないパラメーターで、より低い計算コストで達成されました。
Capturing global contextual representations by exploiting long-range pixel-pixel dependencies has shown to improve semantic segmentation performance. However, how to do this efficiently is an open question as current approaches of utilising attention schemes or very deep models to increase the models field of view, result in complex models with large memory consumption. Inspired by recent work on graph neural networks, we propose the Self-Constructing Graph (SCG) module that learns a long-range dependency graph directly from the image and uses it to propagate contextual information efficiently to improve semantic segmentation. The module is optimised via a novel adaptive diagonal enhancement method and a variational lower bound that consists of a customized graph reconstruction term and a Kullback-Leibler divergence regularization term. When incorporated into a neural network (SCG-Net), semantic segmentation is performed in an end-to-end manner and competitive performance (mean F1-scores of 92.0% and 89.8% respectively) on the publicly available ISPRS Potsdam and Vaihingen datasets is achieved, with much fewer parameters, and at a lower computational cost compared to related pure convolutional neural network (CNN) based models.
updated: Sun Jan 03 2021 14:59:24 GMT+0000 (UTC)
published: Thu Sep 03 2020 12:13:09 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アソシエイト