Future state generation

A auxiliary future state generation framework using diffusion model

In this work, we propose a lightweight auxiliary framework that integrates a diffusion-based future state predictor to reassess and refine initial actions.

Project Report PDF here