The frequent display timeout that is causing your device to lose connection when the screen is off can be rather annoying. To retain that connection whenver your screen is turned off you can use Screen Standby for that.

It will also prevent your device from overheating especially if you are connecting it via MHL/HDMI for a long gaming or movie streaming session.