mlsanityData to 14 Oct 2026

Batch Matrix-form Equations and Implementation of Multilayer Perceptrons

arXiv 2511.11918 ↗First posted 14 Nov 2025cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Multilayer perceptrons (MLPs) remain fundamental to modern deep learning, yet their algorithmic details are rarely presented in complete, explicit batch matrix-form. Rather, most references express gradients per sample or rely on automatic differentiation. Although automatic differentiation can achieve equally high computational efficiency, the usage of batch matrix-form makes the computational structure explicit, which is essential for transparent, systematic analysis, and optimization in settings such as sparse neural networks. This paper fills that gap by providing a mathematically rigorous and implementation-ready specification of MLPs in batch matrix-form. We derive forward and backward equations for all standard and advanced layers, including batch normalization and softmax, and validate all equations using the symbolic mathematics library SymPy. From these specifications, we construct uniform reference implementations in NumPy, PyTorch, JAX, TensorFlow, and a high-performance C++ backend optimized for sparse operations. Our main contributions are: (1) a complete derivation of batch matrix-form backpropagation for MLPs, (2) symbolic validation of all gradient equations, (3) uniform Python and C++ reference implementations grounded in a small set of matrix primitives, and (4) demonstration of how explicit formulations enable efficient sparse computation. Together, these results establish a validated, extensible foundation for understanding, teaching, and researching neural network algorithms.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 9 of its 29 references that are in the map. First posted on arXiv 14 Nov 2025.

Builds on

9 of 29 references.

References · 9

  1. 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
  2. 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
  3. 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,611
  4. 2015Automatic differentiation in machine learning: a surveycited by 855
  5. 2021Activation Functions in Deep Learning: A Comprehensive Survey and Benchmarkcited by 155
  6. 2023Ten Lessons We Have Learned in the New "Sparseland": A Short Handbook for Sparse Neural Network Researcherscited by 25
  7. 2019Demystifying Learning Rate Policies for High Accuracy Training of Deep Neural Networkscited by 12
  8. 2018On the Selection of Initialization and Activation Function for Deep Neural Networkscited by 11
  9. 2024Nerva: a Truly Sparse Implementation of Neural Networkscited by 2

Built on by

Not cited yet.