Differentiable Entropy Regularization: A Complexity-Aware Approach for Neural Optimization
Not on the map. Why?
Abstract (arXiv)
We introduce the first differentiable approximation of range-partition entropy, a complexity measure from computational geometry that directly bounds algorithmic runtime. Unlike architectural modifications, our method is a complementary regularizer that provides orthogonal efficiency gains when combined with existing optimizations. We establish theoretical guarantees in computational geometry, achieving 4--5× provable speedups on convex hull and triangulation with <0.2% error. On ImageNet-1K with ViT-Base, entropy regularization achieves 80.1% top-1 accuracy at 80% sparsity (1.60× standalone speedup), and when combined with FlashAttention yields 2.07× speedup versus 1.63× for FlashAttention alone. On large language models (LLaMA-2 7B, Mistral-7B, Phi-2), we achieve 1.48--1.60× inference speedups at 70--75% sparsity with minimal quality degradation (ROUGE-L drops of 0.3--0.4 points, perplexity increase of 0.9). Unlike prior regularization methods that target output distributions, we directly minimize representation complexity, yielding both efficiency gains and improved robustness through semantically structured sparsity patterns (IoU 0.73 vs 0.41 for magnitude pruning, CIFAR-100-C mCE 48.7 vs 55.4). Benefits are strongest for geometry and vision transformers, with more modest but measurable gains on LLMs, demonstrating that complexity regularization offers a principled pathway to joint efficiency-robustness optimization.
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 33 of its 43 references that are in the map. First posted on arXiv 3 Sep 2025.
Builds on
33 of 43 references.
References · 33
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
- 2017Towards Deep Learning Models Resistant to Adversarial Attackscited by 6,415
- 2015Rethinking the Inception Architecture for Computer Visioncited by 5,475
- 2016PointNet: Deep Learning on Point Sets for 3D Classification and Segmentationcited by 5,248
- 2017PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Spacecited by 4,401
- 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
- 2020Longformer: The Long-Document Transformercited by 3,413
- 2023Mistral 7Bcited by 3,164
- 2015ShapeNet: An Information-Rich 3D Model Repositorycited by 2,897
21 more
- 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,877
- 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,739
- 2017Billion-scale similarity search with GPUscited by 2,497
- 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
- 2019Scalability in Perception for Autonomous Driving: Waymo Open Datasetcited by 2,451
- 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,316
- 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
- 2020Shortcut Learning in Deep Neural Networkscited by 1,969
- 2020Big Bird: Transformers for Longer Sequencescited by 1,684
- 2020Rethinking Attention with Performerscited by 1,601
- 2019Generating Long Sequences with Sparse Transformerscited by 1,546
- 2020Linformer: Self-Attention with Linear Complexitycited by 1,474
- 2019Energy and Policy Considerations for Deep Learning in NLPcited by 1,254
- 2017A Closer Look at Memorization in Deep Networkscited by 942
- 2015Deep Learning and the Information Bottleneck Principlecited by 802
- 2020Fast is better than free: Revisiting adversarial trainingcited by 727
- 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 613
- 2023Better Diffusion Models Further Improve Adversarial Trainingcited by 208
- 2020Algorithms with Predictionscited by 153
- 2019Stochastic Optimization of Sorting Networks via Continuous Relaxationscited by 111
- 2021AugMax: Adversarial Composition of Random Augmentations for Robust Trainingcited by 90
Built on by
Citing papers per year: 2025: 1; 2026: 3.
4 papers.
- 2026Beyond Variance: Knowledge-Aware LLM Compression via Fisher-Aligned Subspace Diagnosticscited by 4
- 2026Grassmannian Mixture-of-Experts: Concentration-Controlled Routing on Subspace Manifoldscited by 2
- 2025Adaptive Constraint Propagation: Scaling Structured Inference for Large Language Models via Meta-Reinforcement Learningcited by 0
- 2026Calibrated Adaptation: Bayesian Stiefel Manifold Priors for Reliable Parameter-Efficient Fine-Tuningcited by 0
- 2026Calibrated Adaptation: Bayesian Stiefel Manifold Priors for Reliable Parameter-Efficient Fine-Tuning19 Feb 2026
- 2026Grassmannian Mixture-of-Experts: Concentration-Controlled Routing on Subspace Manifolds19 Feb 2026
- 2026Beyond Variance: Knowledge-Aware LLM Compression via Fisher-Aligned Subspace Diagnostics12 Jan 2026
- 2025Adaptive Constraint Propagation: Scaling Structured Inference for Large Language Models via Meta-Reinforcement Learning31 Dec 2025
Similar, not cited
Close in wording, no citation either way
- 2024An Information-Theoretic Regularizer for Lossy Neural Image Compression
- 2026PRISM: A Geometric Risk Bound for Decomposing Drift into Scale, Shape, and Head
- 2026Taming the Entropy Cliff: Variable Codebook Size Quantization for Autoregressive Visual Generation
- 2025Quantization-based Bounds on the Wasserstein Metric
- 2025AME: Aligned Manifold Entropy for Robust Vision-Language Distillation
- 2025Entropy Regularizing Activation: Boosting Continuous Control, Large Language Models, and Image Classification with Activation as Entropy Constraints
- 2026From Rigid to Dynamic: Entropy-Guided Adaptive Inference for Long-Context LLMs
- 2022On the pitfalls of entropy-based uncertainty for multi-class semi-supervised segmentation