AI-based workflow optimization