Abstract:The great success of deep neural networks is built upon their over-parameterization, which smooths the optimization landscape without degrading the generalization ability. Despite the benefits of over-parameterization, a huge amount of parameters makes deep networks cumbersome in daily life applications. On the other hand, training neural networks without over-parameterization faces many practical problems, e.g., being trapped in the local optimal. Though techniques such as pruning and distillation are developed, they are expensive in fully training a dense network as backward selection methods; and there is still a void on systematically exploring forward selection methods for learning structural sparsity in deep networks. To fill in this gap, this paper proposes a new approach based on differential inclusions of inverse scale spaces. Specifically, our method can generate a family of models from simple to complex ones along the dynamics via coupling a pair of parameters, such that over-parameterized deep models and their structural sparsity can be explored simultaneously. This kind of differential inclusion scheme has a simple discretization, dubbed Deep structure splitting Linearized Bregman Iteration (DessiLBI), whose global convergence in learning deep networks could be established under the Kurdyka-Łojasiewicz framework. Particularly, we explore several applications of DessiLBI, including finding sparse structures of networks directly via the coupled structure parameter and growing networks from simple to complex ones progressively. Experimental evidence shows that our method achieves comparable and even better performance than the competitive optimizers in exploring the sparse structure of several widely used backbones on the benchmark datasets. Remarkably, with early stopping, our method unveils "winning tickets" in early epochs: the effective sparse network structures with comparable test accuracy to fully trained over-parameterized models, that are further transferable to similar alternative tasks. Furthermore, our method is able to grow networks efficiently with adaptive filter configurations, demonstrating the good performance with much less computational cost. Codes and models can be downloaded at https://github.com/DessiLBI2020/DessiLBI.

Sparsify Dynamically Expandable Network Via Variational Dropout

Bayesian compression for dynamically expandable networks

DSD$^2$: Can We Dodge Sparse Double Descent and Compress the Neural Network Worry-Free?

Always-Sparse Training by Growing Connections with Guided Stochastic Exploration

Spiking generative networks empowered by multiple dynamic experts for lifelong learning

Exploring Structural Sparsity of Deep Networks Via Inverse Scale Spaces

Subdomain contraction in deep networks for robust representation learning

Dynamic sparse coding-based value estimation network for deep reinforcement learning

Differentiable Sparsification for Deep Neural Networks

Dynamic Sparse Graph for Efficient Deep Learning.

Sparsity Winning Twice: Better Robust Generalization from More Efficient Training

Dep-$L_0$: Improving $L_0$-based Network Sparsification via Dependency Modeling

Robust Learning with Progressive Data Expansion Against Spurious Correlation

Sparse and Expandable Network for Google's Pathways

Bayesian Sparsification of Recurrent Neural Networks

A Survival Analysis Based Volatility and Sparsity Modeling Network for Student Dropout Prediction.

Pushing the Limits of Sparsity: A Bag of Tricks for Extreme Pruning

Divergent Ensemble Networks: Enhancing Uncertainty Estimation with Shared Representations and Independent Branching

Efficient Marginalization of Discrete and Structured Latent Variables via Sparsity

Learning Sparse Networks Using Targeted Dropout

Sparsity and Heterogeneous Dropout for Continual Learning in the Null Space of Neural Activations