Tổng quan
Chú ý nhiều truy vấn (MQA) là một giải pháp tiết kiệm bộ nhớ cho sự chú ý của máy biến áp, chia sẻ một bộ khóa và giá trị trên tất cả các đầu chú ý. Nó tăng tốc đáng kể việc tạo văn bản bằng cách thu nhỏ bộ nhớ mà mô hình phải xáo trộn xung quanh.
Chú ý nhiều truy vấn là một phần của ngăn xếp ngôn ngữ-AI dùng để đọc, tạo, phân loại và chuyển đổi văn bản cũng như lời nói trên quy mô lớn.
Lặn sâu
Sự chú ý nhiều đầu tiêu chuẩn cung cấp cho mỗi đầu các dự đoán truy vấn, khóa và giá trị riêng. Trong quá trình tạo, các khóa và giá trị cho tất cả các mã thông báo trước đây phải được lưu vào bộ đệm và tải lại ở mỗi bước — bộ đệm KV này trở thành nút thắt cổ chai chính vì việc đọc nó từ bộ nhớ chậm hơn so với chính phép toán. Chú ý nhiều truy vấn, do Noam Shazeer đề xuất vào năm 2019, giữ các dự đoán truy vấn riêng biệt cho mỗi đầu nhưng thu gọn các khóa và giá trị vào một đầu chung duy nhất. Điều này sẽ thu nhỏ bộ đệm KV theo hệ số bằng số lượng đầu, đôi khi nhỏ hơn từ 8 đến 64 lần. Kết quả là khả năng giải mã tự hồi quy nhanh hơn nhiều và dung lượng bộ nhớ nhẹ hơn mà chất lượng chỉ bị giảm ở mức khiêm tốn. Nền tảng trung gian, Chú ý truy vấn theo nhóm, cân bằng sự đánh đổi.
Hiểu biết kỹ thuật
Trong MQA, trọng số truy vấn vẫn tạo ra H vectơ truy vấn riêng biệt, nhưng một phép chiếu khóa duy nhất và phép chiếu giá trị duy nhất được chia sẻ trên tất cả các đầu. Mỗi đầu tính toán sự chú ý bằng cách sử dụng truy vấn riêng của nó dựa trên các khóa và giá trị giống nhau. Bởi vì các tenxơ K và V được lưu trong bộ nhớ đệm không còn tỷ lệ theo số lượng đầu, băng thông bộ nhớ trong quá trình giải mã giảm mạnh — và băng thông, chứ không phải tính toán, mới là tốc độ tạo cổng trên các máy gia tốc hiện đại.
Nắm vững sự chú ý nhiều truy vấn
Chú ý nhiều truy vấn (MQA) là một giải pháp tiết kiệm bộ nhớ cho sự chú ý của máy biến áp, chia sẻ một bộ khóa và giá trị trên tất cả các đầu chú ý. Nó tăng tốc đáng kể việc tạo văn bản bằng cách thu nhỏ bộ nhớ mà mô hình phải xáo trộn xung quanh. Chú ý nhiều truy vấn là một phần của ngăn xếp ngôn ngữ-AI dùng để đọc, tạo, phân loại và chuyển đổi văn bản cũng như lời nói trên quy mô lớn. Để xây dựng sự hiểu biết sâu sắc, hãy coi Chú ý nhiều truy vấn như một mô hình vận hành chứ không phải một tính năng duy nhất: xác định kết quả mong muốn, làm rõ các giả định và tách biệt những gì hệ thống có thể thực hiện một cách đáng tin cậy với những gì vẫn cần đến sự đánh giá của chuyên gia.
Trong thực tế, các nhóm mạnh sử dụng các lời nhắc, truy xuất và vòng lặp xem xét Chú ý nhiều truy vấn được thiết kế như một hệ thống liên lạc tích hợp. Họ ghi lại các tiêu chí thành công rõ ràng, kiểm tra dựa trên dữ liệu và quy trình làm việc thực tế, đồng thời lặp lại dựa trên các kiểu thất bại được quan sát thay vì chiến thắng điểm chuẩn một lần. Đây là nơi sự hiểu biết về mặt lý thuyết biến thành khả năng bền vững trên toàn bộ sản phẩm, chính sách và hoạt động.
Quy trình công việc ngôn ngữ có thể di chuyển nhanh hơn mà không làm mất tính nhất quán. Đồng thời, các sự kiện ảo giác có thể âm thầm đưa vào các báo cáo, luồng hỗ trợ hoặc kết quả nghiên cứu. Cách tiếp cận linh hoạt nhất là kết hợp tốc độ thử nghiệm với kỷ luật quản trị: chạy thử nghiệm, thu thập bằng chứng, xuất bản nhật ký quyết định và liên tục cập nhật các biện pháp bảo vệ khi hành vi của mô hình, kỳ vọng của người dùng và các yêu cầu pháp lý phát triển.
Tác động chiến lược
Quy trình công việc ngôn ngữ có thể di chuyển nhanh hơn mà không làm mất tính nhất quán.
Quy trình công việc ngôn ngữ có thể di chuyển nhanh hơn mà không làm mất tính nhất quán. Trong quá trình triển khai chất lượng cao, điều này được chuyển thành các quy tắc vận hành, ranh giới quyền sở hữu và quy trình đánh giá định kỳ có thể đo lường được để các nhóm có thể mở rộng quy mô một cách tự tin thay vì mở rộng quy mô sự mơ hồ.
Nó mở rộng quyền truy cập vào các ngôn ngữ và phong cách giao tiếp.
Nó mở rộng quyền truy cập vào các ngôn ngữ và phong cách giao tiếp. Trong quá trình triển khai chất lượng cao, điều này được chuyển thành các quy tắc vận hành, ranh giới quyền sở hữu và quy trình đánh giá định kỳ có thể đo lường được để các nhóm có thể mở rộng quy mô một cách tự tin thay vì mở rộng quy mô sự mơ hồ.
Các nhóm có thể dành nhiều thời gian hơn để đánh giá trong khi quá trình tự động hóa xử lý sự lặp lại.
Các nhóm có thể dành nhiều thời gian hơn để đánh giá trong khi quá trình tự động hóa xử lý sự lặp lại. Trong quá trình triển khai chất lượng cao, điều này được chuyển thành các quy tắc vận hành, ranh giới quyền sở hữu và quy trình đánh giá định kỳ có thể đo lường được để các nhóm có thể mở rộng quy mô một cách tự tin thay vì mở rộng quy mô sự mơ hồ.
Triển khai trong thế giới thực
Tăng tốc quá trình tạo mã thông báo theo mã thông báo trong trợ lý trò chuyện trong đó bộ nhớ đệm KV chứ không phải điện toán thô sẽ hạn chế thông lượng.
PaLM của Google, sử dụng Chú ý nhiều truy vấn để cho phép suy luận quy mô lớn hiệu quả.
Phục vụ nhiều người dùng đồng thời trên một GPU bằng cách thu nhỏ bộ nhớ đệm KV theo yêu cầu.
Chú ý truy vấn được nhóm trong Llama 2 70B và Llama 3, một thế hệ con cháu trực tiếp cân bằng tốc độ của MQA với chất lượng chú ý đầy đủ.
Các mẫu triển khai
Chú ý nhiều truy vấn trong thực tế
Tăng tốc quá trình tạo mã thông báo theo mã thông báo trong trợ lý trò chuyện trong đó bộ nhớ đệm KV chứ không phải điện toán thô sẽ hạn chế thông lượng.
Tăng tốc quá trình tạo mã thông báo theo mã thông báo trong trợ lý trò chuyện trong đó bộ nhớ đệm KV chứ không phải tính toán thô giới hạn thông lượng Các nhóm thường đạt được kết quả tốt hơn khi họ xác định trước các ngưỡng chất lượng, duy trì lộ trình leo thang của con người đối với các trường hợp khó khăn và theo dõi cả mức tăng năng suất và chi phí lỗi theo thời gian.
Chú ý nhiều truy vấn trong thực tế
PaLM của Google, sử dụng Chú ý nhiều truy vấn để cho phép suy luận quy mô lớn hiệu quả.
PaLM của Google, sử dụng Chú ý nhiều truy vấn để cho phép suy luận hiệu quả trên quy mô lớn. Các nhóm thường đạt được kết quả tốt hơn khi họ xác định trước các ngưỡng chất lượng, duy trì lộ trình leo thang của con người đối với các trường hợp phức tạp và theo dõi cả mức tăng năng suất và chi phí lỗi theo thời gian.
Chú ý nhiều truy vấn trong thực tế
Phục vụ nhiều người dùng đồng thời trên một GPU bằng cách thu nhỏ bộ nhớ đệm KV theo yêu cầu.
Phục vụ nhiều người dùng đồng thời trên một GPU bằng cách thu nhỏ bộ nhớ đệm KV theo yêu cầu Các nhóm thường đạt được kết quả tốt hơn khi họ xác định trước các ngưỡng chất lượng, duy trì lộ trình leo thang của con người đối với các trường hợp khó khăn và theo dõi cả mức tăng năng suất và chi phí lỗi theo thời gian.
Chú ý nhiều truy vấn trong thực tế
Chú ý truy vấn được nhóm trong Llama 2 70B và Llama 3, một thế hệ con cháu trực tiếp cân bằng tốc độ của MQA với chất lượng chú ý đầy đủ.
Chú ý truy vấn theo nhóm trong Llama 2 70B và Llama 3, một thế hệ kế thừa trực tiếp cân bằng tốc độ của MQA với chất lượng được chú ý đầy đủ Các nhóm thường nhận được kết quả tốt hơn khi họ xác định trước các ngưỡng chất lượng, duy trì lộ trình leo thang của con người đối với các trường hợp phức tạp và theo dõi cả mức tăng năng suất và chi phí lỗi theo thời gian.
Rủi ro & lan can
Sự thật ảo giác có thể lặng lẽ đi vào báo cáo, luồng hỗ trợ hoặc kết quả nghiên cứu.
Sự nhạy cảm kịp thời có thể tạo ra kết quả không nhất quán đối với các yêu cầu tương tự.
Dữ liệu văn bản nhạy cảm có thể bị lộ nếu khả năng kiểm soát quyền truy cập yếu.
Lộ trình thực hiện
Xác định định dạng đầu ra, âm thanh và tiêu chuẩn chất lượng trước khi triển khai.
Xác định định dạng đầu ra, âm thanh và tiêu chuẩn chất lượng trước khi triển khai. Hãy coi mỗi bước như một cổng bằng chứng: nếu không đáp ứng được các tiêu chí, hãy tạm dừng triển khai, thu hẹp khoảng cách và chỉ sau đó mới mở rộng mức sử dụng.
Phản hồi mặt đất với các nguồn đáng tin cậy bất cứ khi nào độ chính xác quan trọng.
Phản hồi mặt đất với các nguồn đáng tin cậy bất cứ khi nào độ chính xác quan trọng. Hãy coi mỗi bước như một cổng bằng chứng: nếu không đáp ứng được các tiêu chí, hãy tạm dừng triển khai, thu hẹp khoảng cách và chỉ sau đó mới mở rộng mức sử dụng.
Duy trì điểm kiểm tra đánh giá của con người đối với các kết quả đầu ra có mức độ rủi ro cao.
Duy trì điểm kiểm tra đánh giá của con người đối với các kết quả đầu ra có mức độ rủi ro cao. Hãy coi mỗi bước như một cổng bằng chứng: nếu không đáp ứng được các tiêu chí, hãy tạm dừng triển khai, thu hẹp khoảng cách và chỉ sau đó mới mở rộng mức sử dụng.
Theo dõi các kiểu lỗi và đào tạo lại các lời nhắc hoặc quy trình làm việc thường xuyên.
Theo dõi các kiểu lỗi và đào tạo lại các lời nhắc hoặc quy trình làm việc thường xuyên. Hãy coi mỗi bước như một cổng bằng chứng: nếu không đáp ứng được các tiêu chí, hãy tạm dừng triển khai, thu hẹp khoảng cách và chỉ sau đó mới mở rộng mức sử dụng.