Lightweight Adaptive ReduNet via Hyperspherical Manifold Learning
Organizations: School of Information Science and Technology, Southwest Jiaotong University, Chengdu 610031, China · Information Coding and Transmission Key Laboratory of Sichuan Province, CSNMT Int. Coop. Res. Centre (MoST), Southwest Jiaotong University, Chengdu 611756, China
Abstract
In recent years, a white-box neural network called ReduNet has been proposed, which employs the maximal coding rate reduction (MCR) principle to transform raw data into low-dimensional discriminative features via a forward layer-wise construction process. Unlike traditional deep networks that rely on backpropagation, ReduNet explicitly derives the parameters of each layer from the features of its preceding layer, offering a mathematically interpretable paradigm. However, this layer-wise construction often requires a large number of layers for the MCR objective to reach a stable value, which increases the parameter storage of the unfolded module. To address this issue, we propose LA-ReduNet, a lightweight adaptive architecture that refines the layer-wise update rule and enables discriminative feature representations to be obtained with substantially fewer unfolded layers. Specifically, LA-ReduNet employs hyperspherical manifold learning and adaptive step sizes, thereby reducing by an order of magnitude the number of layers required for the MCR objective to reach a stable value. Simulation results demonstrate that, while maintaining comparable classification accuracy, LA-ReduNet requires significantly fewer layers for the MCR objective to reach a stable value. Remarkably, under the considered experimental settings, LA-ReduNet requires only approximately of the parameter storage of the unfolded ReduNet module for the MCR objective to reach a stable value.
Figures & tables
| CIFAR-10 | CIFAR-100 | CINIC-10 | |
|---|---|---|---|
| Training set | 50000 | 50000 | 90000 |
| Test set | 10000 | 10000 | 90000 |
| Number of classes | 10 | 100 | 10 |
| Data dimension | 32 32 3 | 32 32 3 | 32 32 3 |
| Variant | Update direction | NT | Step-size parameter |
|---|---|---|---|
| Riemannian (Fixed) | No | ||
| Riemannian + NT (Fixed) | Yes | ||
| LA-ReduNet (Adaptive) | Yes | Eq. ( 21 ) |
| Weight coefficient | CIFAR-10 | CIFAR-100 | CINIC-10 |
|---|---|---|---|
| 44.71% | 9.28% | 30.13% | |
| 84.66% | 55.97% | 72.86% | |
| 87.58% | 62.78% | 76.79% | |
| 86.48% | 61.84% | 74.49% | |
| 86.18% | 61.67% | 75.20% |
Neural Network Compression
Neural network compression reduces model size, memory, communication, or computation while preserving useful accuracy.
See all 218 papers