Low-Rank Thinning
Annabelle Michael Carrell, Albert Gong, Abhishek Shetty, Raaz Dwivedi, Lester Mackey
Abstract
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.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 43390583-ee5c-4ca9-bed6-e748778a891bCited by top-tier papers4
- WildCat: Near-Linear Attention in Theory and PracticeTobias Schröder, Lester MackeyICML 2026 · 3 citations
- A Critical Look at Targeted Instruction Selection: Disentangling What Matters (and What Doesn’t)Nihal Nayak, Paula Rodriguez-Diaz, Neha Hulkund, Sara Beery et al.ICML 2026 · 2 citations
- Thinned Mean Field Langevin DynamicsZonghao Chen, Heishiro Kanagawa, Francois-Xavier Briol, Chris J Oates et al.ICML 2026 · 1 citation
- Stationary MMD PointsZonghao Chen, Toni Karvonen, Heishiro Kanagawa, Francois-Xavier Briol et al.ICML 2026
Builds on16
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn et al.ICLR 2021 · 21,477 citations
- FlashAttention: Fast and Memory-Efficient Exact Attention with IO-AwarenessTri Dao, Daniel Y. Fu, Stefano Ermon, Atri Rudra et al.NeurIPS 2022 · 5,493 citations
- Reformer: The Efficient TransformerNikita Kitaev, Lukasz Kaiser, Anselm LevskayaICLR 2020 · 2,878 citations
- FlashAttention-2: Faster Attention with Better Parallelism and Work PartitioningTri DaoICLR 2024 · 2,600 citations
- Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNetLi Yuan, Yunpeng Chen, Tao Wang, Weihao Yu et al.ICCV 2021 · 2,462 citations
Related papers
- Distribution Compression in Near-Linear TimeAbhishek Shetty, Raaz Dwivedi, Lester MackeyICLR 2022 · 24 citations
- Generalized Kernel ThinningRaaz Dwivedi, Lester MackeyICLR 2022 · 37 citations
- Debiased Distribution CompressionLingxiao Li, Raaz Dwivedi, Lester MackeyICML 2024 · 7 citations
- Supervised Kernel ThinningAlbert Gong, Kyuseong Choi, Raaz DwivediNeurIPS 2024 · 6 citations
- Block Subsampled Randomized Hadamard Transform for Nyström Approximation on Distributed ArchitecturesOleg Balabanov, Matthias Beaupère, Laura Grigori, Victor LedererICML 2023 · 13 citations
