Correlation modelling on the sphere using a generalized diffusion equation