AI-driven protection