arXiv · 2412.15077
Till the Layers Collapse: Compressing a Deep Neural Network through the Lenses of Batch Normalization Layers
Abstract
Today, deep neural networks are widely used since they can handle a variety of complex tasks. Their generality makes them very powerful tools in modern technology. However, deep neural networks are often overparameterized. The usage of these large models consumes a lot of computation resources. In this paper, we introduce a method called \textbf{T}ill the \textbf{L}ayers \textbf{C}ollapse (TLC), which compresses deep neural networks through the lenses of batch normalization layers. By reducing the depth of these networks, our method decreases deep neural networks' computational requirements and overall latency. We validate our method on popular models such as Swin-T, MobileNet-V2, and RoBERTa, across both image classification and natural language processing (NLP) tasks.
Explore related subjects
Keep this discovery
Explore connections, maps & timelines
Zhu Liao, Nour Hezbri, Victor Quétu, Van-Tam Nguyen, Enzo Tartaglione. 2024-12-19. Till the Layers Collapse: Compressing a Deep Neural Network through the Lenses of Batch Normalization Layers. https://arxiv.org/abs/2412.15077
Cite the original work for its findings. Save a collection to share your selection of sources.