Diversity Impact on AI Model Generalization