Yeah sounds like a battery problem.  I watched movies and played games on a 5 hour flight and still had battery left.  Try the battery pull and see if that helps.  I assume the display is shutting down when not in use?