SSA: Improving Performance With a Better Scoring Function

Omar Naim, Swarnadeep Bhar, Jerome Bolte, Nicholas Asher


Abstract
While transformer models exhibit strong in-context learning (ICL) abilities, they often fail to generalize under simple distribution shifts. We analyze these failures and identify Softmax, the scoring function in the attention mechanism, as a contributing factor. We propose Scaled Signed Averaging (SSA), a novel attention scoring function that mitigates these failures. SSA significantly improves performance on our ICL tasks and outperforms transformer models with Softmax on several NLP benchmarks and linguistic probing tasks, in both decoder-only and encoder-only architectures.
Anthology ID:
2026.acl-long.1385
Volume:
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
30033–30051
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.acl-long.1385/
DOI:
Bibkey:
Cite (ACL):
Omar Naim, Swarnadeep Bhar, Jerome Bolte, and Nicholas Asher. 2026. SSA: Improving Performance With a Better Scoring Function. In Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pages 30033–30051, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
SSA: Improving Performance With a Better Scoring Function (Naim et al., ACL 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.acl-long.1385.pdf
Checklist:
 2026.acl-long.1385.checklist.pdf