I don't think android 5.1, 6 or 7 will made Mi Note Pro better or not.
It's a good Phone, with a great hardware, wonderful display, a great audio section, etc.
It was an unfortunate phone, perhaps because of rumors about the overheating of the CPU (I've never had overheating of my Mi Note Pro)...