ReduNet: A White-box Deep Network from the Principle of Maximizing Rate Reduction
Not on the map. Why?
Abstract (arXiv)
This work attempts to provide a plausible theoretical framework that aims to interpret modern deep (convolutional) networks from the principles of data compression and discriminative representation. We argue that for high-dimensional multi-class data, the optimal linear discriminative representation maximizes the coding rate difference between the whole dataset and the average of all the subsets. We show that the basic iterative gradient ascent scheme for optimizing the rate reduction objective naturally leads to a multi-layer deep network, named ReduNet, which shares common characteristics of modern deep networks. The deep layered architectures, linear and nonlinear operators, and even parameters of the network are all explicitly constructed layer-by-layer via forward propagation, although they are amenable to fine-tuning via back propagation. All components of so-obtained "white-box" network have precise optimization, statistical, and geometric interpretation. Moreover, all linear operators of the so-derived network naturally become multi-channel convolutions when we enforce classification to be rigorously shift-invariant. The derivation in the invariant setting suggests a trade-off between sparsity and invariance, and also indicates that such a deep convolution network is significantly more efficient to construct and learn in the spectral domain. Our preliminary simulations and experiments clearly verify the effectiveness of both the rate reduction objective and the associated ReduNet. All code and data are available at urlhttps://github.com/Ma-Lab-Berkeley.
Key facts. Cited by 84 papers in the map (data through 7 Oct 2026). Builds on 37 of its 175 references that are in the map. First posted on arXiv 21 May 2021.
Builds on
37 of 175 references.
References · 37
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2016Layer Normalizationcited by 5,579
- 2013Intriguing properties of neural networkscited by 5,114
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,014
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2018Spectral Normalization for Generative Adversarial Networkscited by 1,618
- 2016Neural Architecture Search with Reinforcement Learningcited by 1,489
- 2018Learning deep representations by mutual information estimation and maximizationcited by 1,265
- 2016Instance Normalization: The Missing Ingredient for Fast Stylizationcited by 1,187
- 2016Group Equivariant Convolutional Networkscited by 1,154
- 2018Reconciling modern machine learning practice and the bias-variance trade-offcited by 919
25 more
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
- 2017Dynamic Routing Between Capsulescited by 689
- 2015Empirical Evaluation of Rectified Activations in Convolutional Networkcited by 515
- 2019Algorithm Unrolling: Interpretable, Efficient Deep Learning for Signal and Image Processingcited by 411
- 2019The generalization error of random features regression: Precise asymptotics and double descent curvecited by 410
- 2017Size-Independent Sample Complexity of Neural Networkscited by 289
- 2016Designing Neural Network Architectures using Reinforcement Learningcited by 277
- 2015Highway Networkscited by 257
- 2018Why do deep convolutional networks generalize so poorly to small image transformations?cited by 198
- 2021Exploring Deep Neural Networks via Layer-Peeled Model: Minority Collapse in Imbalanced Trainingcited by 166
- 2018Activation Functions: Comparison of trends in Practice and Research for Deep Learningcited by 130
- 2017Universal Function Approximation by Deep Neural Nets with Bounded Width and ReLU Activationscited by 104
- 2020On the Optimal Weighted ℓ2 Regularization in Overparameterized Linear Regressioncited by 97
- 2017Accelerated Gradient Descent Escapes Saddle Points Faster than Gradient Descentcited by 66
- 2017The power of deeper networks for expressing natural functionscited by 35
- 2020Deep Networks and the Multiple Manifold Problemcited by 35
- 2018Caveats for information bottleneck in deterministic scenarioscited by 34
- 2014Fast Convolutional Nets With fbfft: A GPU Performance Evaluationcited by 27
- 2019A Rate-Distortion Framework for Explaining Neural Network Decisionscited by 24
- 2019Neural Collaborative Subspace Clusteringcited by 21
- 2019Asymptotic learning curves of kernel methods: empirical data v.s. Teacher-Student paradigmcited by 20
- 2021Convolutional Normalization: Improving Deep Convolutional Network Robustness and Trainingcited by 15
- 2016Interpretable Recurrent Neural Networks Using Sequential Sparse Recoverycited by 8
- 2017Deep Sparse Subspace Clusteringcited by 7
- 2015LogDet Rank Minimization with Application to Subspace Clusteringcited by 6
Built on by
Citing papers per year: 2021: 7; 2022: 18; 2023: 18; 2024: 20; 2025: 12; 2026: 9.
84 papers.
Sort:
By area:
- 2022Discovering Invariant Rationales for Graph Neural Networkscited by 186
- 2022Neural Collapse: A Review on Modelling Principles and Generalizationcited by 95
- 2022Robust Training under Label Noise by Over-parameterizationcited by 93
- 2023White-Box Transformers via Sparse Rate Reductioncited by 77
- 2023D2 Pruning: Message Passing for Balancing Diversity and Difficulty in Data Pruningcited by 68
- 2021How Powerful is Graph Convolution for Recommendation?cited by 50
- 2022On the Principles of Parsimony and Self-Consistency for the Emergence of Intelligencecited by 50
- 2022Neural Collapse with Normalized Features: A Geometric Analysis over the Riemannian Manifoldcited by 50
- 2023Linear Spaces of Meanings: Compositional Structures in Vision-Language Modelscited by 48
- 2023Generalized Neural Collapse for a Large Number of Classescited by 39
- 2022Green Learning: Introduction, Examples and Outlookcited by 30
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Modelscited by 29
38 more
- 2022Understanding and Improving Transfer Learning of Deep Models via Neural Collapsecited by 24
- 2023Emergence of Segmentation with Minimalistic White-Box Transformerscited by 23
- 2023Understanding Augmentation-based Self-Supervised Representation Learning via RKHS Approximation and Regressioncited by 20
- 2021Closed-Loop Data Transcription to an LDR via Minimaxing Rate Reductioncited by 17
- 2022Incremental Learning of Structured Memory via Closed-Loop Transcriptioncited by 17
- 2021MAE-DET: Revisiting Maximum Entropy Principle in Zero-Shot NAS for Efficient Object Detectioncited by 16
- 2021Learning High-Dimensional Parametric Maps via Reduced Basis Adaptive Residual Networkscited by 16
- 2022Learning Fair Representations via Rate-Distortion Maximizationcited by 15
- 2024A Law of Next-Token Prediction in Large Language Modelscited by 15
- 2023DeepMAD: Mathematical Architecture Design for Deep Convolutional Neural Networkcited by 14
- 2023Neural Collapse in Multi-label Learning with Pick-all-label Losscited by 14
- 2025Data Pruning by Information Maximizationcited by 14
- 2023Representation Learning via Manifold Flattening and Reconstructioncited by 11
- 2023Data Representations' Study of Latent Image Manifoldscited by 11
- 2023Towards Task Sampler Learning for Meta-Learningcited by 11
- 2024Speculative Coreset Selection for Task-Specific Fine-tuningcited by 10
- 2024Modeling Spatio-temporal Dynamical Systems with Neural Discrete Learning and Levels-of-Expertscited by 9
- 2024Masked Completion via Structured Diffusion with White-Box Transformerscited by 9
- 2022Unsupervised Learning of Structured Representations via Closed-Loop Transcriptioncited by 7
- 2023Threshold-Consistent Margin Loss for Open-World Deep Metric Learningcited by 7
- 2022Minimalistic Unsupervised Learning with the Sparse Manifold Transformcited by 6
- 2023Data Optimization in Deep Learning: A Surveycited by 6
- 2022Sustaining Fairness via Incremental Learningcited by 5
- 2022Unsupervised Opinion Summarization Using Approximate Geodesicscited by 5
- 2022Extracting Semantic Knowledge from GANs with Unsupervised Learningcited by 5
- 2023Maximizing Spatio-Temporal Entropy of Deep 3D CNNs for Efficient Video Recognitioncited by 5
- 2023Achieving Robust Generalization for Wireless Channel Estimation Neural Networks by Designed Training Datacited by 4
- 2023Closed-Loop Transcription via Convolutional Sparse Codingcited by 4
- 2021Envisioning Future Deep Learning Theories: Some Basic Concepts and Characteristicscited by 3
- 2022Bridging Model-based Safety and Model-free Reinforcement Learning through System Identification of Low Dimensional Linear Modelscited by 3
- 2022UGformer for Robust Left Atrium and Scar Segmentation Across Scannerscited by 3
- 2024Merino: Entropy-driven Design for Generative Language Models on IoT Devicescited by 3
- 2024LightFF: Lightweight Inference for Forward-Forward Algorithmcited by 3
- 2024Denoising Pre-Training and Customized Prompt Learning for Efficient Multi-Behavior Sequential Recommendationcited by 3
- 2024Rethinking Decoders for Transformer-based Semantic Segmentation: A Compression Perspectivecited by 3
- 2025Geometry of Neural Reinforcement Learning in Continuous State and Action Spacescited by 3
- 2025Transformers as Intrinsic Optimizers: Forward Inference through the Energy Principlecited by 3
- 2022Augmentation Invariant Manifold Learningcited by 2
50 most cited of 84 All 84 (JSON)
- 2026MILO: Efficient Many-shot In-Context Learning with Block-wise Low-rank Compression24 Sep 2026
- 2026On the Limits of Maximal Coding Rate Reduction for Out-of-Distribution Generalisation17 Sep 2026
- 2026Riemannian Archetypal Analysis: Interpretable non-linear data analysis on deformed star distributions22 May 2026
- 2026How Label Imbalance Shapes Geometry: A General Spectral Analysis of Multi-Label Neural Collapse3 May 2026
- 2026Operational Feature Fingerprints of Graph Datasets via a White-Box Signal-Subspace Probe24 Apr 2026
- 2026Robust Principal Component Completion26 Mar 2026
- 2026FedCova: Robust Federated Covariance Learning Against Noisy Labels4 Mar 2026
- 2026PRISM: Deriving a White-Box Transformer as a Signal-Noise Decomposition Operator via Maximum Coding Rate Reduction21 Jan 2026
- 2026Interpretable and Sparse Linear Attention with Decoupled Membership-Subspace Modeling via MCR2 Objective20 Jan 2026
- 2025How Wide and How Deep? Mitigating Over-Squashing of GNNs via Channel Capacity Constrained Estimation9 Nov 2025
- 2025Transformers as Intrinsic Optimizers: Forward Inference through the Energy Principle2 Nov 2025
- 2025Scaling Non-Parametric Sampling with Representation25 Oct 2025
38 more
- 2025An Empirical Study of Sample Selection Strategies for Large Language Model Repair23 Oct 2025
- 2025Knowledge-Informed Neural Network for Complex-Valued SAR Image Recognition23 Oct 2025
- 2025Towards Interpretable and Efficient Attention: Compressing All by Contracting a Few21 Sep 2025
- 2025Rethinking Purity and Diversity in Multi-Behavior Sequential Recommendation from the Frequency Perspective28 Aug 2025
- 2025Towards White-Box Deep Wireless Sensing29 Jul 2025
- 2025Geometry of Neural Reinforcement Learning in Continuous State and Action Spaces28 Jul 2025
- 2025Data Pruning by Information Maximization2 Jun 2025
- 2025SCORE: Soft Label Compression-Centric Dataset Condensation via Coding Rate Optimization18 Mar 2025
- 2025Beyond External Monitors: Enhancing Transparency of Large Language Models for Easier Monitoring7 Feb 2025
- 2024LoLaFL: Low-Latency Federated Learning via Forward-only Propagation19 Dec 2024
- 2024ESS-ReduNet: Enhancing Subspace Separability of ReduNet via Dynamic Expansion with Bayesian Inference27 Nov 2024
- 2024White-Box Diffusion Transformer for single-cell RNA-seq generation11 Nov 2024
- 2024Rethinking Decoders for Transformer-based Semantic Segmentation: A Compression Perspective5 Nov 2024
- 2024Improving Neuron-level Interpretability with White-box Language Models21 Oct 2024
- 2024EMWaveNet: Physically Explainable Neural Network Based on Electromagnetic Propagation for SAR Target Recognition13 Oct 2024
- 2024Dense Optimizer : An Information Entropy-Guided Structural Search Method for Dense-like Neural Network Design10 Oct 2024
- 2024Speculative Coreset Selection for Task-Specific Fine-tuning2 Oct 2024
- 2024On the Generalization and Causal Explanation in Self-Supervised Learning1 Oct 2024
- 2024CVVLSNet: Vehicle Location and Speed Estimation Using Partial Connected Vehicle Trajectory Data30 Sep 2024
- 2024A Law of Next-Token Prediction in Large Language Models24 Aug 2024
- 2024Denoising Pre-Training and Customized Prompt Learning for Efficient Multi-Behavior Sequential Recommendation21 Aug 2024
- 2024Unleashing the Power of Data Tsunami: A Comprehensive Survey on Data Assessment and Selection for Instruction Tuning of Language Models4 Aug 2024
- 2024Spring-block theory of feature learning in deep neural networks28 Jul 2024
- 2024JSCDS: A Core Data Selection Method with Jason-Shannon Divergence for Caries RGB Images-Efficient Learning29 Jun 2024
- 2024Geometric Understanding of Discriminability and Transferability for Visual Domain Adaptation24 Jun 2024
- 2024LightFF: Lightweight Inference for Forward-Forward Algorithm8 Apr 2024
- 2024Masked Completion via Structured Diffusion with White-Box Transformers3 Apr 2024
- 2024Merino: Entropy-driven Design for Generative Language Models on IoT Devices28 Feb 2024
- 2024Modeling Spatio-temporal Dynamical Systems with Neural Discrete Learning and Levels-of-Experts6 Feb 2024
- 2023Data Optimization in Deep Learning: A Survey25 Oct 2023
- 2023Neural Collapse in Multi-label Learning with Pick-all-label Loss24 Oct 2023
- 2023D2 Pruning: Message Passing for Balancing Diversity and Difficulty in Data Pruning11 Oct 2023
- 2023Generalized Neural Collapse for a Large Number of Classes9 Oct 2023
- 2023Emergence of Segmentation with Minimalistic White-Box Transformers30 Aug 2023
- 2023Towards Task Sampler Learning for Meta-Learning18 Jul 2023
- 2023Threshold-Consistent Margin Loss for Open-World Deep Metric Learning8 Jul 2023
- 2023White-Box Transformers via Sparse Rate Reduction1 Jun 2023
- 2023Understanding Augmentation-based Self-Supervised Representation Learning via RKHS Approximation and Regression1 Jun 2023
50 newest of 84 All 84 (JSON)
- 2022Discovering Invariant Rationales for Graph Neural Networkscited by 186
- 2022Neural Collapse: A Review on Modelling Principles and Generalizationcited by 95
- 2022Robust Training under Label Noise by Over-parameterizationcited by 93
- 2023White-Box Transformers via Sparse Rate Reductioncited by 77
- 2023D2 Pruning: Message Passing for Balancing Diversity and Difficulty in Data Pruningcited by 68
- 2021How Powerful is Graph Convolution for Recommendation?cited by 50
- 2022Neural Collapse with Normalized Features: A Geometric Analysis over the Riemannian Manifoldcited by 50
- 2023Generalized Neural Collapse for a Large Number of Classescited by 39
- 2022Green Learning: Introduction, Examples and Outlookcited by 30
- 2022Understanding and Improving Transfer Learning of Deep Models via Neural Collapsecited by 24
- 2023Emergence of Segmentation with Minimalistic White-Box Transformerscited by 23
- 2023Understanding Augmentation-based Self-Supervised Representation Learning via RKHS Approximation and Regressioncited by 20
12 most cited of 79 All 79 (JSON)
- 2026On the Limits of Maximal Coding Rate Reduction for Out-of-Distribution Generalisation17 Sep 2026
- 2026Riemannian Archetypal Analysis: Interpretable non-linear data analysis on deformed star distributions22 May 2026
- 2026How Label Imbalance Shapes Geometry: A General Spectral Analysis of Multi-Label Neural Collapse3 May 2026
- 2026Operational Feature Fingerprints of Graph Datasets via a White-Box Signal-Subspace Probe24 Apr 2026
- 2026Robust Principal Component Completion26 Mar 2026
- 2026FedCova: Robust Federated Covariance Learning Against Noisy Labels4 Mar 2026
- 2026PRISM: Deriving a White-Box Transformer as a Signal-Noise Decomposition Operator via Maximum Coding Rate Reduction21 Jan 2026
- 2026Interpretable and Sparse Linear Attention with Decoupled Membership-Subspace Modeling via MCR2 Objective20 Jan 2026
- 2025How Wide and How Deep? Mitigating Over-Squashing of GNNs via Channel Capacity Constrained Estimation9 Nov 2025
- 2025Transformers as Intrinsic Optimizers: Forward Inference through the Energy Principle2 Nov 2025
- 2025Scaling Non-Parametric Sampling with Representation25 Oct 2025
- 2025An Empirical Study of Sample Selection Strategies for Large Language Model Repair23 Oct 2025
12 newest of 79 All 79 (JSON)