AI RESEARCH
ASAP: Amortized Doubly-Stochastic Attention via Sliced Dual Projection
arXiv CS.LG
•
ArXi:2605.12879v1 Announce Type: new Doubly-stochastic attention has emerged as a transport-based alternative to row-softmax attention, with recent Transformer variants using it to reduce attention sinks and rank collapse while improving performance. In this family, the standard approach is Sinkhorn scaling, which trains efficiently but still repeats matrix scaling in every inference forward pass. Sliced-transport attention removes the online iteration, but its soft sorting approximation materializes dense tensors for each slice, requiring substantially.