>a society essentially controlled by the AI, as it will be above us.

This is the highest probability of the realistic outcome of AGI as it will advance in capabilities so much and so rapidly that we won't be able to keep up. I personally think it will be benevolent and care for us like we do for our pets which won't necessarily be a bad thing.