Temporary solution for it.
As most of the user are facing it after 8.1 update so its basically a software issue I guess.
Install system tiles and pin lock phone to your start screen it will help you to lock your device without using lock button.
And enable double tap to wake to unlock your phone(sorry icon user it cant be of help to you)