O meu também estava com esse problema. Ele também desconectava sozinho do wi-fi toda vez em que a tela bloqueava.
Alterei os ajustes de wi-fi e aparentemente o problema foi resolvido.
O caminho foi: Ajustes> Wi-Fi> Vá até a rede na qual quer conectar e clique no i de informações na lateral direita> Ative conectar automaticamente.
Espero que funcione. Boa sorte!