I have never understood why you would take a ball out of a bearing. A full set of balls spread the load just that much more.
Since the headset puts extreme pressure on the bearings, I use the heaviest auto bearing grease I can find. It also serves to keep loose balls in place when you reassemble the head set.