In Airbus a/c (only fly 320/321 but I guess all the FBW ones are the same) you retract the flap by one stage as soon as you go around and the rest in stages as speed builds once at accel. height, which is normally 1500ft agl. Seems to work very well, whatever config you have used for the approach you always put the flaps up one stage for the Go-around, even my small brain can cope with that.