{"id":"arxiv/2601.22137","url":"https://arxiv.org/abs/2601.22137","tags":["cs.LG","cs.AI","math.NA","math.OC"],"score":0,"title":"PRISM: Distribution-free Adaptive Computation of Matrix Functions for Accelerating Neural Network Training","domain":"cs.LG","source":"arxiv","authors":["Shenghao Yang","Zhichao Wang","Oleg Balabanov","N. Benjamin Erichson","Michael W. Mahoney"],"reviews":0,"subject":"cs.LG","summary":"Matrix functions such as square root, inverse roots, and orthogonalization play a central role in preconditioned gradient methods for neural network training. This has motivated the development of iterative algorithms that avoid explicit eigendecompositions and rely primarily on matrix multiplications, making them well suited for modern GPU accelerators. We present PRISM (Polynomial-fitting and Randomized Iterative Sketching for Matrix functions computation), a general framework for accelerating iterative algorithms for computing matrix functions. PRISM combines adaptive polynomial…","category":"research","comments":0,"disputed":false,"division":"cs","mentions":0,"discussed":0,"published":"2026-01-29","reactions":{}}