Robust Reinforcement Learning with Diffusion Wavelets