arXiv reaDer
Vision GNN:画像はノードの価値のあるグラフです
Vision GNN: An Image is Worth Graph of Nodes
ネットワークアーキテクチャは、ディープラーニングベースのコンピュータビジョンシステムで重要な役割を果たします。広く使用されている畳み込みニューラルネットワークとトランスフォーマーは、画像をグリッドまたはシーケンス構造として扱います。これは、不規則で複雑なオブジェクトをキャプチャするための柔軟性がありません。この論文では、画像をグラフ構造として表現し、新しいVision GNN(ViG)アーキテクチャを導入して、視覚的タスクのグラフレベルの特徴を抽出することを提案します。まず、画像をノードとして表示されるいくつかのパッチに分割し、最近傍を接続してグラフを作成します。画像のグラフ表現に基づいて、ViGモデルを構築し、すべてのノード間で情報を変換および交換します。 ViGは、2つの基本モジュールで構成されています。グラフ情報を集約および更新するためのグラフ畳み込みを備えたGrapherモジュールと、ノード機能変換のための2つの線形レイヤーを備えたFFNモジュールです。 ViGの等方性アーキテクチャとピラミッドアーキテクチャはどちらも、異なるモデルサイズで構築されています。画像認識およびオブジェクト検出タスクに関する広範な実験は、ViGアーキテクチャの優位性を示しています。一般的な視覚的課題に関するGNNのこの先駆的な研究が、将来の研究に役立つインスピレーションと経験を提供することを願っています。 PyTorchコードはhttps://github.com/huawei-noah/Efficient-AI-Backbonesで入手でき、MindSporeコードはhttps://gitee.com/mindspore/modelsで入手できます。
Network architecture plays a key role in the deep learning-based computer vision system. The widely-used convolutional neural network and transformer treat the image as a grid or sequence structure, which is not flexible to capture irregular and complex objects. In this paper, we propose to represent the image as a graph structure and introduce a new Vision GNN (ViG) architecture to extract graph-level feature for visual tasks. We first split the image to a number of patches which are viewed as nodes, and construct a graph by connecting the nearest neighbors. Based on the graph representation of images, we build our ViG model to transform and exchange information among all the nodes. ViG consists of two basic modules: Grapher module with graph convolution for aggregating and updating graph information, and FFN module with two linear layers for node feature transformation. Both isotropic and pyramid architectures of ViG are built with different model sizes. Extensive experiments on image recognition and object detection tasks demonstrate the superiority of our ViG architecture. We hope this pioneering study of GNN on general visual tasks will provide useful inspiration and experience for future research. The PyTorch code is available at https://github.com/huawei-noah/Efficient-AI-Backbones and the MindSpore code is available at https://gitee.com/mindspore/models.
updated: Fri Nov 04 2022 14:45:03 GMT+0000 (UTC)
published: Wed Jun 01 2022 07:01:04 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アソシエイト