Một dự án mã nguồn mở mới có tên TinyWorldFormer vừa cho thấy AI có thể "mơ" ra cả một trò chơi ngay trên một vi điều khiển giá chỉ vài đô la. Điều đặc biệt: không có game thật nào chạy trên chip, mọi khung hình đều do mạng neural sinh ra.

Hai mạng neural nhỏ, một giấc mơ chung

Dự án dùng hai mạng nhỏ chạy song song trên Raspberry Pi Pico 2 W. Mạng thứ nhất là "mô hình thế giới": nhận khung hình hiện tại cùng hành động, rồi dự đoán khung hình kế tiếp và phần thưởng. Mạng thứ hai là "người chơi": nhìn khung hình vừa sinh ra và chọn hành động. Hai mạng nuôi nhau liên tục, và người dùng chỉ cần bấm nút để tự tay chơi trong giấc mơ đó.

Mẹo tiết kiệm bộ nhớ

Tác giả dùng looped transformer: một khối transformer duy nhất được lặp lại nhiều lần với cùng trọng số, thay vì xếp chồng nhiều lớp khác nhau. Cách này tiết kiệm bộ nhớ đáng kể, giúp mô hình đủ nhỏ để nằm gọn trong chip. Các trò như Pac-Man, Breakout, Pong, Flappy Bird và Doom-lite đều được thử nghiệm, kể cả một agent học tăng cường có thể tự học ngay trên board.

Ý nghĩa với người dùng

Đây là bước tiến thú vị cho AI chạy trên thiết bị nhỏ, không cần đám mây hay GPU đắt tiền. Nếu hướng đi này phát triển, các thiết bị gia dụng và đồ chơi thông minh có thể tự mô phỏng thế giới xung quanh một cách tiết kiệm điện. Xem thêm tin công nghệ mới nhất trên VnRead.

#Học Tăng Cường

TinyWorldFormer: mô hình thế giới AI chạy game ngay trên Raspberry Pi Pico 2 W