AI-based mechanism