arXiv cs.LG
5/13/2026

Vertex-Softmax: Tight Transformer Verification via Exact Softmax Optimization
Short summary
Vertex-Softmax provides a new algorithm for certified verification of transformer attention with significantly tighter bounds than existing approaches. It proves optimal softmax bounds occur at constraint box vertices, enabling log-linear complexity search. Evaluated on MNIST, Fashion-MNIST, and CIFAR-10, it outperforms alpha-CROWN and branch-and-bound baselines while maintaining formal soundness guarantees.
- •New Vertex-Softmax algorithm for tight transformer attention verification
- •Reduces search complexity to log-linear by proving optimum occurs at box vertices
- •Outperforms alpha-CROWN and branch-and-bound on image classification benchmarks
Generated with AI, which can make mistakes.
Is this a good recommendation for you?