YiteWang / NTK-SAP
[ICLR2023] NTK-SAP: Improving neural network pruning by aligning training dynamics
☆17Updated last year
Related projects: ⓘ
- Deep Learning & Information Bottleneck☆45Updated last year
- ☆11Updated 2 years ago
- [ICLR 2021] "Long Live the Lottery: The Existence of Winning Tickets in Lifelong Learning" by Tianlong Chen*, Zhenyu Zhang*, Sijia Liu, S…☆22Updated 2 years ago
- ☆10Updated 2 years ago
- [NeurIPS‘2021] "MEST: Accurate and Fast Memory-Economic Sparse Training Framework on the Edge", Geng Yuan, Xiaolong Ma, Yanzhi Wang et al…☆17Updated 2 years ago
- Code repository for the paper "Invariant and Transportable Representations for Anti-Causal Domain Shifts"☆12Updated 2 years ago
- ☆24Updated 3 years ago
- A modern look at the relationship between sharpness and generalization [ICML 2023]☆42Updated last year
- Source code of "What can linearized neural networks actually say about generalization?☆17Updated 2 years ago
- Prospect Pruning: Finding Trainable Weights at Initialization Using Meta-Gradients☆28Updated 2 years ago
- [ICLR 2023] 'Revisiting Pruning At Initialization Through The Lens of Ramanujan Graph" by Duc Hoang, Shiwei Liu, Radu Marculescu, Atlas W…☆13Updated last year
- Code for Sanity-Checking Pruning Methods: Random Tickets can Win the Jackpot☆43Updated 3 years ago
- Recycling diverse models☆42Updated last year
- Latest Weight Averaging (NeurIPS HITY 2022)☆21Updated last year
- A generic code base for neural network pruning, especially for pruning at initialization.☆30Updated 2 years ago
- A simple Jax implementation of influence functions.☆15Updated 5 months ago
- Repo for the paper: "Agree to Disagree: Diversity through Disagreement for Better Transferability"☆34Updated last year
- ☆20Updated last year
- [ICLR 2022] "Sparsity Winning Twice: Better Robust Generalization from More Efficient Training" by Tianlong Chen*, Zhenyu Zhang*, Pengjun…☆37Updated 2 years ago
- Reproducing RigL (ICML 2020) as a part of ML Reproducibility Challenge 2020☆26Updated 2 years ago
- ☆17Updated last year
- [ICML2022] Training Your Sparse Neural Network Better with Any Mask. Ajay Jaiswal, Haoyu Ma, Tianlong Chen, ying Ding, and Zhangyang Wang☆23Updated 2 years ago
- [ICML 2021] "Do We Actually Need Dense Over-Parameterization? In-Time Over-Parameterization in Sparse Training" by Shiwei Liu, Lu Yin, De…☆46Updated 10 months ago
- Good Subnetworks Provably Exist: Pruning via Greedy Forward Selection☆21Updated 3 years ago
- Code for "Training Neural Networks with Fixed Sparse Masks" (NeurIPS 2021).☆54Updated 2 years ago
- Understanding Rare Spurious Correlations in Neural Network☆11Updated 2 years ago
- Towards Understanding Sharpness-Aware Minimization [ICML 2022]☆34Updated 2 years ago
- ☆14Updated 9 months ago
- [ICML 2021] "Efficient Lottery Ticket Finding: Less Data is More" by Zhenyu Zhang*, Xuxi Chen*, Tianlong Chen*, Zhangyang Wang☆25Updated 2 years ago
- ☆16Updated 2 years ago