AI-driven defense