MemBodied: Recurrent Associative Memory for Vision-Language-Action Models

Nghiên cứu giới thiệu MemBodied, một bộ nhớ episodic kích thước cố định gồm hai thành phần bổ trợ cho nhau: một associative state ghi lại các tương tác qua các lần gọi policy và một episode anchor lưu trữ đại diện gọn nhẹ của cảnh ban đầu. Thay vì làm phình to ngữ cảnh bằng cách giữ lại toàn bộ các quan sát trong quá khứ, mô hình điều kiện hóa việc tạo hành động dựa trên đầu vào hiện tại và các thành phần bộ nhớ này. Kết quả thực nghiệm trên các tác vụ RMBench và LIBERO-Long cho thấy MemBodied cải thiện đáng kể tỷ lệ thành công so với chính sách phi trạng thái và các bộ nhớ hồi quy thông thường.

MemBodied đạt tỷ lệ thành công trung bình gấp 7.81times so với chính sách phi trạng thái và 2.98times so với vanilla recurrent memory trên năm tác vụ RMBench.

Mô hình vượt trội hơn baseline tăng cường bộ nhớ mạnh nhất 1.3times với số lượng tham số thêm vào ít hơn 10times.

Trên bộ LIBERO-Long, mô hình đạt 90.6%, cải thiện 5.4% so với chính sách π_0 không trạng thái.