mlsanityData to 14 Oct 2026

Low-Rank Thinning

arXiv 2502.12063 ↗First posted 17 Feb 2025stat.ML cs.LG math.OCcited by 7

Not on the map. Why?

Abstract (arXiv)

The goal in thinning is to summarize a dataset using a small set of representative points. Remarkably, sub-Gaussian thinning algorithms like Kernel Halving and Compress can match the quality of uniform subsampling while substantially reducing the number of summary points. However, existing guarantees cover only a restricted range of distributions and kernel-based quality measures and suffer from pessimistic dimension dependence. To address these deficiencies, we introduce a new low-rank analysis of sub-Gaussian thinning that applies to any distribution and any kernel, guaranteeing high-quality compression whenever the kernel or data matrix is approximately low-rank. To demonstrate the broad applicability of the techniques, we design practical sub-Gaussian thinning approaches that improve upon the best known guarantees for approximating attention in transformers, accelerating stochastic gradient training through reordering, and distinguishing distributions in near-linear time.

Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Builds on 31 of its 45 references that are in the map. First posted on arXiv 17 Feb 2025.

Builds on

31 of 45 references.

References · 31

  1. 2017Attention Is All You Needcited by 60,361
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  4. 2014ImageNet Large Scale Visual Recognition Challengecited by 8,498
  5. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  6. 2016Improved Techniques for Training GANscited by 2,970
  7. 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,598
  8. 2020Reformer: The Efficient Transformercited by 1,668
  9. 2020Rethinking Attention with Performerscited by 1,601
  10. 2021Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNetcited by 1,101
  11. 2020Learning Deep Kernels for Non-Parametric Two-Sample Testscited by 130
  12. 2021Scatterbrain: Unifying Sparse and Low-rank Attention Approximationcited by 130
19 more
  1. 2023Fast Attention Requires Bounded Entriescited by 102
  2. 2023HyperAttention: Long-context Attention in Near-Linear Timecited by 85
  3. 2020Closing the convergence gap of SGD without replacementcited by 57
  4. 2021Kernel Thinningcited by 47
  5. 2018Massively scalable Sinkhorn distances via the Nystr"om methodcited by 41
  6. 2018Near-Optimal Coresets of Kernel Density Estimatescited by 30
  7. 2014A Wild Bootstrap for Degenerate Kernel Testscited by 29
  8. 2021Generalized Kernel Thinningcited by 29
  9. 2020Discrepancy Minimization via a Self-Balancing Walkcited by 22
  10. 2023Tighter Lower Bounds for Shuffling SGD: Random Permutations and Beyondcited by 22
  11. 2017LDMNet: Low Dimensional Manifold Regularized Neural Networkscited by 21
  12. 2021Distribution Compression in Near-linear Timecited by 19
  13. 2022GraB: Finding Provably Better Data Permutations than Random Reshufflingcited by 19
  14. 2023Compress Then Test: Powerful Kernel Testing in Near-linear Timecited by 14
  15. 2019Balancing Covariates in Randomized Experiments with the Gram-Schmidt Walk Designcited by 13
  16. 2017The Gram-Schmidt Walk: A Cure for the Banaszczyk Bluescited by 10
  17. 2023Differentially Private Permutation Testscited by 10
  18. 2024Debiased Distribution Compressioncited by 8
  19. 2023Coordinating Distributed Example Orders for Provably Accelerated Trainingcited by 7

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 1; 2026: 6.

7 papers.