Tiếp theoHướng dẫn tiếp theo
Mô hình không gian trạng thái và Mamba
Nguyên tắc cơ bản
HƯỚNG DẪN AI về ngôn ngữ
Mamba là mô hình trình tự được xây dựng trên các mô hình không gian trạng thái (SSM) xử lý văn bản theo thời gian tuyến tính, cung cấp giải pháp thay thế nhanh chóng cho sự chú ý bậc hai của Transformer.
Nghe hướng dẫn nàyAI thuật lại · ~2 min
Its key trick is making the model selectively decide what to remember and forget based on the input itself.
Mamba được Albert Gu và Trí Đạo giới thiệu vào cuối năm 2023, được xây dựng trên các mô hình không gian trạng thái có cấu trúc. SSM cổ điển nén toàn bộ lịch sử của chuỗi vào trạng thái ẩn có kích thước cố định và cập nhật nó từng bước, giống như một mạng lặp lại phức tạp. Điểm đột phá là tính chọn lọc: Mamba làm cho các tham số của SSM (giữ bao nhiêu, cho vào bao nhiêu) phụ thuộc vào mã thông báo hiện tại, do đó mô hình có thể tập trung vào các từ có liên quan và bỏ qua phần bổ sung. Điều này cho phép một trạng thái có kích thước cố định hoạt động giống như bộ nhớ nhận biết nội dung. Vì tránh so sánh mọi mã thông báo với mọi mã thông báo khác nên Mamba chia tỷ lệ tuyến tính theo độ dài chuỗi và duy trì nhanh chóng trên các đầu vào rất dài như bộ gen, âm thanh hoặc văn bản có độ dài sách.
Mô hình hóa các chuỗi DNA cực dài trong đó Máy biến áp hàng triệu mã thông báo quá đắt
Hỗ trợ trợ lý ngôn ngữ ngữ cảnh dài tóm tắt toàn bộ sách mà không cần cắt bớt
Tạo mô hình giọng nói và tạo âm thanh theo thời gian thực để xử lý dạng sóng thô một cách hiệu quả
Triển khai trên thiết bị hoặc biên trong đó trạng thái lặp lại có kích thước cố định nhỏ giúp tiết kiệm bộ nhớ so với bộ nhớ đệm chú ý ngày càng tăng
Mamba và người kế nhiệm Mamba-2 đang đẩy mạnh các kiến trúc kết hợp xen kẽ một số lớp chú ý với nhiều lớp SSM, nắm bắt được điểm mạnh của cả hai. Yêu cầu SSM trong trợ lý ngữ cảnh dài, các mẫu trên thiết bị có bộ nhớ bị hạn chế và các miền không phải văn bản như DNA và âm thanh. Nghiên cứu đang thăm dò xem liệu SSM thuần túy có thể sánh ngang với Transformers trong các nhiệm vụ cần thu hồi chính xác hay không và liệu chúng có mở rộng đến kích thước mô hình lớn nhất hay không.
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
Mamba là mô hình trình tự được xây dựng trên các mô hình không gian trạng thái (SSM) xử lý văn bản theo thời gian tuyến tính, cung cấp giải pháp thay thế nhanh chóng cho sự chú ý bậc hai của Transformer. Bí quyết chính của nó là làm cho mô hình quyết định có chọn lọc những gì cần nhớ và quên dựa trên chính dữ liệu đầu vào.
Mamba tránh so sánh mã thông báo tất cả các cặp, do đó chi phí của nó tăng tuyến tính theo độ dài chuỗi thay vì theo phương trình bậc hai như sự chú ý.
Tính chọn lọc có nghĩa là các tham số như B, C và delta trở thành chức năng của đầu vào hiện tại, mang lại bộ nhớ nhận biết nội dung.
SSM là các mô hình kiểu lặp lại, gấp quá khứ vào trạng thái có kích thước cố định, tương tự như RNN.
Chế độ xem tích chập yêu cầu các tham số cố định (bất biến theo thời gian); các tham số phụ thuộc vào đầu vào sẽ phá vỡ điều đó nên Mamba sử dụng tính năng quét song song để thay thế.
Mamba sử dụng tính năng quét chọn lọc nhận biết phần cứng để giữ các trạng thái trung gian trong SRAM nhanh nhằm khôi phục thông lượng.
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
Mô hình không gian trạng thái và Mamba
Nguyên tắc cơ bản