AI-driven Threat Detection