The Chern-Simons diffusion rate in improved holographic QCD