Hay una opción que hace que el WiFi se apague cuando el teléfono entra en suspensión (esta en ajustes-wifi-opciones avanzadas).
Si no tienes datos activos (de fondo) la WiFi permanecerá apagada hasta que enciendas la pantalla, momento en el que reconectará y tras un breve lapso te llegarían las notificaciones