Technology

Yes. But this would probably cause friction with the overall public, as the AI would then give a full range of human traits, but people would still expect very narrow default outputs. And thinking more about it, what is the full range of human traits anyways? Does such a thing exist? Can we access it? Like, if we only looked at the societies the AI is present in, we still don't get all the people to actually be documented for AI to be trained upon. That's partially the cause for the racist bias of AI in the first place, isn't it? Because white cishet ablebodied people are proportionally much more frequently depicted in media.

If you gave the AI a prompt, e.g. "a man with a hat". What would you expect a good AI to produce? You have a myriad of choices to make and a machine, i.e. the AI, will not be able to make all these choices by itself. Will the result be a black person? Visibly queer or trans? In a wheelchair?

I guess the problem really is, there is no default output for anything. But when people draw something then they so have a default option ready in their mind because of societal biases and personal experiences. So I would probably draw a white cishet man with a boring hat if I were to execute that prompt. Because I'm unfortunately heavily biased, like we all are. And an AI, based on our biases, would draw the same.

But repeating the question from before, what would we expect a "fair" and "neutral" AI to draw? This is really tricky. In the meantime your solution is probably good, i.e. training the AI with more diverse data.

(Oh and I ignored the whole celebrity or known people thingy, your solution is definitely the way to go.)