Explain My Surprise: Learning Efficient Long-Term Memory by predicting uncertain outcomes