arXiv reaDer
GradInit:安定した効率的なトレーニングのためにニューラルネットワークを初期化する方法を学ぶ
GradInit: Learning to Initialize Neural Networks for Stable and Efficient Training
ニューラルアーキテクチャの革新は、言語モデリングとコンピュータビジョンの大きな進歩を促進しました。残念ながら、新しいアーキテクチャでは、ネットワークパラメータが適切に初期化されていないと、ハイパーパラメータの選択が困難になり、トレーニングが不安定になることがよくあります。アーキテクチャ固有の初期化スキームがいくつか提案されていますが、これらのスキームは常に新しいアーキテクチャに移植できるとは限りません。このホワイトペーパーでは、ニューラルネットワークを初期化するための自動化されたアーキテクチャにとらわれない方法であるGradInitについて説明します。 GradInitは、単純なヒューリスティックに基づいています。各ネットワーク層のノルムは、規定のハイパーパラメータを使用したSGDまたはAdamの1つのステップで、可能な限り最小の損失値が得られるように調整されます。この調整は、各パラメーターブロックの前にスカラー乗数変数を導入し、単純な数値スキームを使用してこれらの変数を最適化することによって行われます。 GradInitは、スキップ接続の有無にかかわらず、さらに正規化レイヤーがなくても、多くの畳み込みアーキテクチャの収束とテストパフォーマンスを加速します。また、機械翻訳用の元のTransformerアーキテクチャの安定性が向上し、幅広い学習率と運動量係数の下でAdamまたはSGDを使用して、学習率のウォームアップなしでトレーニングできるようになります。コードはhttps://github.com/zhuchen03/gradinitで入手できます。
Innovations in neural architectures have fostered significant breakthroughs in language modeling and computer vision. Unfortunately, novel architectures often result in challenging hyper-parameter choices and training instability if the network parameters are not properly initialized. A number of architecture-specific initialization schemes have been proposed, but these schemes are not always portable to new architectures. This paper presents GradInit, an automated and architecture agnostic method for initializing neural networks. GradInit is based on a simple heuristic; the norm of each network layer is adjusted so that a single step of SGD or Adam with prescribed hyperparameters results in the smallest possible loss value. This adjustment is done by introducing a scalar multiplier variable in front of each parameter block, and then optimizing these variables using a simple numerical scheme. GradInit accelerates the convergence and test performance of many convolutional architectures, both with or without skip connections, and even without normalization layers. It also improves the stability of the original Transformer architecture for machine translation, enabling training it without learning rate warmup using either Adam or SGD under a wide range of learning rates and momentum coefficients. Code is available at https://github.com/zhuchen03/gradinit.
updated: Wed Nov 24 2021 09:13:08 GMT+0000 (UTC)
published: Tue Feb 16 2021 11:45:35 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アソシエイト