mlsanityData to 14 Oct 2026

Scalability Analysis of Distributed Kolmogorov-Arnold Network Training on High-Performance Computing Systems

arXiv 2609.07740 ↗First posted 7 Sep 2026cs.DC cs.LG cs.PFcited by 0

Not on the map. Why?

Abstract (arXiv)

Kolmogorov-Arnold Networks (KANs) replace the fixed activation functions and linear weights of Multi-Layer Perceptrons (MLPs) with learnable univariate functions on network edges, offering improved interpretability and, in some settings, competitive parameter efficiency. While the approximation properties of KANs have received considerable attention, their behavior under distributed, multi-GPU training has not been systematically characterized. This paper presents an empirical scalability study of data-parallel KAN training on multi-node, multi-GPU high-performance computing (HPC) infrastructure, evaluated along four dimensions: strong scaling, weak scaling, communication overhead, and model-size scaling. Experiments were conducted on the FinisTerrae III supercomputer using up to 8 NVIDIA A100 GPUs across 4 nodes with PyTorch Distributed Data Parallel (DDP). KAN training reaches 74.7% parallel efficiency at 8 GPUs with a 5.97x speedup, consistent with conventional deep learning workloads. Weak scaling shows an initial single-to-multi-GPU throughput drop followed by strong stability. Communication overhead follows a non-monotonic pattern (1.3%-6.1%), driven primarily by All-Reduce algorithm selection and inter-node latency rather than KAN's edge-wise gradient structure. The parameter-to-memory ratio improves with model size even as training time scales unfavorably. These results indicate that operator-level and data-parallel optimizations for KAN are complementary. We provide deployment guidelines for GPU topology and model-size selection, and discuss the limitations of a synthetic-regression evaluation.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 8 of its 17 references that are in the map. First posted on arXiv 7 Sep 2026.

Builds on

8 of 17 references.

References · 8

  1. 2019Megatron-LM: Training Multi-Billion Parameter Language Models Using Model Parallelismcited by 1,868
  2. 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
  3. 2019ZeRO: Memory Optimizations Toward Training Trillion Parameter Modelscited by 1,301
  4. 2024KAN: Kolmogorov-Arnold Networkscited by 649
  5. 2017Revisiting Distributed Synchronous SGDcited by 142
  6. 2019MLPerf Training Benchmarkcited by 115
  7. 2025MatrixKAN: Parallelized Kolmogorov-Arnold Networkcited by 7
  8. 2025Concurrent training methods for Kolmogorov-Arnold networks: Disjoint datasets and FPGA implementationcited by 2

Built on by

Not cited yet.