Low-Rank Thinning
Annabelle Michael Carrell, Albert Gong, Abhishek Shetty, Raaz Dwivedi, Lester Mackey
摘要
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.
问问这篇 Paper
智能体会读完全文。
Lune 把这篇 Paper 索引到了每一个公式,引用它的顶会 Paper 也一样。你提问,回答直接引用原文。
引用它的顶会 Paper4
- WildCat: Near-Linear Attention in Theory and PracticeTobias Schröder, Lester MackeyICML 2026 · 被引用 3 次
- A Critical Look at Targeted Instruction Selection: Disentangling What Matters (and What Doesn’t)Nihal Nayak, Paula Rodriguez-Diaz, Neha Hulkund, Sara Beery 等ICML 2026 · 被引用 2 次
- Thinned Mean Field Langevin DynamicsZonghao Chen, Heishiro Kanagawa, Francois-Xavier Briol, Chris J Oates 等ICML 2026 · 被引用 1 次
- Stationary MMD PointsZonghao Chen, Toni Karvonen, Heishiro Kanagawa, Francois-Xavier Briol 等ICML 2026
它引用的顶会 Paper16
- An Image is Worth 16x16 Words: Transformers for Image Recognition at ScaleAlexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn 等ICLR 2021 · 被引用 21,477 次
- FlashAttention: Fast and Memory-Efficient Exact Attention with IO-AwarenessTri Dao, Daniel Y. Fu, Stefano Ermon, Atri Rudra 等NeurIPS 2022 · 被引用 5,493 次
- Reformer: The Efficient TransformerNikita Kitaev, Lukasz Kaiser, Anselm LevskayaICLR 2020 · 被引用 2,878 次
- FlashAttention-2: Faster Attention with Better Parallelism and Work PartitioningTri DaoICLR 2024 · 被引用 2,600 次
- Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNetLi Yuan, Yunpeng Chen, Tao Wang, Weihao Yu 等ICCV 2021 · 被引用 2,462 次
相关 Paper
- Distribution Compression in Near-Linear TimeAbhishek Shetty, Raaz Dwivedi, Lester MackeyICLR 2022 · 被引用 24 次
- Generalized Kernel ThinningRaaz Dwivedi, Lester MackeyICLR 2022 · 被引用 37 次
- Debiased Distribution CompressionLingxiao Li, Raaz Dwivedi, Lester MackeyICML 2024 · 被引用 7 次
- Supervised Kernel ThinningAlbert Gong, Kyuseong Choi, Raaz DwivediNeurIPS 2024 · 被引用 6 次
- Block Subsampled Randomized Hadamard Transform for Nyström Approximation on Distributed ArchitecturesOleg Balabanov, Matthias Beaupère, Laura Grigori, Victor LedererICML 2023 · 被引用 13 次
