AI model enhancement