Learning what to remember