AI outbreak prediction