English

GPU Parallelization Strategies for Forward and Backward Propagation in Shallow Neural Networks: A CUDA-Based Comparative Study

Distributed, Parallel, and Cluster Computing 2026-06-29 v1 Machine Learning

Abstract

We present a comparative study of CUDA optimization strategies applied to forward and backward propagation in a shallow neural network. Three stacked optimizations are evaluated: (1) tiled shared memory with bank-conflict elimination via +1-column padding, (2) pre-transposed weight matrices for coalesced global memory access, and (3) a fused MatMul+ReLU kernel that eliminates intermediate global-memory round-trips. Experiments on an NVIDIA Tesla T4 (CUDA 13.0) across three dataset sizes show that the fully optimized implementation achieves a 1.41x speedup over the baseline CUDA version on the large dataset (25,600 samples), reducing execution time from 21.0s to 14.8s. Results are compared against a sequential CPU baseline and an OpenMP parallel implementation, demonstrating the effectiveness of memory-access optimization in GPU-accelerated deep learning primitives.

Cite

@article{arxiv.2606.30497,
  title  = {GPU Parallelization Strategies for Forward and Backward Propagation in Shallow Neural Networks: A CUDA-Based Comparative Study},
  author = {Rania Zitouni and Nadine Bousdjira and Sarah Hasnaoui and Amel Sadoun and Fatma Salhi},
  journal= {arXiv preprint arXiv:2606.30497},
  year   = {2026}
}

Comments

7 pages, 5 figures. Technical report, ESI Algiers, 2025--2026