Restoring Noisy Demonstration for Imitation Learning With Diffusion Models
Published in IEEE Transactions on Neural Networks and Learning Systems, 2026
Most imitation learning methods assume perfect expert demonstrations, yet real data is often noisy. We propose a filter-and-restore framework that isolates clean samples and uses conditional diffusion models to recover noisy demonstrations. Experiments on robot arm manipulation, dexterous manipulation, and locomotion show consistent improvements over existing methods, with ablations confirming robustness to diverse noise types and levels.
