học tăng cườngPPOđộ dốc chính sáchhọc máytrí tuệ nhân tạo
Huấn luyện ổn định trong PPO so với các phương pháp gradient chính sách không ổn định
Tối ưu hóa chính sách gần đúng (Proximal Policy Optimization - PPO) mang đến các hàm mục tiêu được cắt xén và tư duy vùng tin cậy cho học tăng cường, giảm đáng kể sự biến động thường gặp trong các phương pháp gradient chính sách truyền thống. Trong khi các phương pháp truyền thống như REINFORCE và các thuật toán actor-critic tiêu chuẩn có thể phân kỳ hoặc sụp đổ giữa chừng quá trình huấn luyện, thiết kế của PPO giữ cho các cập nhật được giới hạn và có thể tái tạo được giữa các lần chạy.
Điểm nổi bật
Mục tiêu được điều chỉnh của PPO giúp ngăn chặn sự sụp đổ chính sách thảm khốc mà các chính sách phân cấp thông thường thường gặp phải.
Thuật toán gradient chính sách cơ bản đòi hỏi phải điều chỉnh cẩn thận tốc độ học và đường cơ sở để tránh sự phân kỳ.
PPO tái sử dụng các lần triển khai trên nhiều kỷ nguyên, mang lại hiệu quả lấy mẫu tốt hơn so với các phương pháp thuần túy dựa trên chính sách.
PPO đã trở thành thuật toán tiêu chuẩn đằng sau các hệ thống RLHF được sử dụng để huấn luyện các mô hình ngôn ngữ quy mô lớn hiện đại.
Huấn luyện ổn định tại PPO là gì?
Một mục tiêu thay thế được cắt bớt nhằm giữ cho các cập nhật chính sách nằm trong phạm vi an toàn, ngăn ngừa các bước học tập mang tính phá hoại.
PPO được nhóm của John Schulman tại OpenAI giới thiệu vào năm 2017 như một sự cải tiến của TRPO.
Cơ chế cốt lõi sử dụng tỷ lệ xác suất được giới hạn trong khoảng từ 0,8 đến 1,2 để hạn chế mức độ sai lệch của chính sách mới so với chính sách cũ.
PPO có khả năng mở rộng hiệu quả trên hàng triệu bước môi trường và chạy trên một cụm GPU hoặc CPU duy nhất.
Nó đã trở thành thuật toán mặc định đằng sau nhiều hệ thống RLHF nổi tiếng được sử dụng để huấn luyện các mô hình ngôn ngữ quy mô lớn.
Các thử nghiệm thực nghiệm cho thấy PPO phục hồi từ các lỗi khởi tạo kém hiệu quả hơn nhiều so với các phương pháp cơ bản dựa trên thuật toán gradient chính sách thông thường.
Phương pháp Gradient Chính sách Không ổn định là gì?
Các thuật toán học tăng cường cổ điển cập nhật chính sách trực tiếp theo độ dốc của lợi nhuận kỳ vọng, thường tạo ra các đường cong học tập không ổn định.
REINFORCE, thuật toán gradient chính sách nền tảng, được Ronald Williams công bố vào năm 1992.
Các thuật toán gradient chính sách thông thường có độ biến động cao vì chúng dựa vào lợi nhuận Monte Carlo từ các tập dữ liệu đầy đủ.
Nếu không có các vùng tin cậy, một bản cập nhật lớn duy nhất có thể khiến chính sách sụp đổ và biến thành một hành động mang tính xác định suy biến.
Các phương pháp này thường yêu cầu tinh chỉnh siêu tham số rộng rãi, bao gồm cả việc giảm tốc độ học và định hình phần thưởng, để đạt được sự hội tụ.
Các biến thể của mô hình Actor-Critic như A2C giúp giảm sự biến thiên nhưng vẫn thiếu các ràng buộc cập nhật nghiêm ngặt mà mô hình PPO áp đặt.
Bảng So Sánh
Tính năng
Huấn luyện ổn định tại PPO
Phương pháp Gradient Chính sách Không ổn định
Cơ chế cập nhật
Mục tiêu thay thế bị cắt xén với tỷ lệ xác suất được giới hạn gần 1.0
Thuật toán leo dốc gradient thô dựa trên lợi nhuận kỳ vọng mà không có giới hạn cập nhật cứng.
Luyện tập sự ổn định
Cao — phục hồi nhanh chóng sau những bước đi sai lầm và hiếm khi đi chệch hướng.
Thấp — nhạy cảm với tốc độ học tập và thang điểm phần thưởng, dễ bị sụp đổ.
Hiệu quả mẫu
Độ khó trung bình; sử dụng nhiều chu kỳ SGD minibatch cho mỗi lần triển khai.
Thường kém hiệu quả trừ khi được kết hợp với các chỉ số cơ sở hoặc các thủ thuật giảm phương sai.
Độ phức tạp triển khai
Đơn giản — dung lượng mã tương đương với thuật toán Policy Gradient thông thường.
Về cơ bản thì cấu tạo đơn giản, nhưng việc ổn định nó đòi hỏi thêm kỹ thuật.
Độ nhạy của siêu tham số
Khá dễ sử dụng với nhiều tỷ lệ khung hình và tốc độ học khác nhau.
Rất nhạy cảm; những thay đổi nhỏ cũng có thể phá hỏng hoàn toàn quá trình huấn luyện.
Xử lý sai lệch
Chức năng cắt xén tích hợp hoạt động như một công cụ giảm thiểu phương sai ngầm định.
Cần sử dụng các kỹ thuật riêng biệt như đường cơ sở, GAE hoặc chuẩn hóa ưu thế.
Hiệu suất đồng hồ treo tường
Tốc độ nhanh trên phần cứng hiện đại nhờ tối ưu hóa bậc nhất.
Hiệu suất tương đương nhau ở mỗi bước, nhưng sự không ổn định thường làm lãng phí thời gian thực tế vào những lần chạy thất bại.
Các trường hợp sử dụng phổ biến
RLHF dành cho mô hình ngôn ngữ, robot, chơi game, điều khiển liên tục
Phân tích lý thuyết, môi trường đơn giản, giảng dạy học tăng cường
So sánh chi tiết
Triết lý cốt lõi của thuật toán
Ý tưởng cốt lõi của PPO là các cập nhật chính sách nên nhỏ và có thể đảo ngược. Bằng cách cắt bớt tỷ lệ xác suất giữa chính sách mới và cũ, thuật toán ngăn chặn bộ tối ưu hóa thực hiện một bước làm thay đổi hành vi quá mạnh mẽ chỉ trong một lần lặp. Các phương pháp gradient chính sách không ổn định lại đi theo hướng ngược lại: chúng theo dõi gradient thô của lợi nhuận kỳ vọng, tin tưởng rằng tốc độ học được điều chỉnh tốt sẽ giữ mọi thứ trong tầm kiểm soát. Trên thực tế, niềm tin đó thường sai lầm.
Tính ổn định và hành vi hội tụ
Một lần chạy PPO thường cho thấy đường cong học tập có nhiều nhiễu nhưng cải thiện đều đặn, với những lần giảm đột ngột và phục hồi trong vòng vài lần lặp. Ngược lại, thuật toán gradient chính sách thông thường có thể đạt đến trạng thái ổn định trong hàng nghìn bước và sau đó đột ngột sụp đổ khi một quỹ đạo có phần thưởng cao hiếm hoi đẩy các tham số vào vùng xấu. Mục tiêu bị cắt trong PPO hoạt động như một phanh an toàn, giới hạn ảnh hưởng của bất kỳ một tập hợp kinh nghiệm nào.
Chi phí kỹ thuật và điều chỉnh
Để thuật toán gradient chính sách (RPG) hoạt động ổn định thường đòi hỏi phải tinh chỉnh thủ công tốc độ học, hệ số chiết khấu, phần thưởng entropy và ngưỡng cắt gradient. PPO hợp nhất phần lớn công đoạn tinh chỉnh đó vào một siêu tham số cắt duy nhất, thường được đặt trong khoảng 0,1 đến 0,3, vốn hoạt động mạnh mẽ trên nhiều nhiệm vụ khác nhau. Đối với các nhóm triển khai hệ thống RL trong môi trường sản xuất, việc giảm bớt gánh nặng tinh chỉnh này sẽ trực tiếp dẫn đến chu kỳ lặp lại nhanh hơn.
Sự đánh đổi về hiệu quả lấy mẫu
PPO tái sử dụng mỗi lần chạy cho nhiều epoch cập nhật minibatch, giúp cải thiện hiệu quả lấy mẫu so với các phương pháp thuần túy dựa trên chính sách như REINFORCE. Tuy nhiên, việc tái sử dụng này cũng là lý do tại sao việc cắt xén lại quan trọng: nếu không có nó, thuật toán sẽ bị quá khớp với các quỹ đạo cũ. Các phương pháp gradient chính sách không ổn định thường chỉ thực hiện một lần chạy cho mỗi lần chạy, điều này khiến chúng kém hiệu quả hơn về mặt lấy mẫu nhưng cũng ít bị lỗi hơn ở chế độ cụ thể đó.
Áp dụng trong thế giới thực
PPO đã trở thành lựa chọn mặc định cho học tăng cường ứng dụng, cung cấp sức mạnh cho các hệ thống từ các tác nhân Dota 5v5 của OpenAI đến các đường dẫn RLHF đằng sau ChatGPT và các chatbot hiện đại khác. Các phương pháp gradient chính sách truyền thống vẫn có giá trị như các công cụ giảng dạy và là cơ sở trong các bài báo nghiên cứu, nhưng chúng hiếm khi xuất hiện trong các hệ thống sản xuất nơi độ tin cậy là điều quan trọng. Sự chuyển dịch sang PPO phản ánh một xu hướng rộng hơn trong học máy hướng tới các phương pháp hoạt động ngay lập tức mà không cần cấu hình phức tạp.
Ưu & Nhược điểm
Huấn luyện ổn định tại PPO
Ưu điểm
+Cập nhật cực kỳ ổn định
+Các siêu tham số dễ tha thứ
+Dễ thực hiện
+Kết quả thực nghiệm mạnh mẽ
Đã lưu
−Các bản cập nhật có phần thiên vị
−Có thể triển khai quá mức
−Cần chỉnh âm thanh.
−Ít tính thanh lịch về mặt lý thuyết hơn
Phương pháp Gradient Chính sách Không ổn định
Ưu điểm
+Về mặt lý thuyết là sạch sẽ
+Dễ dàng suy ra
+Rất tốt cho việc giảng dạy
+Tính toán thấp trên mỗi bước
Đã lưu
−Ước tính phương sai cao
−Dễ phân kỳ
−Cần phải tinh chỉnh nhiều.
−Hiệu quả lấy mẫu kém
Những hiểu lầm phổ biến
Huyền thoại
PPO chỉ là một phiên bản hoa mỹ hơn của REINFORCE mà không có cơ sở lý thuyết thực sự nào.
Thực tế
PPO được xây dựng dựa trên ý tưởng vùng tin cậy từ TRPO nhưng thay thế tối ưu hóa ràng buộc bằng một phép cắt thay thế bậc nhất. Phép cắt này cung cấp một xấp xỉ thực tế cho ràng buộc vùng tin cậy, đó là lý do tại sao nó hoạt động tốt trên thực tế mặc dù việc triển khai đơn giản hơn.
Huyền thoại
Thuật toán gradient chính sách thông thường luôn hội tụ nếu bạn sử dụng tốc độ học đủ nhỏ.
Thực tế
Tốc độ học chậm giúp làm chậm sự phân kỳ nhưng không loại bỏ hoàn toàn. Các quỹ đạo xấu vẫn có thể đẩy chính sách vào vùng suy thoái, và độ biến thiên cao của lợi nhuận Monte Carlo có nghĩa là các cập nhật hiệu quả lớn thỉnh thoảng là điều gần như không thể tránh khỏi nếu không có các ràng buộc rõ ràng.
Huyền thoại
PPO không thể được sử dụng cho các nhiệm vụ điều khiển liên tục.
Thực tế
PPO hoạt động cực kỳ hiệu quả trên các bộ dữ liệu chuẩn về điều khiển liên tục như chuyển động MuJoCo và thao tác robot. Mục tiêu bị cắt xén không phụ thuộc vào không gian hành động, và PPO với các chính sách Gaussian vẫn là một nền tảng vững chắc cho các bài toán từ đi bộ bốn chân đến thao tác tay khéo léo.
Huyền thoại
Các lý thuyết về độ dốc chính sách không ổn định đã lỗi thời và không còn được sử dụng trong nghiên cứu.
Thực tế
Phương pháp gradient chính sách cơ bản vẫn là nền tảng trong nghiên cứu học tăng cường. Chúng xuất hiện như những phương pháp cơ bản trong hầu hết các bài báo về thuật toán mới, và các biến thể như gradient chính sách tự nhiên vẫn là cơ sở cho các nghiên cứu hiện đại về vùng tin cậy và tối ưu hóa có ràng buộc.
Huyền thoại
PPO đảm bảo sự cải thiện đều đặn trong mỗi lần chạy luyện tập.
Thực tế
PPO cải thiện đáng kể tính ổn định nhưng không đảm bảo sự tiến bộ đều đặn. Đường cong học tập vẫn chứa nhiễu, và các hàm thưởng bất thường hoặc tín hiệu cực kỳ thưa thớt vẫn có thể gây ra thất bại. Tính ổn định có nghĩa là ít sự sụp đổ thảm khốc hơn, chứ không phải là không có thất bại nào.
Các câu hỏi thường gặp
Điều gì khiến PPO ổn định hơn so với phương pháp gradient chính sách thông thường?
Tỷ lệ xác suất bị cắt bớt trong mục tiêu của PPO ngăn chính sách thay đổi quá nhiều trong một lần cập nhật. Các thuật toán gradient chính sách thông thường không có cơ chế bảo vệ như vậy, do đó một tập dữ liệu kinh nghiệm có độ biến thiên cao có thể đẩy các tham số vào vùng mà chính sách sụp đổ. Về cơ bản, PPO đánh đổi một lượng nhỏ độ lệch để giảm đáng kể độ biến thiên.
PPO thuộc diện bảo hiểm hay không thuộc diện bảo hiểm?
Về mặt kỹ thuật, PPO là một thuật toán dựa trên chính sách vì nó sử dụng dữ liệu từ chính sách hiện tại để cập nhật. Tuy nhiên, nó tái sử dụng mỗi lần triển khai cho nhiều chu kỳ cập nhật theo lô nhỏ, điều này mang lại cho nó một số lợi ích về hiệu quả lấy mẫu của các phương pháp ngoài chính sách mà không cần đến sự phức tạp của bộ đệm phát lại kinh nghiệm.
Tại sao các thuật toán tối ưu hóa chính sách thông thường lại có độ biến thiên cao?
Lợi nhuận từ phương pháp Monte Carlo thu được từ các tập đầy đủ có thể thay đổi rất nhiều tùy thuộc vào quỹ đạo nào được lấy mẫu. Nếu không có đường cơ sở hoặc bộ ước lượng lợi thế, ước tính độ dốc về cơ bản là tổng của phần thưởng nhân với các chỉ số hành động, điều này có độ biến thiên cao, đặc biệt là trong môi trường có thời gian dài hoặc phần thưởng ít ỏi.
Liệu PPO có thể kết hợp với các thủ thuật ổn định khác như cắt xén gradient không?
Đúng vậy, và điều đó thường xảy ra. Nhiều chuyên gia áp dụng kỹ thuật cắt độ dốc (gradient clipping) lên trên kỹ thuật cắt mục tiêu (objective clipping) của PPO, sử dụng ước lượng lợi thế tổng quát (Generalized Advantage Estimation) để giảm phương sai và chuẩn hóa lợi thế giữa các lô nhỏ (minibatch). Những bổ sung này hỗ trợ chứ không thay thế cơ chế cắt cốt lõi của PPO.
Tỷ lệ cắt điển hình được sử dụng trong PPO là bao nhiêu?
Tỷ lệ cắt mặc định là 0,2, có nghĩa là tỷ lệ xác suất bị giới hạn trong khoảng từ 0,8 đến 1,2. Các giá trị từ 0,1 đến 0,3 thường hoạt động tốt trên nhiều nhiệm vụ khác nhau, mặc dù một số môi trường sẽ được hưởng lợi từ việc cắt chặt hơn hoặc lỏng hơn tùy thuộc vào cấu trúc phần thưởng.
Phương pháp PPO có hiệu quả với không gian hành động rời rạc và liên tục không?
PPO xử lý cả hai loại không gian hành động một cách tự nhiên. Đối với các hành động rời rạc, chính sách sẽ xuất ra một phân phối theo danh mục. Đối với các hành động liên tục, nó thường xuất ra một phân phối Gaussian với giá trị trung bình được học và phương sai cố định hoặc được học. Cơ chế cắt xén hoạt động trên tỷ lệ xác suất bất kể không gian hành động nào.
PPO và TRPO có những điểm gì khác biệt?
Về bản chất, PPO là một phép xấp xỉ bậc nhất của TRPO, nhưng dễ thực hiện hơn nhiều. TRPO sử dụng ràng buộc phân kỳ KL được giải quyết bằng phương pháp gradient liên hợp và tìm kiếm đường thẳng, trong khi PPO thay thế tất cả những điều đó bằng một thao tác cắt duy nhất. PPO nhanh hơn trong mỗi lần lặp và dễ điều chỉnh hơn, mặc dù TRPO cung cấp các đảm bảo lý thuyết mạnh mẽ hơn một chút.
Tại sao PPO được sử dụng cho RLHF trong quá trình huấn luyện mô hình ngôn ngữ?
Tính ổn định và khả năng xử lý các mô hình lớn trên phần cứng phân tán của PPO đã khiến nó trở thành lựa chọn tự nhiên khi OpenAI cần tinh chỉnh các mô hình GPT với dữ liệu sở thích của con người. Mục tiêu được cắt bớt ngăn chặn chính sách đi quá xa so với mô hình được tinh chỉnh có giám sát, giúp duy trì tính trôi chảy trong khi vẫn kết hợp các tín hiệu phần thưởng.
Liệu thuật toán Policy Gradients thông thường vẫn có thể vượt trội hơn Policy Poisson trong mọi trường hợp?
Trong các thiết lập nghiên cứu hẹp với các siêu tham số được tinh chỉnh cẩn thận và môi trường đơn giản, thuật toán gradient chính sách thông thường có thể đạt được hiệu suất cuối cùng tương đương với PPO. Tuy nhiên, chúng thường đòi hỏi nhiều nỗ lực tinh chỉnh hơn và tạo ra kết quả kém nhất quán hơn trên các hạt giống ngẫu nhiên. PPO vượt trội về tính ổn định, không nhất thiết là về hiệu suất tiệm cận.
Mức cơ sở đóng vai trò gì trong các phương pháp phân cấp chính sách?
Phương pháp tính toán đường cơ sở trừ đi một giá trị ước tính từ lợi nhuận trước khi tính toán độ dốc, giúp giảm phương sai mà không gây ra sai lệch. Các lựa chọn phổ biến bao gồm hàm giá trị được học bởi bộ đánh giá, trung bình động của lợi nhuận, hoặc đơn giản là phần thưởng trung bình trong lô. PPO thường sử dụng hàm giá trị đã học làm đường cơ sở.
Phán quyết
Hãy chọn PPO khi bạn cần một thuật toán học tăng cường đáng tin cậy, đa năng, hoạt động trên nhiều môi trường khác nhau mà không cần tinh chỉnh nhiều. Sử dụng các phương pháp gradient chính sách thông thường chủ yếu cho mục đích giáo dục, phân tích lý thuyết, hoặc khi bạn muốn nghiên cứu cụ thể các chế độ lỗi mà PPO được thiết kế để loại bỏ.