mlsanityData to 14 Oct 2026

Differentiable Entropy Regularization: A Complexity-Aware Approach for Neural Optimization

arXiv 2509.03733 ↗First posted 3 Sep 2025cs.LG cs.AIcited by 4

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

  1. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
  4. 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,974
  5. 2017Towards Deep Learning Models Resistant to Adversarial Attackscited by 6,415
  6. 2015Rethinking the Inception Architecture for Computer Visioncited by 5,475
  7. 2016PointNet: Deep Learning on Point Sets for 3D Classification and Segmentationcited by 5,248
  8. 2017PointNet++: Deep Hierarchical Feature Learning on Point Sets in a Metric Spacecited by 4,401
  9. 2022FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awarenesscited by 3,673
  10. 2020Longformer: The Long-Document Transformercited by 3,413
  11. 2023Mistral 7Bcited by 3,164
  12. 2015ShapeNet: An Information-Rich 3D Model Repositorycited by 2,897
21 more
  1. 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,877
  2. 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,739
  3. 2017Billion-scale similarity search with GPUscited by 2,497
  4. 2021Efficiently Modeling Long Sequences with Structured State Spacescited by 2,465
  5. 2019Scalability in Perception for Autonomous Driving: Waymo Open Datasetcited by 2,451
  6. 2013Sinkhorn Distances: Lightspeed Computation of Optimal Transportation Distancescited by 2,316
  7. 2020Transformers are RNNs: Fast Autoregressive Transformers with Linear Attentioncited by 2,022
  8. 2020Shortcut Learning in Deep Neural Networkscited by 1,969
  9. 2020Big Bird: Transformers for Longer Sequencescited by 1,684
  10. 2020Rethinking Attention with Performerscited by 1,601
  11. 2019Generating Long Sequences with Sparse Transformerscited by 1,546
  12. 2020Linformer: Self-Attention with Linear Complexitycited by 1,474
  13. 2019Energy and Policy Considerations for Deep Learning in NLPcited by 1,254
  14. 2017A Closer Look at Memorization in Deep Networkscited by 942
  15. 2015Deep Learning and the Information Bottleneck Principlecited by 802
  16. 2020Fast is better than free: Revisiting adversarial trainingcited by 727
  17. 2023Retentive Network: A Successor to Transformer for Large Language Modelscited by 613
  18. 2023Better Diffusion Models Further Improve Adversarial Trainingcited by 208
  19. 2020Algorithms with Predictionscited by 153
  20. 2019Stochastic Optimization of Sorting Networks via Continuous Relaxationscited by 111
  21. 2021AugMax: Adversarial Composition of Random Augmentations for Robust Trainingcited by 90

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 1; 2026: 3.

4 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2024An Information-Theoretic Regularizer for Lossy Neural Image Compression
  2. 2026PRISM: A Geometric Risk Bound for Decomposing Drift into Scale, Shape, and Head
  3. 2026Taming the Entropy Cliff: Variable Codebook Size Quantization for Autoregressive Visual Generation
  4. 2025Quantization-based Bounds on the Wasserstein Metric
  5. 2025AME: Aligned Manifold Entropy for Robust Vision-Language Distillation
  6. 2025Entropy Regularizing Activation: Boosting Continuous Control, Large Language Models, and Image Classification with Activation as Entropy Constraints
  7. 2026From Rigid to Dynamic: Entropy-Guided Adaptive Inference for Long-Context LLMs
  8. 2022On the pitfalls of entropy-based uncertainty for multi-class semi-supervised segmentation