Unveiling Invariances via Neural Network Pruning
Not on the map. Why?
Abstract (arXiv)
Invariance describes transformations that do not alter data's underlying semantics. Neural networks that preserve natural invariance capture good inductive biases and achieve superior performance. Hence, modern networks are handcrafted to handle well-known invariances (ex. translations). We propose a framework to learn novel network architectures that capture data-dependent invariances via pruning. Our learned architectures consistently outperform dense neural networks on both vision and tabular datasets in both efficiency and effectiveness. We demonstrate our framework on multiple deep learning models across 3 vision and 40 tabular datasets.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 70 of its 89 references that are in the map. First posted on arXiv 15 Sep 2023.
Builds on
70 of 89 references.
References · 70
- 2017Attention Is All You Needcited by 60,361
- 2015Deep Residual Learning for Image Recognitioncited by 52,096
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
- 2016XGBoost: A Scalable Tree Boosting Systemcited by 5,166
- 2017PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Spacecited by 4,401
- 2020wav2vec 2.0: A Framework for Self-Supervised Learning of Speech Representationscited by 3,905
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
- 2019A Comprehensive Survey on Graph Neural Networkscited by 2,919
- 2015FaceNet: A Unified Embedding for Face Recognition and Clusteringcited by 2,849
- 2018The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networkscited by 2,262
- 2016Convolutional Neural Networks on Graphs with Fast Localized Spectral Filteringcited by 2,089
38 more
- 2018DARTS: Differentiable Architecture Searchcited by 1,798
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,616
- 2021MLP-Mixer: An all-MLP Architecture for Visioncited by 1,446
- 2015Spatial Transformer Networkscited by 1,429
- 2020Graph Contrastive Learning with Augmentationscited by 1,208
- 2016Group Equivariant Convolutional Networkscited by 1,165
- 2017Learning Transferable Architectures for Scalable Image Recognitioncited by 1,101
- 2016Geometric deep learning: going beyond Euclidean datacited by 1,048
- 2021E(n) Equivariant Graph Neural Networkscited by 965
- 2017A Closer Look at Memorization in Deep Networkscited by 942
- 2018Efficient Neural Architecture Search via Parameter Sharingcited by 837
- 2021Revisiting Deep Learning Models for Tabular Datacited by 740
- 2018AutoAugment: Learning Augmentation Policies from Datacited by 730
- 2017Dynamic Routing Between Capsulescited by 689
- 2018On Lazy Training in Differentiable Programmingcited by 675
- 2022TabPFN: A Transformer That Solves Small Tabular Classification Problems in a Secondcited by 674
- 2019TabNet: Attentive Interpretable Tabular Learningcited by 648
- 2018Rethinking the Value of Network Pruningcited by 599
- 2017Learning Sparse Neural Networks through L0 Regularizationcited by 583
- 2020What is the State of Neural Network Pruning?cited by 583
- 2020A Survey on Contrastive Self-supervised Learningcited by 554
- 2021Tabular Data: Deep Learning is Not All You Needcited by 535
- 2020Graph Contrastive Learning with Adaptive Augmentationcited by 519
- 2019Rigging the Lottery: Making All Tickets Winnerscited by 495
- 2021Deep Neural Networks and Tabular Data: A Surveycited by 491
- 2020Picking Winning Tickets Before Training by Preserving Gradient Flowcited by 471
- 2015Deep Metric Learning via Lifted Structured Feature Embeddingcited by 449
- 2020TabTransformer: Tabular Data Modeling Using Contextual Embeddingscited by 409
- 2017Snapshot Ensembles: Train 1, get M for freecited by 350
- 2014In Search of the Real Inductive Bias: On the Role of Implicit Regularization in Deep Learningcited by 342
- 2021Vector Neurons: A General Framework for SO(3)-Equivariant Networkscited by 305
- 2019Optimizing Millions of Hyperparameters by Implicit Differentiationcited by 304
- 2021Layer-wise Analysis of a Self-supervised Speech Representation Modelcited by 298
- 2019Neural Oblivious Decision Ensembles for Deep Learning on Tabular Datacited by 203
- 2022Why do tree-based models still outperform deep learning on tabular data?cited by 199
- 2022On Embeddings for Numerical Features in Tabular Deep Learningcited by 186
- 2018Neural Architecture Optimizationcited by 167
- 2021SCARF: Self-Supervised Contrastive Learning using Random Feature Corruptioncited by 156
50 most cited of 70 All 70 (JSON)
Built on by
Not cited yet.