Is Treason ever okay?

If i were to leave the USA to another country that the USA was at war with and i joined the military there and just started killing everyone there and i was able to escape back to the USA, would they take me in or arrest me or kill me or what?