AI-driven risk management