about
Laser: Attention with Exponential Transformation (arxiv.org)
13 points by PaulHoule on Nov 28, 2024 | hide | past | pdf | discuss on HN

In plain words: Attention normally uses a softmax step to weight past words, but its gradients can be tiny, slowing learning in earlier layers. Replacing it with an exponential weighting sends stronger gradients and raised downstream scores by up to 1.44% over standard attention in language models.

Abstract · LASER: Attention with Exponential Transformation

Transformers have had tremendous impact for several sequence related tasks, largely due to their ability to retrieve from any part of the sequence via softmax based dot-product attention. This mechanism plays a crucial role in Transformer's performance. We analyze the gradients backpropagated through the softmax operation in the attention mechanism and observe that these gradients can often be small. This poor gradient signal backpropagation can lead to inefficient learning of parameters preceeding the attention operations. To this end, we introduce a new attention mechanism called LASER, which we analytically show to admit a larger gradient signal. We show that LASER attention can be implemented by making small modifications to existing attention implementations. We conduct experiments on autoregressive large language models (LLMs) with upto 7.7 billion parameters with an average improvement of upto 1.44% over standard attention on downstream evaluations and 1.65% finetuning improvements. Additionally, LASER demonstrates generalization performance improvement across a variety of tasks (vision, text and speech):Vision Transformer (ViT) on Imagenet, Conformer on the Librispeech speech-to-text and BERT with 2.2 billion parameters.

Sai Surya Duvvuri, Inderjit S. Dhillon
arXiv:2411.03493 · cs.LG, cs.CL · submitted Nov 5, 2024 · updated Jul 13, 2025
abstract · pdf · html · ICML 2025

add comment on HN