State space collapse for critical multistage epidemics