Mira la configuración avanzada del wifi por defecto está configurada para que se desconecte al apagar la pantalla si este es tu problema simplemente lo cambias a las otras opciones y ya está ,lo llevan todas las roms.
Si lo tienes puesto para que no se desconecte y lo hace comprueba tu router.