AI-driven risk assessment