Big2Small: A Unifying Neural Network Framework for Model Compression
Authors: Jing-Xiao Liao, Haoran Wang, Tao Li, Daoming Lyu, Yi Zhang, Chengjun Cai, Feng-Lei Fan
Organizations: Frontier of Artificial Networks (FAN) Lab, Department of Data Science, City University of Hong Kong, Kowloon, Hong Kong SAR, China · Hong Kong Applied Science and Technology Research Institute (ASTRI) · Sichuan Engineering Research Center for Big Data Visual Analytics, Sichuan University of Science and Engineering, Zigong, China · School of Cyber Science and Engineering, Sichuan University, Chengdu, China · Key Laboratory of Data Protection and Intelligent Management, Ministry of Education, Sichuan University, Chengdu, China · Computer Science and Information Technology Center, City University of Hong Kong (Dongguan), Dongguan, China
With the development of foundational models, model compression has become a critical requirement. Various model compression approaches have been proposed such as low-rank decomposition, pruning, quantization, ergodic dynamic systems, and knowledge distillation, which are based on different heuristics. To elevate the field from fragmentation to a principled discipline, we construct a unifying mathematical framework for model compression grounded in measure theory. We further demonstrate that each model compression technique is mathematically equivalent to a neural network subject to a regularization. Building upon this mathematical and structural equivalence, we propose an experimentally-verified data-free model compression framework, termed Big2Small, which translates Implicit Neural Representations (INRs) from data domain to the domain of network parameters. Big2Small trains compact INRs to encode the weights of larger models and reconstruct the weights during inference. To enhance reconstruction fidelity, we introduce Outlier-Aware Preprocessing to handle extreme weight values and a Frequency-Aware Loss function to preserve high-frequency details. Experiments on image classification and segmentation demonstrate that Big2Small achieves competitive accuracy and compression ratios compared to state-of-the-art baselines.
Figures & tables
Feature
Low-Rank Decomposition
Quantization
Pruning
Ergodic Dynamic Systems
Knowledge Distillation
Primary Goal
Matrix Factorization
Precision Reduction
Redundancy Removal
Trajectory Representation
Behavior Mimicry
Sparsity Type
Structured
Dense
Unstructured or Structured
Continuous
Dense
Core Operations
SVD, CP, Tucker
Integer Mapping
Sparsity Induction
Ergodic Hyperfunctions
Soft Target Matching
Data Reliance
Low (Often Data-free)
Moderate (Calibration)
High (Fine-tuning)
Low (Direct Encoding)
Very High (Teacher Output)
TABLE I: Properties of model compression methods.
Fig. 1: The theoretical framework of Theorem 1.
Fig. 2: The framework of Theorem 2.
Fig. 3: An overview of the proposed Big2Small framework. It uses a “Compression–Decompression” architecture that encodes discrete weight parameter tensors with lightweight INRs and reconstructs the original weights at inference time.
Fig. 4: The structure of INR.
Size (MB)
W/A Bits
Compression Ratio
Top1 Acc (%)
ResNet18 [ 63 ]
44.6
32/32
-
71.47
DSG [ 68 ]
8.4
6/6
5.3
70.46
Squant [ 69 ]
8.4
6/6
5.3
70.74
UDFC [ 30 ]
8.4
6/6
5.3
72.76
RieM [ 56 ]
8.4
8/16
5.3
71.80
Big2Small-32bit
30.4
32/32
1.5
71.85
TABLE II: Comparison of different compression methods for three notable image recognition networks on the ImageNet dataset. W/A means the numbers of weight/activation bits.
Fig. 5: Visualizing segmentation results of the original UNet models and the compressed ones.
Size (MB)
W/A bits
mACC(%)
mIOU(%)
UNet [ 62 ]
51.5
32/32
99.52
95.87
RieM [ 56 ]
25.8
16/16
98.52
94.67
Big2Small-32bit
22.9
32/32
99.07
95.32
Big2Small-16bit
11.5
16/16
98.75
94.86
UNet++ [ 73 ]
9.2
32/32
99.01
95.62
RieM [ 56 ]
4.6
16/16
97.15
90.35
TABLE III: Comparison of different compression methods on the Carvana dataset. Where W/A means the numbers of weight/activation bits.
Fig. 6: Comparison of weight reconstruction qualities on the convolutional layer of ResNet50.
Fig. 7: Reconstruction error vs compression ratios on the convolutional layer of ResNet50.
Layer
Strategy
Rel MAE
Cosine
PSNR
w/o pre
0.0832
0.953
29.32
pre
0.0085
1.000
54.21
MSE
0.0672
0.996
34.29
MSE+Grad
0.0421
0.996
35.21
MSE+Freq
0.0351
0.996
35.52
Convolution
Ours
0.0085
1.000
54.21
TABLE IV: Ablation study of the Big2Small using ResNet50. “w/o pr” denotes without using our preprocessing strategy; “Rel MAE” is relative MAE; “Cosine” represents cosine similarity.
Fig. 8: The Q-Q plot of using different loss functions to reconstruct the weight of the linear layer.
Size (MB)
Compression Ratio
Top1 Acc (%)
ResNet18
44.59
-
71.47
Big2Small
30.4
1.5
71.85
Big2Small+SVD
10.32
2.9
70.32
Big2Small+P (60%)
9.12
4.9
69.82
Big2Small+Q (6bit)
7.59
5.9
71.24
TABLE V: Compressing Big2Small on the ImageNet dataset. Where P denotes pruning and Q denotes quantization.
Fig. 9: The throughput of raw and Big2Small models on the ImageNet test set.
Fig. 10: Comparison of different encoders for compressing a convolution weight of ResNet50.
The escalating size of pretrained neural networks has rendered model compression a prerequisite for deployment under stringent memory and compute constraints. With the irrational winding as an example, earlier work introduced a dynamic system (DS) paradigm that reconceptualizes compression as compact weight representation: high-dimensional parameters are encoded by the index of a trajectory produced by a dynamic system, from which the vector is recovered during decompression. This mechanism is fundamentally distinct from pruning, quantization, knowledge distillation, and low-rank decomposition. Along this direction, we prove that under a Diophantine condition, a finite trajectory of M=O(ε−(d+ν)) states in the irrational winding constitutes an ε-net over the d-dimensional weight space, thereby linking state resolution, decompression error, and compression ratio in a predictable manner. Furthermore, we propose a generalized DS-based model compression framework by unifying four DS families---space-filling curves (Hilbert, Peano, Morton/Z-order, Snake), chaotic systems (Lorenz), congruential and pseudo-random generators (LCG, PCG), and low-discrepancy sequences (Halton). Also, we introduce the KD-tree and coordinate-template acceleration to scale to large models as well as outlier identification to control the error. Experiments on ResNet-18 and Qwen2.5-1.5B/Qwen1.5-7B validate that DS-based compression achieves competitive compression ratios without post-hoc retraining, with controllable decompression error and flexible state-space design, establishing it as a principled and practical compression approach.
Fan Gao, Wei Su, Juntong Fan +4
School of Information Science and Engineering, Lanzhou University, Lanzhou, Gansu Province, China · Department of Data Science, City University of Hong Kong, Hong Kong, China SAR · Department of Mathematics, City University of Hong Kong, Hong Kong, China SAR +1
Neural network compression is commonly achieved by pruning parameters based on local importance scores, e.g., magnitude-based pruning. We propose a complementary approach that compresses models by aggregating neurons with similar functional behavior rather than removing weights independently. Our method encodes a trained network as a polynomial ODE system and applies a lumping method called Approximate Forward Differential Equivalence to identify neurons with approximately matching induced dynamics. A single tolerance parameter, ε, controls the compression level and induces a smooth trade-off between model size and predictive accuracy. We evaluate the method on synthetic datasets derived from nonlinear dynamical systems with known ground-truth behavior and on public regression benchmarks. Across both settings, the proposed approach achieves substantial parameter reduction while preserving accuracy, and consistently compares favorably with magnitude-based pruning and Wanda at similar compression levels. These results suggest that differential equivalence-based aggregation is a principled and effective alternative to conventional weight-centric pruning.
Ravi Dhiman, Andrea Passarella, Mirco Tribastone +1
IMT School for Advanced Studies Lucca 55100, Italy · IIT CNR, Pisa 56124, Italy
Compressing neural networks without retraining is vital for deployment at scale. We study calibration-free compression through the lens of projection geometry: structured pruning is an axis-aligned projection, whereas model folding performs a low-rank projection via weight clustering. We formalize both as orthogonal operators and show that, within a rank distance of one, folding provably yields smaller parameter reconstruction error, and under mild smoothness assumptions, smaller functional perturbations than pruning. At scale, we evaluate >1000 checkpoints spanning ResNet18, PreActResNet18, ViT-B/32, and CLIP ViT-B/32 on CIFAR-10 and ImageNet-1K, covering diverse training hyperparameters (optimizers, learning rates, augmentations, regularization, sharpness-aware training), as well as multiple LLaMA-family 60M and 130M parameter models trained on C4. We show that folding typically achieves higher post-compression accuracy, with the largest gains at moderate-high compression. The gap narrows and occasionally reverses at specific training setups. Our results position folding as a geometry-aware, calibration-free alternative to pruning that is often superior in practice and principled in theory.
Olga Saukh, Dong Wang, Haris Šikić +2
Graz University of Technology · Complexity Science Hub · Swiss Data Science Center +1