AI-based control