arXiv reaDer
CONet:畳み込みニューラルネットワークのチャネル最適化
CONet: Channel Optimization for Convolutional Neural Networks
Neural Architecture Search(NAS)は、ネットワーク設計を人間の直感の使用から評価メトリックに基づく検索アルゴリズムの活用にシフトしました。畳み込みニューラルネットワーク(CNN)でチャネルサイズの最適化を研究し、モデルの精度と複雑さにおいてそれが果たす役割を特定します。現在のチャネルサイズの選択方法は、通常、手動の反復と単純なヒューリスティックに悩まされている一方で、離散サンプルスペースによって制限されています。これを解決するために、特定のCNNのネットワークレイヤー全体でチャネルサイズを自動的に最適化する効率的な動的スケーリングアルゴリズム(CONet)を導入します。トレーニングで蓄積された情報を識別するために、「ランク」と「ランク平均勾配」の2つのメトリックが導入されています。アルゴリズムは、固定検索フェーズでチャネルサイズを動的に拡大または縮小します。CIFAR10/ 100とImageNetで実験を行います。データセットとは、CONetがResNet、DARTS、およびDARTS +スペースで検索された、ベースラインモデルよりも優れた効率的で正確なアーキテクチャを見つけることができることを示しています。
Neural Architecture Search (NAS) has shifted network design from using human intuition to leveraging search algorithms guided by evaluation metrics. We study channel size optimization in convolutional neural networks (CNN) and identify the role it plays in model accuracy and complexity. Current channel size selection methods are generally limited by discrete sample spaces while suffering from manual iteration and simple heuristics. To solve this, we introduce an efficient dynamic scaling algorithm -- CONet -- that automatically optimizes channel sizes across network layers for a given CNN. Two metrics -- ``Rank" and "Rank Average Slope" -- are introduced to identify the information accumulated in training. The algorithm dynamically scales channel sizes up or down over a fixed searching phase. We conduct experiments on CIFAR10/100 and ImageNet datasets and show that CONet can find efficient and accurate architectures searched in ResNet, DARTS, and DARTS+ spaces that outperform their baseline models.
updated: Sun Aug 15 2021 21:48:25 GMT+0000 (UTC)
published: Sun Aug 15 2021 21:48:25 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アソシエイト