Stochastic Gradient Descent (SGD)는 convex 또는 non-convex function의 optimization에 일반적으로 사용되는 알고리즘입니다. SGD의 한가지 문제점은 learning rate에 매우 민감합니다. 특히 데이터가 sparse이며 features가 서로다른 frequencies (빈도수)를 갖고 있다면, 단 하나의 learning rate가 모든 weight update에 영향을 주는것은 문제가 될 수 있습니다.
즉 모든 weightsθθ에 대해서 동일한 learning rateηη를 사용하여 update를 하였습니다.
AgaGrad는 각각의 데이터에 dynamically adapt함으로서 궁극적으로 각각의 feature들마다 서로다른 learning rate로 연산을 합니다.
gt,i=∇θJ(θi)gt,i=∇θJ(θi)
이렇게 해주기 위해서는.. 실제로는 sum을 제거해야 합니다. 2N∑y−^y2N∑y−y^ 이게 아니고2Ny−^y2Ny−y^ 즉 weight값의 gradient값을 구하는게 아니라, 각각 weight elementθiθi들의 gradientgt,igt,i값을 구해야 합니다.
쉽게 이야기해서, 빈도수가 낮은 데이터에는 높은 learning rate를 적용하고, 빈도수가 높은 데이터에는 낮은 learning rate를 자동으로 적용합니다. 따라서 sparse한 데이터에 적합하며, 대표적인 예가 구글 X lab에서 16000대의 cores를 사용하여 유튜브 동영상안의 고양이를 detection하는 모델에 Adagrad를 사용한것입니다.
Global Learning Rateηη가 필요합니다.
weightsθθ에 대한 초기화가 필요합니다.
small constantϵϵ값을 설정합니다. (numerical stability를 위해서 약10−810−8로 설정, 즉 분모에 들어가게 되는데 0이 안되도록 매우 작은 수를 넣음)
gradient accumulation variableGG값은 weight와 동일한 shape인 0으로 초기화합니다.
먼저 gradient를 구한뒤 sqaured gradient (gradient의 제곱)을 r에 accumulation해줍니다.
AdaGrad의 장점은 보통 다른 optimizers의 경우에는 learning rate를 일반적으로 0.01로 놓고 변경되지 않지만, Adagram같은 경우는 자동으로 learning rate를 계산해준다는 것입니다. 반면 약점은 denominator부분의 accumulation of the squared gradients의 값이 지속적으로 쌓여서 계속 늘어난다는 단점이 있습니다. 결국 learning rate는 점점 작아진다는 약점을 갖고있습니다. 이러한 약점은 극복한 것이 Adadelta입니다.
Neural network의 weight을 조절하는 과정에는 보통‘Gradient Descent’라는 방법을 사용한다. 이는 네트워크의 parameter들을θθ라고 했을 때, 네트워크에서 내놓는 결과값과 실제 결과값 사이의 차이를 정의하는 함수 Loss functionJ(θ)J(θ)의 값을 최소화하기 위해 기울기(gradient)∇θJ(θ)∇θJ(θ)를 이용하는 방법이다. Gradient Descent에서는θθ에 대해 gradient의 반대 방향으로 일정 크기만큼 이동해내는 것을 반복하여 Loss functionJ(θ)J(θ)의 값을 최소화하는θθ의 값을 찾는다. 한 iteration에서의 변화 식은 다음과 같다.
θ=θ−η∇θJ(θ)θ=θ−η∇θJ(θ)
이 때ηη는 미리 정해진 걸음의 크기 ‘step size’ 로서, 보통 0.01~0.001 정도의 적당한 크기를 사용한다.
이 때 Loss Function을 계산할 때 전체 train set을 사용하는 것을BatchGradient Descent 라고 한다. 그러나 이렇게 계산을 할 경우 한번 step을 내딛을 때 전체 데이터에 대해 Loss Function을 계산해야 하므로 너무 많은 계산량이 필요하다. 이를 방지하기 위해 보통은StochasticGradient Descent (SGD) 라는 방법을 사용한다. 이 방법에서는 loss function을 계산할 때 전체 데이터(batch) 대신 일부 조그마한 데이터의 모음(mini-batch)에 대해서만 loss function을 계산한다. 이 방법은 batch gradient descent 보다 다소 부정확할 수는 있지만, 훨씬 계산 속도가 빠르기 때문에 같은 시간에 더 많은 step을 갈 수 있으며 여러 번 반복할 경우 보통 batch의 결과와 유사한 결과로 수렴한다. 또한, SGD를 사용할 경우 Batch Gradient Descent에서 빠질 local minima에 빠지지 않고 더 좋은 방향으로 수렴할 가능성도 있다.
보통 Neural Network를 트레이닝할 때는 이 SGD를 이용한다. 그러나 단순한 SGD를 이용하여 네트워크를 학습시키는 것에는 한계가 있다. 결론부터 살펴보기 위해, 다음과 같은 그림들을 살펴보자.
Gradient Descent Optimization Algorithms at Long Valley
Gradient Descent Optimization Algorithms at Beale's Function
Gradient Descent Optimization Algorithms at Saddle Point
위의 그림들은 각각 SGD 및 SGD의 변형 알고리즘들이 최적값을 찾는 과정을 시각화한 것이다. 빨간색의 SGD가 우리가 알고 있는 Naive Stochastic Gradient Descent 알고리즘이고, Momentum, NAG, Adagrad, AdaDelta, RMSprop 등은 SGD의 변형이다. 보다시피 모든 경우에서 SGD는 다른 알고리즘들에 비해 성능이 월등하게 낮다. 다른 알고리즘들 보다 이동속도가 현저하게 느릴뿐만 아니라, 방향을 제대로 잡지 못하고 이상한 곳에서 수렴하여 이동하지 못하는 모습도 관찰할 수 있다. 즉 단순한 SGD를 이용하여 네트워크를 학습시킬 경우 네트워크가 상대적으로 좋은 결과를 얻지 못할 것이라고 예측할 수 있다. 그렇다면 실제로는 어떤 방법들을 이용해야 하는 것인가? 이 글에서는 Neural Network를 학습시킬 때 실제로 많이 사용하는 다양한 SGD의 변형 알고리즘들을 간략하게 살펴보겠다. 내용과 그림의 상당 부분은Sebastian Ruder의 글에서 차용했다.
Momentum
Momentum 방식은 말 그대로 Gradient Descent를 통해 이동하는 과정에 일종의 ‘관성’을 주는 것이다. 현재 Gradient를 통해 이동하는 방향과는 별개로, 과거에 이동했던 방식을 기억하면서 그 방향으로 일정 정도를 추가적으로 이동하는 방식이다. 수식으로 표현하면 다음과 같다.vtvt를 time step t에서의 이동 벡터라고 할 때, 다음과 같은 식으로 이동을 표현할 수 있다.
vt=γvt−1+η∇θJ(θ)vt=γvt−1+η∇θJ(θ)
θ=θ−vtθ=θ−vt
이 때,γγ는 얼마나 momentum을 줄 것인지에 대한 momentum term으로서, 보통 0.9 정도의 값을 사용한다. 식을 살펴보면 과거에 얼마나 이동했는지에 대한 이동 항 v를 기억하고, 새로운 이동항을 구할 경우 과거에 이동했던 정도에 관성항만큼 곱해준 후 Gradient을 이용한 이동 step 항을 더해준다. 이렇게 할 경우 이동항vtvt는 다음과 같은 방식으로 정리할 수 있어, Gradient들의 지수평균을 이용하여 이동한다고도 해석할 수 있다.
또한 Momentum 방식을 이용할 경우 위의 그림과 같이 local minima를 빠져나오는 효과가 있을 것이라고도 기대할 수 있다. 기존의 SGD를 이용할 경우 좌측의 local minima에 빠지면 gradient가 0이 되어 이동할 수가 없지만, momentum 방식의 경우 기존에 이동했던 방향에 관성이 있어 이 local minima를 빠져나오고 더 좋은 minima로 이동할 것을 기대할 수 있게 된다. 반면 momentum 방식을 이용할 경우 기존의 변수들θθ외에도 과거에 이동했던 양을 변수별로 저장해야하므로 변수에 대한 메모리가 기존의 두 배로 필요하게 된다.
Nesterov Accelerated Gradient (NAG)
Nesterov Accelerated Gradient(NAG)는 Momentum 방식을 기초로 한 방식이지만, Gradient를 계산하는 방식이 살짝 다르다. 빠른 이해를 위해 다음 그림을 먼저 살펴보자.
Difference between Momentum and NAG. Picture from CS231.
Momentum 방식에서는 이동 벡터vtvt를 계산할 때 현재 위치에서의 gradient와 momentum step을 독립적으로 계산하고 합친다. 반면, NAG에서는 momentum step을 먼저 고려하여, momentum step을 먼저 이동했다고 생각한 후 그 자리에서의 gradient를 구해서 gradient step을 이동한다. 이를 수식으로 나타내면 다음과 같다.
vt=γvt−1+η∇θJ(θ−γvt−1)vt=γvt−1+η∇θJ(θ−γvt−1)
θ=θ−vtθ=θ−vt
NAG를 이용할 경우 Momentum 방식에 비해 보다 효과적으로 이동할 수 있다. Momentum 방식의 경우 멈춰야 할 시점에서도 관성에 의해 훨씬 멀리 갈수도 있다는 단점이 존재하는 반면, NAG 방식의 경우 일단 모멘텀으로 이동을 반정도 한 후 어떤 방식으로 이동해야할 지를 결정한다. 따라서 Momentum 방식의 빠른 이동에 대한 이점은 누리면서도, 멈춰야 할 적절한 시점에서 제동을 거는 데에 훨씬 용이하다고 생각할 수 있을 것이다.
Adagrad
Adagrad(Adaptive Gradient)는 변수들을 update할 때 각각의 변수마다 step size를 다르게 설정해서 이동하는 방식이다. 이 알고리즘의 기본적인 아이디어는 ‘지금까지 많이 변화하지 않은 변수들은 step size를 크게 하고, 지금까지 많이 변화했던 변수들은 step size를 작게 하자’ 라는 것이다. 자주 등장하거나 변화를 많이 한 변수들의 경우 optimum에 가까이 있을 확률이 높기 때문에 작은 크기로 이동하면서 세밀한 값을 조정하고, 적게 변화한 변수들은 optimum 값에 도달하기 위해서는 많이 이동해야할 확률이 높기 때문에 먼저 빠르게 loss 값을 줄이는 방향으로 이동하려는 방식이라고 생각할 수 있겠다. 특히 word2vec이나 GloVe 같이 word representation을 학습시킬 경우 단어의 등장 확률에 따라 variable의 사용 비율이 확연하게 차이나기 때문에 Adagrad와 같은 학습 방식을 이용하면 훨씬 더 좋은 성능을 거둘 수 있을 것이다.
Adagrad의 한 스텝을 수식화하여 나타내면 다음과 같다.
Gt=Gt−1+(∇θJ(θt))2Gt=Gt−1+(∇θJ(θt))2
θt+1=θt−ηGt+ϵ‾‾‾‾‾‾√⋅∇θJ(θt)θt+1=θt−ηGt+ϵ⋅∇θJ(θt)
Neural Network의 parameter가 k개라고 할 때,GtGt는 k차원 벡터로서 ‘time step t까지 각 변수가 이동한 gradient의 sum of squares’ 를 저장한다.θθ를 업데이트하는 상황에서는 기존 step sizeηη에GtGt의 루트값에 반비례한 크기로 이동을 진행하여, 지금까지 많이 변화한 변수일 수록 적게 이동하고 적게 변화한 변수일 수록 많이 이동하도록 한다. 이 때ϵϵ은10−410−4~10−810−8정도의 작은 값으로서 0으로 나누는 것을 방지하기 위한 작은 값이다. 여기에서GtGt를 업데이트하는 식에서 제곱은 element-wise 제곱을 의미하며,θθ를 업데이트하는 식에서도⋅⋅은 element-wise한 연산을 의미한다.
Adagrad를 사용하면 학습을 진행하면서 굳이 step size decay등을 신경써주지 않아도 된다는 장점이 있다. 보통 adagrad에서 step size로는 0.01 정도를 사용한 뒤, 그 이후로는 바꾸지 않는다. 반면, Adagrad에는 학습을 계속 진행하면 step size가 너무 줄어든다는 문제점이 있다.GG에는 계속 제곱한 값을 넣어주기 때문에GG의 값들은 계속해서 증가하기 때문에, 학습이 오래 진행될 경우 step size가 너무 작아져서 결국 거의 움직이지 않게 된다. 이를 보완하여 고친 알고리즘이 RMSProp과 AdaDelta이다.
RMSProp
RMSProp은 딥러닝의 대가 제프리 힌톤이 제안한 방법으로서, Adagrad의 단점을 해결하기 위한 방법이다. Adagrad의 식에서 gradient의 제곱값을 더해나가면서 구한GtGt부분을 합이 아니라 지수평균으로 바꾸어서 대체한 방법이다. 이렇게 대체를 할 경우 Adagrad처럼GtGt가 무한정 커지지는 않으면서 최근 변화량의 변수간 상대적인 크기 차이는 유지할 수 있다. 식으로 나타내면 다음과 같다.
G=γG+(1−γ)(∇θJ(θt))2G=γG+(1−γ)(∇θJ(θt))2
θ=θ−ηG+ϵ‾‾‾‾‾‾√⋅∇θJ(θt)θ=θ−ηG+ϵ⋅∇θJ(θt)
AdaDelta
AdaDelta (Adaptive Delta) 는 RMSProp과 유사하게 AdaGrad의 단점을 보완하기 위해 제안된 방법이다. AdaDelta는 RMSProp과 동일하게GG를 구할 때 합을 구하는 대신 지수평균을 구한다. 다만, 여기에서는 step size를 단순하게ηη로 사용하는 대신 step size의 변화값의 제곱을 가지고 지수평균 값을 사용한다.
G=γG+(1−γ)(∇θJ(θt))2G=γG+(1−γ)(∇θJ(θt))2
Δθ=s+ϵ‾‾‾‾‾√G+ϵ‾‾‾‾‾‾√⋅∇θJ(θt)Δθ=s+ϵG+ϵ⋅∇θJ(θt)
θ=θ−Δθθ=θ−Δθ
s=γs+(1−γ)Δ2θs=γs+(1−γ)Δθ2
얼핏 보면 왜 이러한 식이 도출되었는지 이해가 안될 수 있지만, 이는 사실 Gradient Descent와 같은 first-order optimization 대신 Second-order optimization을 approximate 하기 위한 방법이다. 실제로논문의 저자는 SGD, Momentum, Adagrad와 같은 식들의 경우ΔθΔθ의 unit(단위)을 구해보면θθ의 unit이 아니라θθ의 unit의 역수를 따른다는 것을 지적한다.θθ의 unit을u(θ)u(θ)라고 하고 J는 unit이 없다고 생각할 경우, first-order optimization에서는
Δθ∝∂J∂θ∝1u(θ)Δθ∝∂J∂θ∝1u(θ)
이다. 반면, Newton method와 같은 second-order optimization을 생각해보면
Δθ∝∂J∂θ∂2J∂θ2∝u(θ)Δθ∝∂J∂θ∂2J∂θ2∝u(θ)
이므로 바른 unit을 가지게 된다. 따라서 저자는 Newton’s method를 이용하여ΔθΔθ가∂J∂θ∂2J∂θ2∂J∂θ∂2J∂θ2라고 생각한 후,1∂2J∂θ2=Δθ∂J∂θ1∂2J∂θ2=Δθ∂J∂θ이므로 이를 분자의 Root Mean Square, 분모의 Root Mean Square 값의 비율로 근사한 것이다. 더욱 자세한 설명을 원하시는 분은 논문을 직접 읽어보시길 바란다.
Adam
Adam (Adaptive Moment Estimation)은 RMSProp과 Momentum 방식을 합친 것 같은 알고리즘이다. 이 방식에서는 Momentum 방식과 유사하게 지금까지 계산해온 기울기의 지수평균을 저장하며, RMSProp과 유사하게 기울기의 제곱값의 지수평균을 저장한다.
다만, Adam에서는 m과 v가 처음에 0으로 초기화되어 있기 때문에 학습의 초반부에서는mt,vtmt,vt가 0에 가깝게 bias 되어있을 것이라고 판단하여 이를 unbiased 하게 만들어주는 작업을 거친다.mtmt와vtvt의 식을∑∑형태로 펼친 후 양변에 expectation을 씌워서 정리해보면, 다음과 같은 보정을 통해 unbiased 된 expectation을 얻을 수 있다. 이 보정된 expectation들을 가지고 gradient가 들어갈 자리에mt^mt^,GtGt가 들어갈 자리에vt^vt^를 넣어 계산을 진행한다.
mt^=mt1−βt1mt^=mt1−β1t
vt^=vt1−βt2vt^=vt1−β2t
θ=θ−ηvt^+ϵ‾‾‾‾‾‾√mt^θ=θ−ηvt^+ϵmt^
보통β1β1로는 0.9,β2β2로는 0.999,ϵϵ으로는10−810−8정도의 값을 사용한다고 한다.
