Effects of non-equilibrium segregation on near-surface diffusion