There isnt an AOSP for mi2 on the new partition layout. I find that even on the old days Ivan is the only one guy to work on AOSP for mi2.
AOSP4.4.4 is good, but record audio quality is terrible bad, and doesnt support new partition layout.
AOSP5.1.1 is good too, but I cant configure home button...