AI-driven optimization