Tiếp theoHướng dẫn tiếp theo
Trình tối ưu hóa Zero và Sharded
kỹ thuật
HƯỚNG DẪN KỸ THUẬT
Adam là trình tối ưu hóa đặc biệt đằng sau hầu hết các mạng thần kinh hiện đại, tự động điều chỉnh tốc độ học tập riêng biệt cho mọi tham số.
Nghe hướng dẫn nàyAI thuật lại · ~2 min
Điều này quan trọng vì nó giúp huấn luyện các mô hình sâu nhanh hơn và ít khó chịu hơn nhiều so với việc hạ độ dốc thông thường.
Adam (Ước tính thời điểm thích ứng), được Kingma và Ba giới thiệu vào năm 2014, kết hợp hai ý tưởng. Đầu tiên, động lượng: nó giữ mức trung bình giảm dần theo cấp số nhân của độ dốc trong quá khứ (thời điểm đầu tiên) để cập nhật tốc độ xây dựng theo các hướng nhất quán. Thứ hai, chia tỷ lệ theo từng tham số: nó theo dõi mức trung bình của các gradient bình phương (thời điểm thứ hai) và chia mỗi bước cho căn bậc hai của giá trị đó, do đó, các tham số có độ dốc lớn, nhiễu sẽ thực hiện các bước nhỏ hơn và các tham số hiếm khi được cập nhật sẽ thực hiện các bước lớn hơn. Khả năng thích ứng này có nghĩa là bạn có thể thường xuyên sử dụng một tốc độ học tập trên toàn bộ mạng. Một biến thể, AdamW, tách riêng sự suy giảm trọng lượng khỏi bản cập nhật độ dốc và đã trở thành mặc định để đào tạo các mô hình ngôn ngữ và máy biến áp lớn.
Đào tạo các mô hình ngôn ngữ lớn như GPT và Llama, sử dụng AdamW làm trình tối ưu hóa tiêu chuẩn.
Tinh chỉnh trình phân loại hình ảnh được huấn luyện trước (ví dụ: ResNet) trên tập dữ liệu tùy chỉnh chỉ với tốc độ học Adam mặc định.
Đào tạo các mô hình khuếch tán đằng sau các trình tạo hình ảnh như Khuếch tán ổn định.
Chạy Adam 8 bit trong các thư viện như bitandbyte để phù hợp với trạng thái tối ưu hóa trong bộ nhớ GPU hạn chế.
Adam và AdamW vẫn chiếm ưu thế, nhưng nghiên cứu đang nâng cao hiệu quả cho các mô hình nghìn tỷ tham số, trong đó việc lưu trữ thêm hai giá trị cho mỗi trọng lượng là rất tốn kém. Các biến thể ánh sáng bộ nhớ như Adafactor, Adam 8 bit và các trình tối ưu hóa mới hơn như Lion (chỉ sử dụng động lượng dựa trên dấu hiệu) và Sophia nhằm mục đích phù hợp với chất lượng của Adam với ít bộ nhớ hơn hoặc hội tụ nhanh hơn. Mong đợi các trình tối ưu hóa thích ứng được điều chỉnh riêng cho hoạt động đào tạo phân tán, có độ chính xác thấp để tiếp tục phát triển.
Bản tin miễn phí
Ba câu chuyện AI đã được xác minh vào mỗi buổi sáng các ngày trong tuần, được viết bằng tiếng Anh đơn giản. Miễn phí mãi mãi, không có quảng cáo.
Một email mỗi ngày trong tuần. Hủy đăng ký trong một cú nhấp chuột. Chúng tôi không bao giờ bán hoặc chia sẻ địa chỉ của bạn.
Tự kiểm tra
Phản hồi tức thì về mọi câu trả lời và chứng chỉ có thể chia sẻ với ID có thể xác minh sau khi bạn vượt qua khóa học.
Hỗ trợ giáo dục AI miễn phí. AI Understanding là tổ chức phi lợi nhuận 501(c)(3) — không bao giờ có quảng cáo, không có tường phí. Hãy quyên góp
Adam là trình tối ưu hóa đặc biệt đằng sau hầu hết các mạng thần kinh hiện đại, tự động điều chỉnh tốc độ học tập riêng biệt cho mọi tham số. Điều này quan trọng vì nó giúp việc đào tạo các mô hình sâu nhanh hơn và ít phức tạp hơn so với việc giảm độ dốc đơn giản.
Adam giữ mức trung bình giảm dần theo cấp số nhân của độ dốc (thời điểm đầu tiên) và độ dốc bình phương (thời điểm thứ hai) cho mọi tham số.
Cả m và v đều được khởi tạo bằng 0, do đó các ước tính ban đầu có độ lệch thấp; chia cho (1 - beta^t) sẽ sửa lỗi này.
AdamW áp dụng sự suy giảm trọng số trực tiếp cho các trọng số thay vì trộn nó vào thuật ngữ gradient thích ứng, giúp cải thiện khả năng khái quát hóa trong máy biến áp.
Epsilon (khoảng 1e-8) được thêm vào mẫu số để các tham số có độ dốc trung bình bình phương gần bằng 0 không gây ra vụ nổ.
Bằng cách chia cho căn bậc hai của mức trung bình gradient bình phương của mỗi tham số, Adam chia tỷ lệ kích thước bước cho mỗi tham số thay vì sử dụng một giá trị chung.
Tiếp tục học hỏi
Đã chọn thêm hướng dẫn cho chủ đề này
Tiếp theoHướng dẫn tiếp theo
Trình tối ưu hóa Zero và Sharded
kỹ thuật