arXiv reaDer
MobileViG: モバイル ビジョン アプリケーション向けのグラフベースのスパース アテンション
MobileViG: Graph-Based Sparse Attention for Mobile Vision Applications
従来、畳み込みニューラル ネットワーク (CNN) とビジョン トランスフォーマー (ViT) がコンピューター ビジョンを支配してきました。しかし、最近提案されたビジョン グラフ ニューラル ネットワーク (ViG) は、探索のための新しい手段を提供します。残念ながら、モバイル アプリケーションの場合、画像をグラフ構造として表現するオーバーヘッドのため、ViG は計算コストが高くなります。この研究では、モバイル デバイス上で実行される ViG 用に設計された、新しいグラフベースのスパース アテンション メカニズムであるスパース ビジョン グラフ アテンション (SVGA) を提案します。さらに、SVGA を使用する、モバイル デバイス上のビジョン タスク用の初のハイブリッド CNN-GNN アーキテクチャである MobileViG を提案します。広範な実験により、画像分類、物体検出、インスタンス セグメンテーション タスクの精度や速度の点で、MobileViG が既存の ViG モデルや既存のモバイル CNN および ViT アーキテクチャを上回ることが示されています。当社の最速モデルである MobileViG-Ti は、iPhone 13 Mini NPU (CoreML でコンパイル) 上で 0.78 ミリ秒の推論レイテンシーを備え、ImageNet-1K で 75.7% のトップ 1 精度を達成します。これは、MobileNetV2x1.4 (1.02 ミリ秒、トップ 1 の 74.7%) よりも高速です。 1) および MobileNetV2x1.0 (0.81 ミリ秒、71.8% トップ 1)。当社の最大のモデルである MobileViG-B は、わずか 2.30 ミリ秒の遅延で 82.6% のトップ 1 精度を達成しています。これは、同様のサイズの EfficientFormer-L3 モデル (2.77 ミリ秒、82.4%) よりも高速かつ正確です。私たちの研究は、適切に設計されたハイブリッド CNN-GNN アーキテクチャが、モバイル デバイス上で非常に高速かつ正確なモデルを設計するための新たな探索手段となり得ることを証明しています。私たちのコードは https://github.com/SLDGroup/MobileViG で公開されています。
Traditionally, convolutional neural networks (CNN) and vision transformers (ViT) have dominated computer vision. However, recently proposed vision graph neural networks (ViG) provide a new avenue for exploration. Unfortunately, for mobile applications, ViGs are computationally expensive due to the overhead of representing images as graph structures. In this work, we propose a new graph-based sparse attention mechanism, Sparse Vision Graph Attention (SVGA), that is designed for ViGs running on mobile devices. Additionally, we propose the first hybrid CNN-GNN architecture for vision tasks on mobile devices, MobileViG, which uses SVGA. Extensive experiments show that MobileViG beats existing ViG models and existing mobile CNN and ViT architectures in terms of accuracy and/or speed on image classification, object detection, and instance segmentation tasks. Our fastest model, MobileViG-Ti, achieves 75.7% top-1 accuracy on ImageNet-1K with 0.78 ms inference latency on iPhone 13 Mini NPU (compiled with CoreML), which is faster than MobileNetV2x1.4 (1.02 ms, 74.7% top-1) and MobileNetV2x1.0 (0.81 ms, 71.8% top-1). Our largest model, MobileViG-B obtains 82.6% top-1 accuracy with only 2.30 ms latency, which is faster and more accurate than the similarly sized EfficientFormer-L3 model (2.77 ms, 82.4%). Our work proves that well designed hybrid CNN-GNN architectures can be a new avenue of exploration for designing models that are extremely fast and accurate on mobile devices. Our code is publicly available at https://github.com/SLDGroup/MobileViG.
updated: Sat Jul 01 2023 17:49:12 GMT+0000 (UTC)
published: Sat Jul 01 2023 17:49:12 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アソシエイト