(1/9)🧠World models don’t just need more memory. They need the right memory to predict what happens next.
We introduce FAR (Future-Aware Recall), a principled framework that learns episodic recall for world models:
🏆Beats hand-designed recall—even with the same cues
🎛️Automatically learns which recall cues to trust for each query, among all available cues
🔄Recalls the right state, even as the environment changes
Huge thanks to my collaborators @JCJesseLai, @nguyencongbacbk, @_amirbar, Jong Chul Ye, and @mittu1204 for making this work possible!
We introduce FAR (Future-Aware Recall), a principled framework that learns episodic recall for world models:
🏆Beats hand-designed recall—even with the same cues
🎛️Automatically learns which recall cues to trust for each query, among all available cues
🔄Recalls the right state, even as the environment changes
Huge thanks to my collaborators @JCJesseLai, @nguyencongbacbk, @_amirbar, Jong Chul Ye, and @mittu1204 for making this work possible!