If you assign a headset button to mic mute/unmute it works perfectly until the screen goes off. After that it doesn't work anymore
Though it does work for other apps (e.g. i can stop/play music while on lock screen), so it's most probably on the app side
it would be very useful being able to mute/unmute the mic and not having to get you phone out of the pocket for that