+4

Một số thuật toán tối ưu trong DeepLearning

Vài lời bộc bạch

Đây là lần đầu tiên mình viết những bài viết mở và public như vậy, nếu như bài viết có bất cứ sai sót, hay góp ý gì cứ thì đừng ngại nói cho mình biết trong phần bình luận để mình sửa nhé. Chúc mọi người đọc vui vẻ

Gradient decent (GD)

Gradient decent là một thuật toán cơ bản trong machine learning nói chung và deep learning nói riêng để tối ưu các tham số cho mô hình học máy. Ý tưởng cơ bản nhất của gradient decent là dựa vào tính chất f(x)-\nabla f(x) luôn hướng về phía cực trị của hàm số f(x)f(x), và càng gần với cực trị thì giá trị của f(x)-\nabla f(x) càng tiến về không. Vì thế để tìm kiếm cực trị lân cận của một hàm số f(x)f(x) bất kì, ta chọn một điểm x0x_0 để bắt đầu thuật toán GD và cập nhật giá trị của x0x_0 theo công thức truy hồi sau:

xn+1=xnηf(xn)x_{n+1}=x_n-\eta\cdot\nabla f(x_n)

Trong đó, η\eta được gọi là tốc độ học của thuật toán.

Đối với gradient decent, người ta thường chọn η\eta có độ lớn khoảng 0.010.01 đến 0.0010.001, 0.00010.0001 tùy vào loại dữ liệu.

https://machinelearningcoban.com/assets/GD/1dimg_5_0.1_-5.gif

Gradient trong GD thường được tính bằng hai cách, cách đầu tiên là dùng analytical gradient, có được bằng cách áp dụng công thức đạo hàm vào hàm mất mát để tính Gradient, ưu điểm của phương pháp này là khối lượng tính toán nhẹ và độ chính xác cao, nhưng nhược điểm là đôi lúc ta sẽ gặp những bài toán mà hàm mất mát phức tạp đến nỗi không thể tính được công thức đạo hàm. Điều đó dẫn đến cách tính thứ hai đó là numerical Gradient, cách làm này tuy độ chính xác không cao và khối lượng tính toán lớn nhưng bù lại thì nó có thể dùng cho mọi trường hợp hàm mất mát. numerical gradient thường được dùng để kiểm tra analytical gradient để bào đảm ta tính đúng Gradient. Numerical gradient được tính bằng cách sử dụng công thức sai phân hướng tâm:

f(x)=f(x+ϵ)f(xϵ)2ϵf'(x)=\frac{f(x+\epsilon)-f(x-\epsilon)}{2\epsilon}

Gradient decent với momentum

Trong một số trường hợp, thuật toán GD của chúng ta không tìm được một global minimum để thuật toán thu được một kết quả tốt mà lại bị kẹt tại một local minimum, không mong muốn.

https://machinelearningcoban.com/assets/GD/momentum.png

Để khắc phục điểm trừ này, ta thêm vào công thức cập nhật gradient nhân tố vnv_n có công thức như sau:

vn=γvn1+f(xn)v_n=\gamma v_{n-1}+\nabla f(x_n)

Hệ số γ\gamma trong công thức trên thường được đặt là 0.90.9 và từ đó công thức GD với momentum có công thức như sau

xn+1=xnvnx_{n+1}=x_n-v_{n}

xn+1=xnγvn1f(xn)\Leftrightarrow x_{n+1}=x_n-\gamma v_{n-1}-\nabla f(x_n)

Trong đó, nhìn theo góc nhìn vật lí thì vnv_n bao gồm γvn1\gamma v_{n-1} biểu thì cho “đà” của quả banh màu vàng giúp quả banh vàng vượt qua local minimum không mong muốn và thành phần gradient cơ bản f(xn)\nabla f(x_n), vì thế mà việc thêm “đà” này còn được gọi là rò rỉ động lượng, tức năng lượng ở lần cập nhật trước bị rò rỉ qua lần cập nhật sau với hệ số γ\gamma.

Nesterov accelerated gradient decent (NAG)

Tuy momentum giúp viên bi vàng vượt được dốc để tiến đến cực trị mong muốn nhưng điều này đồng thời cũng khiến nó dao động rất lâu xung quanh điểm cực tiểu trước khi dừng lại. Tuy nhiên, với Nesterov GD thì điều này sẽ được khắc phục. Với hai thuật toán trước, ta chỉ tính Gradient tại điểm đang xét, nhưng với NAG, gradient được tính tại điểm ở “tương lai”:

Thay vì cập nhật với:

vn=γvn1+f(xn)v_n=\gamma v_{n-1}+\nabla f(x_n)

NAG cập nhật với:

vn=γvn1+f(xnγvn1)v_n=\gamma v_{n-1}+\nabla f(x_n-\gamma v_{n-1})

xn+1=xnvn=xnγvn1f(xnγvn1)\Leftrightarrow x_{n+1}=x_n-v_n=x_n-\gamma v_{n-1}-\nabla f(x_n-\gamma v_{n-1})

Cập nhật với NAG

Cập nhật với NAG

GD with momentum

GD with momentum

Stochastic gradient decent (SGD)

Cập nhật Gradient chỉ với một điểm dữ liệu ngẫu nhiên và làm điều này trên toàn bộ dữ liệu. Mỗi lần thuật toán đi qua hết tất cả các điểm dữ liệu gọi là một epoch. Để huấn luyện với SGD, ta có thể sẽ phải chạy nhiều epoch.

Ưu điểm của SGD là tốc độ tính toán rất nhanh và không cần nhiều bộ nhớ. Vì thế phương pháp này được ứng dụng rất nhiều trong mảng online learning.

Batch and mini batch

Thay vì dùng một điểm để cập nhật, ta chia nhiều điểm vào chung một batch rồi dùng để cập nhật Gradient. điều này giúp tận dụng dữ liệu và cũng cải thiện tốc độ tính toán. Hàm mất mát của SGD và Batch, mini batch GD không phải bao giờ cũng giảm nhưng nhìn chung là hội tụ càng về cuối. Phương pháp này được sử dụng rộng rãi trong các bài toán Deep learning.

Stopping criteria

Có các chú ý sau khi dừng thuật toán GD:

  • Giới hạn số vòng lặp, hay số epoch để hạn chế tình trạng GD không tìm được nghiệm do learning rate quá lớn hoặc tránh tình trạng model của chúng ta bị overfit. Nhược điểm là đôi khi thuật toán GD chưa tìm được nghiệm thì đã bị dừng lại
  • So sánh giá trị gradient tại hai lần cập nhật liên tiếp, khi giá trị này đủ nhỏ thì dừng thuật toán. Nhược điểm là đôi khi việc tính gradient như vậy sẽ không có lợi khi dùng SGD hoặc mini batch vì sẽ tốn nhiều bộ nhớ khi có quá nhiều dữ liệu khiến ta không có được lợi thế khi dùng phương pháp này
  • So sánh giá trị của hàm mất mát của nghiệm tại hai lần cập nhật liên tiếp, khi nào giá trị này đủ nhỏ thì dừng lại. Nhược điểm của phương pháp này là nếu tại một thời điểm, điểm cập nhật của ta rơi vào vị trí điểm uốn tương đối bằng phẳng thì thuật toán cũng dừng lại trước khi đạt giá trị mong muốn.
  • Phương pháp cuối cùng là kiểm tra sau một vài lần cập nhật nhất định. Việc này giúp giảm khối lượng tính toán cũng như giúp thuật toán vượt được những điểm uốn và tím được cực trị phù hợp.

Advance optimization

Một số những trở ngại lớn trong bài toán tối ưu là các local minimum, các điểm uốn (điểm yên ngựa), và vấn đề tiêu biến/ bùng nổ Gradient (Vanishing/ exploding Gradient), ...

AdaGrad

Trong một số trường hợp, việc chọn learning rate phù hợp có thể trở thành một vấn đề nhức nhối. Vấn đề này sẽ được giải quyết với AdaGrad, vì thuật toán này sẽ coi learning rate là một tham số biến thiên sau mỗi lần chạy thuật toán.

AdaGrad thường được dùng trong các vấn đề về sparse feature (đặc trưng thưa), ví dụ như các bài toán về xử lí ngôn ngữ tự nhiên, quảng cáo điện toán, và lọc cộng tác. Thường những những đặc trưng này ít khi được cập nhật để có thể được tối ưu những lại mang những ý nghĩa vô cùng to lớn, còn những đặc trưng có tần số xuất hiện cao thì được cập nhật liên tục. Vì thế, việc có một learning rate cố định hoặc là quá cao đối với những feature được cập nhật nhiều lần hoặc là quá thấp đối với những feature ít được cập nhật.

Công thức của AdaGrad như sau:

gn=wL(yn,f(xt,w))g_n=\partial_wL(y_n,f(x_t,w))

sn=sn1+gn2s_n=s_{n-1}+g_n^2

wn=wn1ηsn+ϵgtw_n=w_{n-1}-\frac{\eta}{\sqrt{s_n+\epsilon}}\odot g_t

Trong đó s0s_0 thường được chọn bằng 00.

AdaGrad với lr=0.4

AdaGrad với lr=0.4

AdaGrad với lr=2

AdaGrad với lr=2

Ta có thể hiểu sự thay đổi learning rate này như một hệ số “ma sát”, “ma sát” này có giá trị lớn ở dốc và có giá trị nhỏ ở vùng bằng phẳng.

Phân tích ưu và nhược của AdaGrad:

Ưu điểm:

  • Hoạt động tốt với bài toán sparse feature, cụ thể là tối ưu các bài toán lồi, không dao động nhiều khi đạt vào cực tiểu cần thiết.
  • Learning rate được tự động điều chỉnh cho phù hợp
  • Chi phí tính toán không tăng nếu so với SGD

Nhược điểm:

  • Trong deep learning, các bài toán thường không hoàn toàn lồi và có nhiều local minimum gây nhiễu.
  • sns_n trong công thức cập nhật tăng một cách dường như tuyến tính khiến cho learning rate bị thu nhỏ nhanh.

RMSProp

AdaGrad với tốc độ học thay đổi để thích nghi với dữ liệu là một phương pháp tuyệt vời trong các bài toán tối ưu lồi, thế nhưng nó vẫn có những hạn chế như đã nêu trên. Vì thế RMSProp được đề xuất để khác phục những vấn đề của AdaGrad bằng cách kế thừa ý tưởng từ phương pháp momentum trong gradient decent làm rò rỉ các giá trị của rnr_n trong công thức cập nhật. Cụ thể công thức của RMSProp là:

gn=wL(yn,f(xt,w))g_n=\partial_wL(y_n,f(x_t,w))

sn=γsn1+(1γ)gn2s_n=\gamma s_{n-1}+(1-\gamma)g_n^2

wn=wn1ηsn+ϵgtw_n=w_{n-1}-\frac{\eta}{\sqrt{s_n+\epsilon}}\odot g_t

Việc làm cho sns_n bị rò rỉ cũng đồng thời chuẩn hóa giá của trị số đứng trước giá trị của sns_n, vì thế mà giải quyết được vấn đề learning rate bị thu nhỏ khiến thuật toán dừng sớm.

https://d2l.aivivn.com/_images/output_rmsprop_vn_5d44fd_3_0.svg

Ta thấy rằng RMSProp kế thừa được những ưu điểm của AdaGrad về việc tự điều chỉnh learning rate, thuật toán đã đi được xa hơn so với AdaGrad với cùng learning rate 0.40.4

Adam

Trong một bài toán tối ưu thường gặp, ta muốn thuật toán của mình đầu tiên vượt qua những local minimum nhiễu một cách dễ dàng, nhưng khi đạt được global minimum thì sẽ không dao động quá lâu mà tìm được vị trí phù hợp ngay. Điều này thúc đẩy ta tìm kiếm một thuật có hành vi được mô tả như trên.

Có một thuật toán đã hội tụ đủ mọi ưu điểm của những phương pháp trước đó, đó chính là Adam, được sử dụng rất phổ biến trong bài toán deep learning. Thuật toán của Adam được ví như một quả bóng có khối lượng lớn và có ma sát vì nó có động lượng rò rỉ để vượt qua các local minimum để tiến đến gobal ninimum và không dao động lâu vì có ma sát kế thừa từ AdaGrad

Thuật toán của Adam như sau:

gn=wL(yn,f(xt,w))g_n=\partial_wL(y_n,f(x_t,w))

vn=β1vn1+(1β1)gnv_n=\beta_1v_{n-1}+(1-\beta_1)g_n

sn=β2sn1+(1β2)gn2s_n=\beta_2 s_{n-1}+(1-\beta_2)g_n^2

v^n=vn1β1n,s^n=sn1β2n\hat{v}_n=\frac{v_n}{1-\beta_1^n}, \hat{s}_n=\frac{s_n}{1-\beta_2^n}

gn=ηv^ns^n+ϵg'_n=\frac{\eta \hat{v}_n}{\sqrt{\hat{s}_n}+\epsilon}

xn+1=xngnx_{n+1}=x_n-g'_n

Trong đó, β1\beta_1β2\beta_2 thường được chọn là 0.90.90.990.99. Các giá trị v0v_0s0s_0 cũng thường được cho bằng 00.

Tuy nhiên, Adam tuy có lợi thế về learning rate và động lượng nhưng trong một số trường hợp nhất định, Adam sẽ bắt đầu phân kì. Có một bài báo đã nêu lên và phân tích điều này: [Zaheer et al., 2018]

So sánh các thuật toán tối ưu:

https://miro.medium.com/max/900/1*U224pqhF4WUOZhfmDIWtxA.gif

https://miro.medium.com/max/900/1*STiRp7PW5yIrvYZupZA6nw.gif

Ngoài ra, ta vẫn có thể sử dụng các thuật toán đơn giản như SGD với momentum, hoặc chỉ SGD nhưng kết hợp với bộ định thời tốc độ học ( Learning rate scheduling) để đạt được tốc độ học mong muốn.

Tài liệu tham khảo

  • Sách Machine Learning Cơ bản. Tác giả: bác Vũ Hữu Tiệp
  • Sách Dive into deep learning. Nhiều tác giả. link bản tiếng anh. link bản tiếng Việt.
  • Ngoài ra mình còn tham khảo một số nguồn ngoài khác nhưng chính thì chỉ có hai nguồn dẫn trên thôi .

All rights reserved

Viblo
Hãy đăng ký một tài khoản Viblo để nhận được nhiều bài viết thú vị hơn.
Đăng kí