A two-branch trade-off neural network for balanced scoring sleep stages on multiple cohorts