improved entity extrapolation by sampling several steps