[2P3-J-2] 機械学習: ニューラルネットワークの進歩

[2P3-J-2-02] 貢献度分配を導入した方策勾配によるNeural Architecture Searchの高速化

〇佐藤 怜^1,2、秋本 洋平^1,2、佐久間 淳^1,2 (1. 筑波大学、2. 理化学研究所 革新知能統合研究センター)

キーワード：ニューラルアーキテクチャ探索、深層学習

Neural Architecture Searchが深層学習のアーキテクチャを自動的に設計するためのアプローチとして注目を集めている. 本研究では, アーキテクチャと重みを同時に最適化するone-shotアーキテクチャ探索の高速化を目的とし, 貢献度分配を導入した方策勾配による探索法を提案する. 提案手法では, 貢献度と呼ばれるアーキテクチャを構成する複数の部分それぞれに対しての報酬を定義し, これをアーキテクチャパラメータの勾配の計算に用いる. 実験で, 最近提案された勾配法によるone-shotアーキテクチャ探索の先行研究と比較し, 提案手法では同程度の精度のアーキテクチャをより高速に発見できることを示す.