Back to feed
arXiv cs.LG
arXiv cs.LG
5/13/2026
Vertex-Softmax: Tight Transformer Verification via Exact Softmax Optimization

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?

Explore more