Try soaking your pellet feed , or use sinking type. Sometimes f/g/f gulp air when feeding . Also dry feed can swell in stomach causing pressure against the swim bladder . This then upsets the balance
However if the fish is permanently having balance problems then its probably congenital and wont improve.

Forrest