University of California, Riverside; University of Michigan, UC Riverside; University of Pennsylvania
Davoud Ataee Tarzanagh, Yingcong Li, Xuechen Zhang, Samet Oymak
attention mechanism;implicit bias;margin maximization;nonconvex optimization;prompt tuning
Max-Margin Token Selection in Attention Mechanism