𝗞-𝗺𝗲𝗮𝗻𝘀 𝗶𝘀 𝘀𝗶𝗺𝗽𝗹𝗲. 𝗠𝗮𝗸𝗶𝗻𝗴 𝗶𝘁 𝗳𝗮𝘀𝘁 𝗼𝗻 𝗚𝗣𝗨𝘀 𝗶𝘀𝗻’𝘁. Đó là lý do tại sao chúng tôi xây dựng Flash-KMeans — một triển khai k-means chính xác có nhận thức về IO, suy nghĩ lại thuật toán xung quanh các nút thắt GPU hiện đại. Bằng cách tấn công trực tiếp vào các nút thắt bộ nhớ, Flash-KMeans đạt được tốc độ nhanh gấp 30 lần so với cuML và 200 lần so với FAISS — với cùng một thuật toán chính xác, chỉ được thiết kế cho phần cứng ngày nay. Ở quy mô triệu, Flash-KMeans có thể hoàn thành một vòng lặp k-means trong vài mili giây. Một thuật toán cổ điển — được thiết kế lại cho các GPU hiện đại. Tài liệu: Mã: