arXiv reaDer
ケーキ:効率的な3Dネットワークのためのチャネルごとの自動カーネル縮小
CAKES: Channel-wise Automatic KErnel Shrinking for Efficient 3D Networks
3D畳み込みニューラルネットワーク(CNN)は、ビデオ分析や体積画像認識などの3Dシーンの理解に広く適用されています。ただし、3Dネットワークはパラメータ化の過剰につながりやすく、計算コストが高くなります。この論文では、チャネルごとの自動カーネル縮小(CAKES)を提案し、標準の3D畳み込みを1D、2D畳み込みなどの一連の経済的操作に縮小することにより、効率的な3D学習を可能にします。以前の方法とは異なり、CAKESはチャネルごとのカーネル縮小を実行します。これには、次の利点があります。1)すべてのレイヤーにデプロイされた操作を異種化できるため、学習プロセスに役立つ多様で補完的な情報を抽出できます。 2)効率的で柔軟な置換設計を可能にします。これは、時空間データと体積データの両方に一般化できます。さらに、CAKESに基づく新しい検索空間を提案し、3Dネットワークを簡素化するために置換構成を自動的に決定できるようにします。 CAKESは、同様のモデルサイズを持つ他の方法よりも優れたパフォーマンスを示し、3D医用画像セグメンテーションやビデオアクション認識などのタスクのパラメーターと計算コストを大幅に削減して、最先端の方法と同等のパフォーマンスを実現します。コードとモデルはhttps://github.com/yucornetto/CAKESで入手できます。
3D Convolution Neural Networks (CNNs) have been widely applied to 3D scene understanding, such as video analysis and volumetric image recognition. However, 3D networks can easily lead to over-parameterization which incurs expensive computation cost. In this paper, we propose Channel-wise Automatic KErnel Shrinking (CAKES), to enable efficient 3D learning by shrinking standard 3D convolutions into a set of economic operations e.g., 1D, 2D convolutions. Unlike previous methods, CAKES performs channel-wise kernel shrinkage, which enjoys the following benefits: 1) enabling operations deployed in every layer to be heterogeneous, so that they can extract diverse and complementary information to benefit the learning process; and 2) allowing for an efficient and flexible replacement design, which can be generalized to both spatial-temporal and volumetric data. Further, we propose a new search space based on CAKES, so that the replacement configuration can be determined automatically for simplifying 3D networks. CAKES shows superior performance to other methods with similar model size, and it also achieves comparable performance to state-of-the-art with much fewer parameters and computational costs on tasks including 3D medical imaging segmentation and video action recognition. Codes and models are available at https://github.com/yucornetto/CAKES
updated: Sun Dec 13 2020 21:46:21 GMT+0000 (UTC)
published: Sat Mar 28 2020 14:21: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アソシエイト