Screen keeps turning on
After having switched off the Ambient display the problem has stopped for me. Hope this helps.

Settings > Display > Ambient display (Wake screen when you pick up device or receive notifications)