arXiv reaDer
HUGE2:エッジコンピューティング用の高度に絡み合っていない生成モデルエンジン
HUGE2: a Highly Untangled Generative-model Engine for Edge-computing
深層学習における著名な研究の一種として、生成モデルが最近の研究で広く研究されています。深層学習モデルの2つの研究部門である生成ネットワーク(GAN、VAE)とセマンティックセグメンテーションは、アップサンプリング操作、特に転置畳み込みと拡張畳み込みに大きく依存しています。ただし、これら2種類の畳み込みは、入力特徴マップまたはカーネルにそれぞれゼロを挿入することに関して、標準の畳み込みとは本質的に異なります。この明確な性質により、主に標準の畳み込み用に開発された、Darknet、Tensorflow、PyTorchなどの既存の深層学習エンジンまたはフレームワークのパフォーマンスが大幅に低下します。ディープラーニング領域のもう1つの傾向は、メモリリソースが不足しているエッジ/組み込みデバイスにモデルを展開することです。この作業では、エッジコンピューティング用の高度にもつれのない生成モデルエンジンまたはHUGE2を提案し、カーネルを分解し、基本的な行列乗算を実行してこれらの小さな畳み込みを解くことにより、エッジコンピューティングプラットフォームでこれら2つの特別な畳み込みを加速します。私たちが提案する方法は、はるかに小さいメモリフットプリントを使用するため、メモリアクセスがはるかに少なくなります。また、データアクセスパターンにより、キャッシュにすでにフェッチされているデータの再利用性が劇的に向上し、キャッシュの局所性が高まります。私たちのエンジンは、組み込みCPUで約5倍、組み込みGPUで約10倍の速度向上を実現し、メモリアクセスを50%以上削減します。
As a type of prominent studies in deep learning, generative models have been widely investigated in research recently. Two research branches of the deep learning models, the Generative Networks (GANs, VAE) and the Semantic Segmentation, rely highly on the upsampling operations, especially the transposed convolution and the dilated convolution. However, these two types of convolutions are intrinsically different from standard convolution regarding the insertion of zeros in input feature maps or in kernels respectively. This distinct nature severely degrades the performance of the existing deep learning engine or frameworks, such as Darknet, Tensorflow, and PyTorch, which are mainly developed for the standard convolution. Another trend in deep learning realm is to deploy the model onto edge/ embedded devices, in which the memory resource is scarce. In this work, we propose a Highly Untangled Generative-model Engine for Edge-computing or HUGE2 for accelerating these two special convolutions on the edge-computing platform by decomposing the kernels and untangling these smaller convolutions by performing basic matrix multiplications. The methods we propose use much smaller memory footprint, hence much fewer memory accesses, and the data access patterns also dramatically increase the reusability of the data already fetched in caches, hence increasing the localities of caches. Our engine achieves a speedup of nearly 5x on embedded CPUs, and around 10x on embedded GPUs, and more than 50% reduction of memory access.
updated: Sat May 08 2021 17:07:28 GMT+0000 (UTC)
published: Thu Jul 25 2019 17:21:52 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アソシエイト