atlas

Research map / Architectures and efficiency / Attention and transformers

Outlier-Efficient Hopfield Layers for Large Transformer-Based Models

ICML 2024 · Architectures and efficiency › Attention and transformers · approach: Attention head and block redesign · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes OutEffHop, an outlier-efficient modern Hopfield model whose retrieval dynamics yield a softmax_1 attention layer that reduces activation outliers (kurtosis, max inf-norm) and improves post-quantization performance across BERT, OPT, ViT, and STanHop-Net.

Problem

Transformer attention softmax forces attention onto low-information "no-op" tokens, creating activation outliers that hurt quantization and efficiency of large transformer models.

Approach

Add a no-op classification dimension and a zero-energy point to the modern Hopfield energy via a refined log-sum-exp; its single-step retrieval dynamics give the softmax_1 attention (exp(z)/(1+sum exp(z))), which lets attention place near-zero weight on all tokens and so suppresses outliers.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
replace vanilla attention with OutEffHopBERT-base-uncased · BERT validationaverage kurtosis: 418.724 $\pm$ 0.814 -> 26.564 $\pm$ 0.022Helped
replace vanilla attention with OutEffHopBERT-base-uncased · BERT validationmaximum infinity norm: 255.859 $\pm$ 0.004 -> 33.618 $\pm$ 0.000Helped
replace vanilla attention with OutEffHopBERT-base-uncased · BERT validationperplexity (W8A8): 7.154 $\pm$ 0.009 -> 6.295 $\pm$ 0.001Helped
replace vanilla attention with OutEffHopOPT-125m · OPT validationaverage kurtosis: 23341.513 $\pm$ 27.363 -> 21.542 $\pm$ 0.000Helped
replace vanilla attention with OutEffHopOPT-125m · OPT validationmaximum infinity norm: 92.786 $\pm$ 0.002 -> 13.302 $\pm$ 0.001Helped
replace vanilla attention with OutEffHopOPT-125m · OPT validationperplexity (W8A8): 42.012 $\pm$ 19.514 -> 16.429 $\pm$ 0.013Helped
replace vanilla attention with OutEffHopViT-S_16 · ImageNet-1k validationaverage kurtosis: 37.104 $\pm$ 0.000 -> 31.601 $\pm$ 0.001Helped
replace vanilla attention with OutEffHopViT-S_16 · ImageNet-1k validationmaximum infinity norm: 272.198 $\pm$ 0.000 -> 249.163 $\pm$ 0.000Helped
replace vanilla attention with OutEffHopViT-S_16 · ImageNet-1k validationtop-1 accuracy (W8A8): 74.935 $\pm$ 0.046 -> 76.313 $\pm$ 0.012Helped
replace vanilla Hopfield layer with OutEffHopSTanHop-Net · ETTh1 validationaverage kurtosis: 2.954 $\pm$ 0.063 -> 2.897 $\pm$ 0.011Mixed
replace vanilla Hopfield layer with OutEffHopSTanHop-Net · ETTh1 validationmaximum infinity norm: 5.048 $\pm$ 0.232 -> 4.565 $\pm$ 0.209Helped
combine OutEffHop with Clipped Softmax (ablation)BERT-base-uncased · BERT validationaverage kurtosis: 26.564 $\pm$ 0.022 -> 11.839 $\pm$ 0.001Helped

Surprising findings

Limitations

Experimental setup

Datasets: BookCorpus wiki40b/en ImageNet-1k ETTh1 ETTm1 WTH

Benchmarks: ImageNet-1k ETTh1 ETTm1 WTH wiki40b/en BookCorpus

Metrics: average kurtosis maximum infinity norm perplexity top-1 accuracy MSE MAE W8A8 quantized performance

Baselines: Vanilla attention Clipped Softmax Gated Attention Hopfield SparseHopfield STanHop-Net (GSH)

Builds on: modern Hopfield networks softmax_1 attention Clipped Softmax Gated Attention STanHop-Net

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy