REVIEW 3 major objections 5 minor 1 cited by
ZETA: Leveraging Z-order Curves for Efficient Top-k Attention
T0 review · 3 major / 5 minor · reviewed 2026-08-10 · deepseek-v4-flash
Pith's one-line read This paper claims that top-k attention can be made parallel across entire sequences under causal masks by sorting Z-order projections of low-dimensional keys and queries, and that the resulting method matches or outperforms standard…
desk verdict The Z-order idea is fresh and the experiments are real, but the causal-masking procedure as written is self-contradictory and the main theoretical bound goes imaginary in exactly the regime the paper uses, so the central claims don't hold up. read the letter →
The pith
A machine-rendered reading of the paper's core claim, the machinery that carries it, and where it could break.
The reading
What carries the argument
The load-bearing object is the Z-order curve (Morton code), which maps a $d$-dimensional point to a one-dimensional integer by interleaving the bits of its coordinates so that nearby points tend to get nearby codes. ZETA sorts the key codes once, splits the sorted list into chunks, and for each query in chunk $m$ binary-searches the query's code inside the first $m$ chunks and reads a fixed-size window around that insertion position as the top-$k$ set. The Adaptive Cauchy-Softmax then converts Euclidean distances in the low-dimensional space into attention weights through $\frac{1}{\|q-k\|^2 + \gamma^2}$, with $\gamma$ trainable per layer.
What would settle it
Run ZETA on a trained checkpoint, compute each query's exact $k$ nearest keys by Euclidean distance on the same low-dimensional keys, and measure the recall of ZETA's selected indices against that exact set; if recall is low on real data, the windowed-contiguity assumption fails.
Extended reading notes
Core claim
On its own terms, ZETA's central claim is that the top-$k$ attended tokens for every query in a causally masked sequence can be found in parallel by sorting all keys once in Z-order space, splitting the sorted list into chunks, and restricting each query in chunk $m$ to keys in the first $m$ chunks; the query's insertion position in that prefix locates a window of $k$ nearest neighbours. To make this geometry meaningful, the paper argues that keys and queries should have much lower dimension than values: small $d_K$ preserves relative distances after Z-order projection (supported by a bound built on the Johnson–Lindenstrauss lemma and a Lipschitz regression risk analysis), while a large $d_V$ preserves semantic richness. Attention weights are then computed with an Adaptive Cauchy-Softmax over Euclidean distances, with a trainable $\gamma$ controlling the receptive field, and the paper reports that ZETA matches vanilla attention on Multi-Query Associative Recall, achieves the best average accuracy among the compared models on Long Range Arena, reaches 26.3 perplexity on WikiText-103 against 26.2 for vanilla attention, and runs faster than a FlashAttention baseline at long sequence lengths.
Load-bearing premise
The central assumption is that the true top-$k$ keys for a query are contiguous around the query's insertion position in the sorted Z-order list, so a fixed window over the first $m$ chunks recovers them without scanning the full prefix.
Editorial extensions
If this is right
- A causal top-$k$ attention layer can be trained in parallel over the full sequence, with sorting dominating the cost at $O(N\log N)$ time and space rather than the $O(N^2)$ of standard attention.
- Key and query dimensions can be set far below value dimensions ($d_K = d_Q = 3$ in the paper's experiments) without degrading accuracy, because matching needs relative-distance preservation rather than semantic richness.
- Replacing the exponential softmax with a trainable Cauchy kernel over Euclidean distances keeps long-range tokens influential while letting each layer adjust its receptive field.
- On the reported benchmarks ZETA matches vanilla attention on Multi-Query Associative Recall, posts the best average LRA accuracy among the compared models, and reaches 26.3 WikiText-103 perplexity versus 26.2 for vanilla attention.
- At sequence lengths from 8K to 64K, the paper's custom-kernel implementation is faster than FlashAttention in forward and forward-backward passes, with modestly higher memory use.
Reading between the lines
- Beyond the paper, a natural stress test is to measure recall@k of ZETA's window retrieval against exact kNN on real trained keys; if recall is low, a data-dependent window size or a candidate re-ranking stage would be a direct fix.
- The chunk-prefix causal rule may discard past keys whose Z-order codes land in later chunks; an alternative is to sort each prefix independently or use a hierarchical index, trading speed for exactness.
- If the low-dimensional matching insight generalizes, other distance-based sparse attention schemes (LSH, product quantization, locality-sensitive retrieval) could adopt the same split between small matching dimensions and large value dimensions.
- Because the Cauchy kernel's $\gamma$ is trained per layer, one could test whether learned receptive fields correlate with task structure: sharp in local tasks, broad in long-range retrieval tasks.
Signed reviews
Editorial analysis
A structured set of objections, weighed in public.
Referee Report
Summary. The paper proposes ZETA, a sparse top-k attention mechanism that maps keys and queries to a low-dimensional space, then to one dimension via Z-order curves, sorts the keys, and partitions them into chunks to enable parallel top-k retrieval under causal masks. It claims O(N log N) time and space complexity and reports experiments on the Multi-Query Associative Recall task, Long Range Arena, and WikiText-103, along with a theoretical analysis motivating small key/query dimensionality and an adaptive Cauchy softmax. The paper is written as an ICLR 2025 conference paper and includes a custom Triton implementation with efficiency measurements.
Significance. If the algorithm worked as described, ZETA would address a genuine bottleneck: causal top-k attention is difficult to parallelize because future tokens must be excluded before retrieval. The paper also contains a substantial empirical component and a custom Triton implementation with reported speedups. However, the central algorithmic mechanism is internally inconsistent (see Major Comment 1), and the theoretical bound used to justify the low-dimensional projection is invalid in the relevant regime (Major Comment 2). These issues undermine the paper's main claims, so the current manuscript cannot be accepted.
major comments (3)
- [Section 3.2.2 and Algorithm 1] The described causal chunked search is internally inconsistent. After globally sorting the Z-order keys, the sorted list is a permutation of all N original positions. The paper says that a query in chunk m searches only the first m chunks and 'index[es] the original unsorted keys from 0 to m×M−1 in the sorted list,' and Algorithm 1 Step 4 says to 'Exclude keys from positions j > m×M.' These two conditions cannot hold simultaneously: the first m chunks of the sorted order will generally contain keys with original indices larger than m×M−1, and after excluding them, the number of valid keys is typically much less than k and often zero for early queries. Consequently, the method as specified does not retrieve k past keys per query and does not realize the claimed O(N log N) parallel causal top-k attention. Sorting each prefix separately would cost O(N^2 log N), and no incremental data structure is specified; thus the central efficiency claim is unsupported.
- [Section 3.2.1 and Appendix A] The risk bound in Theorem 3.3 contains the factor sqrt(1 − sqrt(C ln m / dK)). In the proof, epsilon is set to sqrt(C ln m / dK) in the Johnson–Lindenstrauss Lemma, which is only valid for epsilon < 1. For the paper's recommended dK = 3, this requires m < exp(3/C); for the sequence lengths considered in the paper (e.g., N = 4096 in LRA), m will typically exceed this threshold, making the bound imaginary and the derivation invalid. The theorem therefore does not establish the claimed trade-off between the curse of dimensionality and preservation of relative distances in the regime where ZETA operates.
- [Section 3.2.2] The nearest-neighbor retrieval step selects a window of size k around the query's insertion position in the sorted Z-order list and treats the keys in that window as the top-k set. The paper supplies no bound or proof that the Euclidean top-k neighbors are contained in such a window after sorting; Z-order curves preserve locality only approximately, and the locality experiment in Section 4.4 measures neighbor overlap after projection, not contiguity after sorting. Without a retrieval-recall guarantee, the method is not demonstrably a top-k attention method, and the experimental accuracy cannot be attributed to the described selection mechanism.
minor comments (5)
- [Appendix A] The paragraph before Lemma A.1 contains an incomplete sentence and an unresolved URL: 'The recent paper, Reformer, proposed the Shared-QK Transformer... https://arxiv.org/abs/2001.0445a very simple but efficient technique.' This appears to be a leftover note and should be rewritten or removed.
- [Appendix D] There is a typo in 'he @triton.autotune decorator is used...' which should read 'The @triton.autotune decorator is used...'.
- [Table 2] The claim of consistent improvement over baselines is qualified by the Pathfinder row: ZETA's accuracy (68.20) is lower than that of the vanilla Transformer (71.40) and several other variants. The narrative in Section 4.1 should be adjusted to reflect this.
- [Section 3.2.2] The notation is confusing: the top-k value is sometimes written as k and sometimes as K (e.g., in Section 3.2.2, 'a window of size K centered around the insertion point'). Please standardize the notation.
- [Appendix F] The Limitations section acknowledges the general risk of ignoring important tokens in top-k attention, but it does not mention the causal-masking inconsistency described in Major Comment 1, which is a more direct limitation of the proposed algorithm.
Circularity Check
No significant circularity: ZETA's contributions are an algorithmic construction plus empirical evaluation; the theoretical bound is derived from the external Johnson–Lindenstrauss lemma and standard Lipschitz/simplex assumptions, and its tunable quantities are hyperparameters, not fitted constants relabeled as predictions.
full rationale
The paper's claimed derivation chain is self-contained rather than circular. ZETA's central mechanism is algorithmic: keys are projected to one dimension via Z-order curves, sorted, chunked, and searched by insertion position (Section 3.2.2 and Algorithm 1). No prediction in the paper is obtained by fitting a parameter to a target quantity and then reporting that quantity; dK is chosen by ablations (Section 4.4 and Appendix G.1), k is set to 32 after an ablation (Section 4.5), and the Cauchy softmax scale gamma is a trainable parameter (Section 3.3 and Appendix E). The theoretical claim about dK (Theorem 3.3) is derived from the Johnson–Lindenstrauss lemma, Lipschitzness, and a covering argument with an external textbook lemma (Lemma A.4), not from a result whose authors overlap with this paper. The only self-citations appear in the introduction as routine prior-work mentions (Zeng et al. 2023; 2024a; 2024b; Fang et al. 2025) and are not load-bearing. The causal-masking inconsistency raised by a skeptical reading (a single global sort cannot simultaneously supply the first-m-chunk prefix and the original-position exclusion j > m*M) is a substantive correctness concern, but it is not a circularity: the algorithm does not define its output in terms of the quantity it claims to predict. Similarly, the possible invalidity of the sqrt(1 - sqrt(C ln m / dK)) bound at dK = 3 is a mathematical gap, not a circular reduction. Accordingly, the circularity score is 0.
Assumptions & free parameters
free parameters (4)
- key/query dimension dK =
3
- top-k window size k =
32
- Cauchy kernel width gamma =
trainable, sigmoid output in [0, 1]
- number of chunks =
4, 8, 16, 32
assumptions (6)
- standard math Johnson-Lindenstrauss lemma with dK = Omega(ln m / epsilon^2)
- ad hoc to paper Assumption 3.2: an optimal learnable similarity critic Gamma exists and attention weights lie in a simplex
- domain assumption Z-order curves preserve locality well enough that a fixed-size window around the insertion position contains the true top-k nearest neighbors
- ad hoc to paper The chunked causal search can be implemented with a single global sort while excluding future original positions
- domain assumption Euclidean distance with a Cauchy kernel can replace dot-product softmax attention
- domain assumption The conditional mean function h* is l-Lipschitz and projections are bounded by B
Cite this review
Pith. "Pith review of ZETA: Leveraging Z-order Curves for Efficient Top-k Attention." pith.science (2026). https://pith.science/paper/H2A5K2HQ
@misc{pith2026250114577,
author = {Pith},
title = {Pith review of: ZETA: Leveraging Z-order Curves for Efficient Top-k Attention},
year = {2026},
howpublished = {\url{https://pith.science/paper/H2A5K2HQ}},
note = {Machine review of arXiv:2501.14577}
}
abstract
Over recent years, the Transformer has become a fundamental building block for sequence modeling architectures. Yet at its core is the use of self-attention, whose memory and computational cost grow quadratically with the sequence length $N$, rendering it prohibitively expensive for long sequences. A promising approach is top-$k$ attention, which selects only the $k$ most relevant tokens and achieves performance comparable to vanilla self-attention while significantly reducing space and computational demands. However, causal masks require the current query token to only attend to past tokens, preventing the existing top-$k$ attention method from efficiently searching for the most relevant tokens in parallel, thereby limiting training efficiency. In this work, we propose ZETA, leveraging \textbf{Z}-Order Curves for \textbf{E}fficient \textbf{T}op-$k$ \textbf{A}ttention, to enable parallel querying of past tokens for entire sequences. % in both space and time complexity of $\mathcal{O}(N \log N)$. We first theoretically show that the choice of key and query dimensions involves a trade-off between the curse of dimensionality and the preservation of relative distances after projection. In light of this insight, we propose reducing the dimensionality of keys and queries in contrast to values and further leverage $Z$-order curves to map low-dimensional keys and queries into \emph{one}-dimensional space, which permits parallel sorting, thereby largely improving the efficiency for top-$k$ token selection. Experimental results demonstrate that ZETA matches the performance of standard attention on the synthetic \textsc{Multi-Query Associative Recall} task and outperforms attention and its variants on \textsc{Long Range Arena} and \textsc{WikiText-103} language modeling.
Figures
Forward citations
Cited by 1 Pith paper
-
Balancing Computation Load and Representation Expressivity in Parallel Hybrid Neural Networks
FlowHN splits input tokens between parallel attention and SSM branches to balance compute load, improving throughput and hardware utilization in small autoregressive language models.
Reference graph
Works this paper leans on
-
[1]
write newline
" write newline "" before.all 'output.state := FUNCTION n.dashify 't := "" t empty not t #1 #1 substring "-" = t #1 #2 substring "--" = not "--" * t #2 global.max substring 't := t #1 #1 substring "-" = "-" * t #2 global.max substring 't := while if t #1 #1 substring * t #2 global.max substring 't := if while FUNCTION format.date year duplicate empty "emp...
-
[2]
Herv \'e Abdi and Lynne J Williams. Principal component analysis. Wiley interdisciplinary reviews: computational statistics, 2 0 (4): 0 433--459, 2010
work page 2010
-
[3]
Zoology: Measuring and improving recall in efficient language models
Simran Arora, Sabri Eyuboglu, Aman Timalsina, Isys Johnson, Michael Poli, James Zou, Atri Rudra, and Christopher Re. Zoology: Measuring and improving recall in efficient language models. In The Twelfth International Conference on Learning Representations, 2024 a . URL https://openreview.net/forum?id=LY3ukUANko
work page 2024
-
[4]
Simple linear attention language models balance the recall-throughput tradeoff
Simran Arora, Sabri Eyuboglu, Michael Zhang, Aman Timalsina, Silas Alberti, James Zou, Atri Rudra, and Christopher Re. Simple linear attention language models balance the recall-throughput tradeoff. In Forty-first International Conference on Machine Learning, 2024 b
work page 2024
-
[5]
Neural machine translation by jointly learning to align and translate
Dzmitry Bahdanau, Kyunghyun Cho, and Yoshua Bengio. Neural machine translation by jointly learning to align and translate. In International Conference on Learning Representations, 2015
work page 2015
-
[6]
Longformer: The long-document transformer
Iz Beltagy, Matthew E Peters, and Arman Cohan. Longformer: The long-document transformer. arXiv preprint arXiv:2004.05150, 2020
arXiv 2004
-
[7]
Amanda Bertsch, Uri Alon, Graham Neubig, and Matthew R. Gormley. Unlimiformer: Long-range transformers with unlimited length input. In Thirty-seventh Conference on Neural Information Processing Systems, 2023. URL https://openreview.net/forum?id=lJWUJWLCJo
work page 2023
-
[8]
Pythia: A suite for analyzing large language models across training and scaling
Stella Biderman, Hailey Schoelkopf, Quentin Gregory Anthony, Herbie Bradley, Kyle O’Brien, Eric Hallahan, Mohammad Aflah Khan, Shivanshu Purohit, USVSN Sai Prashanth, Edward Raff, et al. Pythia: A suite for analyzing large language models across training and scaling. In International Conference on Machine Learning, pp.\ 2397--2430. PMLR, 2023
2023
Show all 59 references
-
[9]
Probability and Measure
Patrick Billingsley. Probability and Measure. John Wiley and Sons, second edition, 1986
1986
-
[10]
Video generation models as world simulators, 2024
Tim Brooks, Bill Peebles, Connor Holmes, Will DePue, Yufei Guo, Li Jing, David Schnurr, Joe Taylor, Troy Luhman, Eric Luhman, Clarence Ng, Ricky Wang, and Aditya Ramesh. Video generation models as world simulators, 2024. URL https://openai.com/research/video-generation-models-...
2024
-
[11]
Tom B. Brown, Benjamin Mann, Nick Ryder, Melanie Subbiah, Jared Kaplan, Prafulla Dhariwal, Arvind Neelakantan, Pranav Shyam, Girish Sastry, Amanda Askell, Sandhini Agarwal, Ariel Herbert-Voss, Gretchen Krueger, Tom Henighan, Rewon Child, Aditya Ramesh, Daniel M. Ziegler, Jeffr...
2020
-
[12]
Skyformer: Remodel self-attention with gaussian kernel and nystr \"o m method
Yifan Chen, Qi Zeng, Heng Ji, and Yun Yang. Skyformer: Remodel self-attention with gaussian kernel and nystr \"o m method. In A. Beygelzimer, Y. Dauphin, P. Liang, and J. Wortman Vaughan (eds.), Advances in Neural Information Processing Systems, 2021. URL https://openreview.ne...
2021
-
[13]
Generating long sequences with sparse transformers
Rewon Child, Scott Gray, Alec Radford, and Ilya Sutskever. Generating long sequences with sparse transformers. arXiv preprint arXiv:1904.10509, 2019
1904 arXiv
-
[14]
Rethinking attention with performers
Krzysztof Marcin Choromanski, Valerii Likhosherstov, David Dohan, Xingyou Song, Andreea Gane, Tamas Sarlos, Peter Hawkins, Jared Quincy Davis, Afroz Mohiuddin, Lukasz Kaiser, David Benjamin Belanger, Lucy J Colwell, and Adrian Weller. Rethinking attention with performers. In I...
2021
-
[15]
Cover and Joy A
Thomas M. Cover and Joy A. Thomas. Elements of information theory. John Wiley & Sons, 2006
2006
-
[16]
Flash A ttention-2: Faster attention with better parallelism and work partitioning
Tri Dao. Flash A ttention-2: Faster attention with better parallelism and work partitioning. In International Conference on Learning Representations (ICLR), 2024
2024
-
[17]
Fu, Stefano Ermon, Atri Rudra, and Christopher R \'e
Tri Dao, Daniel Y. Fu, Stefano Ermon, Atri Rudra, and Christopher R \'e . Flash A ttention: Fast and memory-efficient exact attention with IO -awareness. In Advances in Neural Information Processing Systems (NeurIPS), 2022
2022
-
[18]
BERT : Pre-training of deep bidirectional transformers for language understanding
Jacob Devlin, Ming-Wei Chang, Kenton Lee, and Kristina Toutanova. BERT : Pre-training of deep bidirectional transformers for language understanding. In Jill Burstein, Christy Doran, and Thamar Solorio (eds.), Proceedings of the 2019 Conference of the North A merican Chapter of...
2019 doi
-
[19]
An image is worth 16x16 words: Transformers for image recognition at scale
Alexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn, Xiaohua Zhai, Thomas Unterthiner, Mostafa Dehghani, Matthias Minderer, Georg Heigold, Sylvain Gelly, Jakob Uszkoreit, and Neil Houlsby. An image is worth 16x16 words: Transformers for image recognition at...
2021
-
[20]
Topology
James Dugundji. Topology. Allyn and Bacon, 1966
1966
-
[21]
Structure-preserving graph representation learning
Ruiyi Fang, Liangjian Wen, Zhao Kang, and Jianzhuang Liu. Structure-preserving graph representation learning. In 2022 IEEE International Conference on Data Mining (ICDM), pp.\ 927--932. IEEE, 2022
2022
-
[22]
On the benefits of attribute-driven graph domain adaptation
Ruiyi Fang, Bingheng Li, Zhao Kang, Qiuhao Zeng, Ruizhi Pu, Nima Hosseini Dashtbayaz, Boyu Wang, and Charles Ling. On the benefits of attribute-driven graph domain adaptation. The Thirteenth International Conference on Learning Representations, 2025
2025
-
[23]
Mamba: Linear-time sequence modeling with selective state spaces
Albert Gu and Tri Dao. Mamba: Linear-time sequence modeling with selective state spaces. In First Conference on Language Modeling, 2024. URL https://openreview.net/forum?id=tEYskw1VY2
2024
-
[24]
Memory-efficient transformers via top-k attention
Ankit Gupta, Guy Dar, Shaya Goodman, David Ciprut, and Jonathan Berant. Memory-efficient transformers via top-k attention. In Nafise Sadat Moosavi, Iryna Gurevych, Angela Fan, Thomas Wolf, Yufang Hou, Ana Marasovi \'c , and Sujith Ravi (eds.), Proceedings of the Second Worksho...
2021 doi
-
[25]
Axial attention in multidimensional transformers, 2020
Jonathan Ho, Nal Kalchbrenner, Dirk Weissenborn, and Tim Salimans. Axial attention in multidimensional transformers, 2020. URL https://openreview.net/forum?id=H1e5GJBtDr
2020
- [26]
-
[27]
Extensions of lipschitz maps into banach spaces
William B Johnson, Joram Lindenstrauss, and Gideon Schechtman. Extensions of lipschitz maps into banach spaces. Israel Journal of Mathematics, 54 0 (2): 0 129--138, 1986
1986
-
[28]
Daniel Jurafsky and James H. Martin. Speech and Language Processing: An Introduction to Natural Language Processing, Computational Linguistics, and Speech Recognition with Language Models. Pearson, 3rd edition, 2024. URL https://web.stanford.edu/ jurafsky/slp3/
2024
-
[29]
Reformer: The efficient transformer
Nikita Kitaev, Lukasz Kaiser, and Anselm Levskaya. Reformer: The efficient transformer. In International Conference on Learning Representations, 2020. URL https://openreview.net/forum?id=rkgNKkHtvB
2020
-
[30]
Iceformer: Accelerated inference with long-sequence transformers on CPU s
Yuzhen Mao, Martin Ester, and Ke Li. Iceformer: Accelerated inference with long-sequence transformers on CPU s. In The Twelfth International Conference on Learning Representations, 2024. URL https://openreview.net/forum?id=6RR3wU4mSZ
2024
-
[31]
Umap: Uniform manifold approximation and projection
Leland McInnes, John Healy, Nathaniel Saul, and Lukas Grossberger. Umap: Uniform manifold approximation and projection. The Journal of Open Source Software, 3 0 (29): 0 861, 2018
2018
-
[32]
Pointer sentinel mixture models
Stephen Merity, Caiming Xiong, James Bradbury, and Richard Socher. Pointer sentinel mixture models. In International Conference on Learning Representations, 2017. URL https://openreview.net/forum?id=Byj72udxe
2017
-
[33]
OpenAI, Josh Achiam, Steven Adler, Sandhini Agarwal, Lama Ahmad, Ilge Akkaya, Florencia Leoni Aleman, Diogo Almeida, Janko Altenschmidt, Sam Altman, Shyamal Anadkat, Red Avila, Igor Babuschkin, Suchir Balaji, Valerie Balcom, Paul Baltescu, Haiming Bao, Mohammad Bavarian, Jeff ...
2024 arXiv
-
[34]
Scaling neural machine translation
Myle Ott, Sergey Edunov, David Grangier, and Michael Auli. Scaling neural machine translation. In Ond r ej Bojar, Rajen Chatterjee, Christian Federmann, Mark Fishel, Yvette Graham, Barry Haddow, Matthias Huck, Antonio Jimeno Yepes, Philipp Koehn, Christof Monz, Matteo Negri, A...
2018
-
[35]
Yang, Zachary DeVito, Martin Raison, Alykhan Tejani, Sasank Chilamkurthy, Benoit Steiner, Lu Fang, Junjie Bai, and Soumith Chintala
Adam Paszke, Sam Gross, Francisco Massa, Adam Lerer, James Bradbury, Gregory Chanan, Trevor Killeen, Zeming Lin, Natalia Gimelshein, Luca Antiga, Alban Desmaison, Andreas K \" o pf, Edward Z. Yang, Zachary DeVito, Martin Raison, Alykhan Tejani, Sasank Chilamkurthy, Benoit Stei...
2019
-
[36]
The devil in linear transformer
Zhen Qin, Xiaodong Han, Weixuan Sun, Dongxu Li, Lingpeng Kong, Nick Barnes, and Yiran Zhong. The devil in linear transformer. In Yoav Goldberg, Zornitsa Kozareva, and Yue Zhang (eds.), Proceedings of the 2022 Conference on Empirical Methods in Natural Language Processing, EMNL...
2022 doi
-
[37]
cosformer: Rethinking softmax in attention
Zhen Qin, Weixuan Sun, Hui Deng, Dongxu Li, Yunshen Wei, Baohong Lv, Junjie Yan, Lingpeng Kong, and Yiran Zhong. cosformer: Rethinking softmax in attention. In International Conference on Learning Representations, 2022 b . URL https://openreview.net/forum?id=Bl8CQrx2Up4
2022
-
[38]
Language models are unsupervised multitask learners, 2019
Alec Radford, Jeffrey Wu, Rewon Child, David Luan, Dario Amodei, Ilya Sutskever, et al. Language models are unsupervised multitask learners, 2019
2019
-
[39]
Zero-shot text-to-image generation
Aditya Ramesh, Mikhail Pavlov, Gabriel Goh, Scott Gray, Chelsea Voss, Alec Radford, Mark Chen, and Ilya Sutskever. Zero-shot text-to-image generation. In Marina Meila and Tong Zhang (eds.), Proceedings of the 38th International Conference on Machine Learning, ICML 2021, 18-24 ...
2021
-
[40]
Understanding Machine Learning: From Theory to Algorithms
Shai Shalev-Shwartz and Shai Ben-David. Understanding Machine Learning: From Theory to Algorithms. Cambridge University Press, 2014
2014
- [41]
-
[42]
Sparse S inkhorn attention
Yi Tay, Dara Bahri, Liu Yang, Donald Metzler, and Da-Cheng Juan. Sparse S inkhorn attention. In Hal Daumé III and Aarti Singh (eds.), Proceedings of the 37th International Conference on Machine Learning, volume 119 of Proceedings of Machine Learning Research, pp.\ 9438--9447. ...
2020
-
[43]
Long range arena : A benchmark for efficient transformers
Yi Tay, Mostafa Dehghani, Samira Abnar, Yikang Shen, Dara Bahri, Philip Pham, Jinfeng Rao, Liu Yang, Sebastian Ruder, and Donald Metzler. Long range arena : A benchmark for efficient transformers. In International Conference on Learning Representations, 2021. URL https://openr...
2021
-
[44]
Efficient transformers: A survey
Yi Tay, Mostafa Dehghani, Dara Bahri, and Donald Metzler. Efficient transformers: A survey. ACM Comput. Surv., 55 0 (6), December 2022. ISSN 0360-0300. doi:10.1145/3530811. URL https://doi.org/10.1145/3530811
2022 doi
-
[45]
Training data-efficient image transformers & distillation through attention
Hugo Touvron, Matthieu Cord, Matthijs Douze, Francisco Massa, Alexandre Sablayrolles, and Herve Jegou. Training data-efficient image transformers & distillation through attention. In Marina Meila and Tong Zhang (eds.), Proceedings of the 38th International Conference on Machin...
2021
-
[46]
Gomez, ukasz Kaiser, and Illia Polosukhin
Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N. Gomez, ukasz Kaiser, and Illia Polosukhin. Attention is all you need. In I. Guyon, U. Von Luxburg, S. Bengio, H. Wallach, R. Fergus, S. Vishwanathan, and R. Garnett (eds.), Advances in Neural Inf...
2017
-
[47]
Linformer: Self-attention with linear complexity
Sinong Wang, Belinda Z Li, Madian Khabsa, Han Fang, and Hao Ma. Linformer: Self-attention with linear complexity. arXiv preprint arXiv:2006.04768, 2020
2006 arXiv
-
[48]
o mformer: A nystr \
Yunyang Xiong, Zhanpeng Zeng, Rudrasis Chakraborty, Mingxing Tan, Glenn Fung, Yin Li, and Vikas Singh. Nystr \"o mformer: A nystr \"o m-based algorithm for approximating self-attention. In Proceedings of the AAAI Conference on Artificial Intelligence, 2021
2021
-
[49]
Big bird: Transformers for longer sequences
Manzil Zaheer, Guru Guruganesh, Kumar Avinava Dubey, Joshua Ainslie, Chris Alberti, Santiago Ontanon, Philip Pham, Anirudh Ravula, Qifan Wang, Li Yang, and Amr Ahmed. Big bird: Transformers for longer sequences. 33: 0 17283--17297, 2020. URL https://proceedings.neurips.cc/pape...
2020
-
[50]
Foresee what you will learn: data augmentation for domain generalization in non-stationary environment
Qiuhao Zeng, Wei Wang, Fan Zhou, Charles Ling, and Boyu Wang. Foresee what you will learn: data augmentation for domain generalization in non-stationary environment. In Proceedings of the AAAI conference on artificial intelligence, volume 37, pp.\ 11147--11155, 2023
2023
-
[51]
Towards understanding evolving patterns in sequential data
QIUHAO Zeng, Long-Kai Huang, Qi Chen, Charles X Ling, and Boyu Wang. Towards understanding evolving patterns in sequential data. Advances in Neural Information Processing Systems, 37: 0 132747--132773, 2024 a
2024
-
[52]
Latent trajectory learning for limited timestamps under distribution shift over time
QIUHAO Zeng, Changjian Shui, Long-Kai Huang, Peng Liu, Xi Chen, Charles Ling, and Boyu Wang. Latent trajectory learning for limited timestamps under distribution shift over time. In The Twelfth International Conference on Learning Representations, 2024 b . URL https://openrevi...
2024
-
[53]
Susskind
Shuangfei Zhai, Tatiana Likhomanenko, Etai Littwin, Dan Busbridge, Jason Ramapuram, Yizhe Zhang, Jiatao Gu, and Joshua M. Susskind. Stabilizing transformer training by preventing attention entropy collapse. In Andreas Krause, Emma Brunskill, Kyunghyun Cho, Barbara Engelhardt, ...
2023
-
[54]
The hedgehog & the porcupine: Expressive linear attentions with softmax mimicry
Michael Zhang, Kush Bhatia, Hermann Kumbong, and Christopher Re. The hedgehog & the porcupine: Expressive linear attentions with softmax mimicry. In The Twelfth International Conference on Learning Representations, 2024. URL https://openreview.net/forum?id=4g02l2N2Nx
2024
-
[55]
H -transformer-1 D : Fast one-dimensional hierarchical attention for sequences
Zhenhai Zhu and Radu Soricut. H -transformer-1 D : Fast one-dimensional hierarchical attention for sequences. In Chengqing Zong, Fei Xia, Wenjie Li, and Roberto Navigli (eds.), Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th...
2021 doi
-
[56]
Efficient attention: Attention with linear complexities
Shen Zhuoran, Zhang Mingyuan, Zhao Haiyu, Yi Shuai, and Li Hongsheng. Efficient attention: Attention with linear complexities. In 2021 IEEE Winter Conference on Applications of Computer Vision (WACV), 2021
2021
-
[57]
@esa (Ref
\@ifxundefined[1] #1\@undefined \@firstoftwo \@secondoftwo \@ifnum[1] #1 \@firstoftwo \@secondoftwo \@ifx[1] #1 \@firstoftwo \@secondoftwo [2] @ #1 \@temptokena #2 #1 @ \@temptokena \@ifclassloaded agu2001 natbib The agu2001 class already includes natbib coding, so you should ...
-
[58]
\@lbibitem[] @bibitem@first@sw\@secondoftwo \@lbibitem[#1]#2 \@extra@b@citeb \@ifundefined br@#2\@extra@b@citeb \@namedef br@#2 \@nameuse br@#2\@extra@b@citeb \@ifundefined b@#2\@extra@b@citeb @num @parse #2 @tmp #1 NAT@b@open@#2 NAT@b@shut@#2 \@ifnum @merge>\@ne @bibitem@firs...
-
[59]
= Y S RF8yV:Ĉ` LB , JDP P[; QX,PueK6ڞ G:EZt x( ɠOU(T P w 3Ѝ * SKE <y فB0;a \!hi BC=lS;6m9h'VlSh D:E L ܓo K i N )pI[ wbk
@open @close @open @close and [1] URL: #1 \@ifundefined chapter * \@mkboth \@ifxundefined @sectionbib * \@mkboth * \@mkboth\@gobbletwo \@ifclassloaded amsart * \@ifclassloaded amsbook * \@ifxundefined @heading @heading NAT@ctr thebibliography [1] @ \@biblabel @NAT@ctr \@bibset...
2023
Reviewed August 10, 2026 · model on record in the stance chip above.
Discussion (0). Continue with ORCID to comment.