AI-driven epidemiology