Exploring the Frontiers of Softmax: Provable Optimization, Applications in Diffusion Model, and Beyond (arXiv 2024)

Jiuxiang Gu, Chenyang Li, Yingyu Liang, Zhenmei Shi, Zhao Song

January 2024