arXiv reaDer
フィルタリング問題を解くことによる確率的勾配分散の減少
Stochastic Gradient Variance Reduction by Solving a Filtering Problem
ディープニューラルネットワーク(DNN)は通常、確率的勾配降下法(SGD)を使用して最適化されます。ただし、確率的サンプルを使用した勾配の推定は、ノイズが多く信頼性が低い傾向があるため、勾配の分散が大きくなり、収束が悪くなります。この論文では、フィルターの異なる設計で適応フィルタリング問題を解くことにより、局所勾配の一貫した推定を行う効率的な確率的最適化アルゴリズムであるフィルター勾配降下法(FGD)を提案します。私たちの方法は、現在の推定を強化するために履歴状態を組み込むことにより、確率的勾配降下法の分散を減らします。ノイズの多い勾配方向を修正するだけでなく、学習の収束を加速することができます。数値最適化とニューラルネットワークのトレーニングに対する提案されたフィルター勾配降下法の有効性を示します。ここでは、従来の運動量ベースの方法と比較して、優れた堅牢なパフォーマンスを実現します。私たちの知る限りでは、信号処理における勾配推定とフィルタリングの問題を類推することにより、フィルタリングを勾配推定に統合する実用的なソリューションを最初に提供しました。 (コードはhttps://github.com/Adamdad/Filter-Gradient-Decentで提供されています)
Deep neural networks (DNN) are typically optimized using stochastic gradient descent (SGD). However, the estimation of the gradient using stochastic samples tends to be noisy and unreliable, resulting in large gradient variance and bad convergence. In this paper, we propose Filter Gradient Decent~(FGD), an efficient stochastic optimization algorithm that makes the consistent estimation of the local gradient by solving an adaptive filtering problem with different design of filters. Our method reduces variance in stochastic gradient descent by incorporating the historical states to enhance the current estimation. It is able to correct noisy gradient direction as well as to accelerate the convergence of learning. We demonstrate the effectiveness of the proposed Filter Gradient Descent on numerical optimization and training neural networks, where it achieves superior and robust performance compared with traditional momentum-based methods. To the best of our knowledge, we are the first to provide a practical solution that integrates filtering into gradient estimation by making the analogy between gradient estimation and filtering problems in signal processing. (The code is provided in https://github.com/Adamdad/Filter-Gradient-Decent)
updated: Sat May 15 2021 01:53:12 GMT+0000 (UTC)
published: Tue Dec 22 2020 23:48:42 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アソシエイト