arXiv reaDer
AutoDropout:ディープネットワークを正規化するためのドロップアウトパターンの学習
AutoDropout: Learning Dropout Patterns to Regularize Deep Networks
ニューラルネットワークはパラメータが多すぎることが多いため、積極的な正則化の恩恵を受けます。ドロップアウトや重み減衰などの従来の正則化方法は、ネットワークの入力と非表示状態の構造を利用しません。その結果、これらの従来の方法は、SpatialDropoutやDropBlockなどの構造を利用する方法よりも効果が低く、非表示状態の特定の連続領域で値をランダムにドロップしてゼロに設定します。ドロップアウトエリアの場所はランダムですが、SpatialDropoutとDropBlockのパターンは手動で設計および修正されています。ここでは、ドロップアウトパターンを学習することを提案します。私たちの方法では、コントローラーは、ConvNetやTransformerなどのターゲットネットワークのすべてのチャネルとレイヤーでドロップアウトパターンを生成することを学習します。次に、ターゲットネットワークはドロップアウトパターンでトレーニングされ、その結果の検証パフォーマンスは、コントローラーが学習するための信号として使用されます。この方法が、CIFAR-10とImageNetでの画像認識、およびPennTreebankとWikiText-2での言語モデリングの両方でうまく機能することを示します。学習したドロップアウトパターンは、PennTreebankの言語モデルからWMT2014の英語-フランス語翻訳など、さまざまなタスクやデータセットにも転送されます。コードが利用可能になります。
Neural networks are often over-parameterized and hence benefit from aggressive regularization. Conventional regularization methods, such as Dropout or weight decay, do not leverage the structures of the network's inputs and hidden states. As a result, these conventional methods are less effective than methods that leverage the structures, such as SpatialDropout and DropBlock, which randomly drop the values at certain contiguous areas in the hidden states and setting them to zero. Although the locations of dropout areas random, the patterns of SpatialDropout and DropBlock are manually designed and fixed. Here we propose to learn the dropout patterns. In our method, a controller learns to generate a dropout pattern at every channel and layer of a target network, such as a ConvNet or a Transformer. The target network is then trained with the dropout pattern, and its resulting validation performance is used as a signal for the controller to learn from. We show that this method works well for both image recognition on CIFAR-10 and ImageNet, as well as language modeling on Penn Treebank and WikiText-2. The learned dropout patterns also transfers to different tasks and datasets, such as from language model on Penn Treebank to Engligh-French translation on WMT 2014. Our code will be available.
updated: Tue Jan 05 2021 19:54:22 GMT+0000 (UTC)
published: Tue Jan 05 2021 19:54:22 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アソシエイト