arXiv reaDer
EEEA-Net:早期終了の進化的ニューラルアーキテクチャ検索
EEEA-Net: An Early Exit Evolutionary Neural Architecture Search
この研究の目的は、コンピューティングリソースが限られているオンデバイスプロセッサに適した畳み込みニューラルネットワーク(CNN)アーキテクチャを検索し、ネットワークアーキテクチャ検索(NAS)のコストを大幅に削減することでした。両方の目標を達成するために、進化的アルゴリズム(EA)の早期終了人口初期化(EE-PI)というタイトルの新しいアルゴリズムが開発されました。 EE-PIは、最大しきい値よりも少ないパラメーターでモデルをフィルター処理することにより、検索プロセスのパラメーターの総数を減らします。これらのモデルをしきい値を超えるパラメーターに置き換える新しいモデルを探します。これにより、同じパフォーマンスまたは精度を維持しながら、パラメータの数、モデルの保存に使用するメモリ、および処理時間を削減できます。検索時間は0.52GPU日に短縮されました。これは、NSGA-Netを使用して達成された4 GPU日、AmoebaNetモデルで3,150 GPU日、NASNetモデルで2,000GPU日というNASと比較して非常に重要な成果です。同様に、早期終了進化的アルゴリズムネットワーク(EEEA-Nets)は、ネットワークアルゴリズムのクラスとして、特定のデータセットに適した最小限のエラーと計算コストでネットワークアーキテクチャを生成します。 CIFAR-10、CIFAR-100、およびImageNetデータセットでEEEA-Netを使用した実験では、EEEA-Netが最先端のNASモデルの中で最も低いエラー率を達成し、CIFAR-10では2.46%、15.02%であることが示されました。 CIFAR-100の場合、およびImageNetデータセットの場合は23.8%。さらに、オブジェクト検出、セマンティックセグメンテーション、キーポイント検出タスクなどの他のタスクにこの画像認識アーキテクチャを実装しました。実験では、EEEA-Net-C2はこれらのさまざまなタスクすべてでMobileNet-V3を上回りました。 (アルゴリズムコードはhttps://github.com/chakkritte/EEEA-Netで入手できます)。
The goals of this research were to search for Convolutional Neural Network (CNN) architectures, suitable for an on-device processor with limited computing resources, performing at substantially lower Network Architecture Search (NAS) costs. A new algorithm entitled an Early Exit Population Initialisation (EE-PI) for Evolutionary Algorithm (EA) was developed to achieve both goals. The EE-PI reduces the total number of parameters in the search process by filtering the models with fewer parameters than the maximum threshold. It will look for a new model to replace those models with parameters more than the threshold. Thereby, reducing the number of parameters, memory usage for model storage and processing time while maintaining the same performance or accuracy. The search time was reduced to 0.52 GPU day. This is a huge and significant achievement compared to the NAS of 4 GPU days achieved using NSGA-Net, 3,150 GPU days by the AmoebaNet model, and the 2,000 GPU days by the NASNet model. As well, Early Exit Evolutionary Algorithm networks (EEEA-Nets) yield network architectures with minimal error and computational cost suitable for a given dataset as a class of network algorithms. Using EEEA-Net on CIFAR-10, CIFAR-100, and ImageNet datasets, our experiments showed that EEEA-Net achieved the lowest error rate among state-of-the-art NAS models, with 2.46% for CIFAR-10, 15.02% for CIFAR-100, and 23.8% for ImageNet dataset. Further, we implemented this image recognition architecture for other tasks, such as object detection, semantic segmentation, and keypoint detection tasks, and, in our experiments, EEEA-Net-C2 outperformed MobileNet-V3 on all of these various tasks. (The algorithm code is available at https://github.com/chakkritte/EEEA-Net).
updated: Fri Aug 13 2021 10:23:19 GMT+0000 (UTC)
published: Fri Aug 13 2021 10:23:19 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アソシエイト