Any good advice to help acheive a nice headset on a stubborn horse?

My horse has a wonderful headset at a walk, but as soon as she transitions into another gait her head shoots up. I have her in a martingale right now, but she still does it. Anybody have any suggestions on what to do?