AI-assisted optimization