Compressed imitation learning