soft max MAB