Predicting excess returns under heterogeneous trading and learning: A diffusive approach