定理証明済み
勾配降下法の収束速度
内容
L が凸で L-滑らか(その勾配がリプシッツ定数 L を持つ: ∥∇L(x)−∇L(y)∥≤L∥x−y∥)であり、θ⋆ が L を最小化するとする。ステップ幅 η=1/L の勾配降下法は、K ステップ後に L(θK)−L(θ⋆)≤2KL∥θ0−θ⋆∥2 を満たす。
なぜ正しいのか?
滑らかさにより、各勾配ステップは勾配の大きさの二乗に比例する量だけ L を減少させることが保証されるため、L(θk) は決して増加しない。凸性により、そのステップごとの減少量をまだ残っているギャップ L(θk)−L(θ⋆) と比較できる。K ステップにわたって保証された減少量を足し合わせる — 望遠鏡和 — と、総減少量が有界であることが分かり、非増加数列における最良点は平均以上に良いはずなので、最終的なギャップは 1/K のように縮小するはずである。
証明の概略
L-滑らかさにより、任意の x,y に対して: L(y)≤L(x)+∇L(x)⊤(y−x)+2L∥y−x∥2。y=θk+1=θk−L1∇L(θk)、x=θk とおくと降下補題が得られる: L(θk+1)≤L(θk)−2L1∥∇L(θk)∥2。
L の凸性により: L(θk)≤L(θ⋆)+∇L(θk)⊤(θk−θ⋆)。これを降下補題に加えると: L(θk+1)−L(θ⋆)≤∇L(θk)⊤(θk−θ⋆)−2L1∥∇L(θk)∥2。
右辺を平方完成すると: ∇L(θk)⊤(θk−θ⋆)−2L1∥∇L(θk)∥2=2L(∥θk−θ⋆∥2−θk−θ⋆−L1∇L(θk)2)=2L(∥θk−θ⋆∥2−∥θk+1−θ⋆∥2)、なぜなら θk+1−θ⋆=θk−θ⋆−L1∇L(θk) であるから。
よって L(θk+1)−L(θ⋆)≤2L(∥θk−θ⋆∥2−∥θk+1−θ⋆∥2)。これを k=0,…,K−1 について足し合わせると、右辺は望遠鏡式に縮約されて 2L(∥θ0−θ⋆∥2−∥θK−θ⋆∥2)≤2L∥θ0−θ⋆∥2 となる。
降下補題はまた L(θk) が非増加であることも示すので、L(θK) は L(θ1),…,L(θK) の平均以下である: L(θK)−L(θ⋆)≤K1∑k=1K(L(θk)−L(θ⋆))≤2KL∥θ0−θ⋆∥2 となり、まさに主張された評価式が得られる。
ステップごとの証明
この定理のステップごとの証明はまだありません。
参考文献
- Ian Goodfellow, Yoshua Bengio, Aaron Courville (2016). Deep Learning
- Christopher M. Bishop (2006). Pattern Recognition and Machine Learning
- Chiyuan Zhang, Samy Bengio, Moritz Hardt, Benjamin Recht, Oriol Vinyals (2017). Understanding deep learning requires rethinking generalization · arXiv:1611.03530
- Arthur Jacot, Franck Gabriel, Clément Hongler (2018). Neural Tangent Kernel: Convergence and Generalization in Neural Networks · arXiv:1806.07572
- Mikhail Belkin, Daniel Hsu, Siyuan Ma, Soumik Mandal (2019). Reconciling modern machine learning practice and the classical bias-variance trade-off · arXiv:1812.11118