A surprisal oracle for when every layer counts