Prediction based pump-off detection