AI-driven prediction