Restoring Noisy Demonstration for Imitation Learning With Diffusion Models
IEEE Transactions on Neural Networks and Learning Systems (TNNLS) 2025
Expert demonstrations often contain imperfections caused by human errors or sensor/control inaccuracies, which most imitation learning methods cannot handle. We propose a filter-and-restore framework that first filters clean samples from noisy demonstrations and then learns conditional diffusion models to recover the noisy ones. Our framework consistently outperforms existing methods across robot arm manipulation, dexterous manipulation, and locomotion tasks, and remains robust to different noise types and levels.
[ Paper ]