Optimize normals prediction