I had the same problem after going through a few heavy rain storms.. it was the control/display unit..just too much moisture has seeped in.. swapped it out for a new one at the place i bought the bike from.. works great now.
Got caught in a bad downpour last night and took shelter, also wrapped plastic bag around the control unit.. Avoid getting it really soaked.