hay una parte en los ajustes de la wifi en donde se escoge el compartamiento de la wifi cuando se apaga la pantalla, si sigue conectado o se desconecta la wifi. Revisa eso debe presionar el boton menu y luego avanzado en donde configuras las redes wifi
si no es eso entonces hay que hilar mas fino, yo por ejemplo tenia el mismo problema y cambiandole un parametro al router mejoro sustancialmente la estabilidad a la hora de conectarme a la wifi