Si fuese un problema de hardware (el cable Flex) no debería fallar siempre y sin excepción? En este caso sólo falla cuando bloquea la pantalla.
Prueba a flashear la última ROM oficial con xiaomimitool a ver si se soluciona
https://www.xiaomitool.com/V2/latest