• Home
  • Search
  • Distributed-Memory Sparse Kernels for Machine Learning
  • Open Access IconOpen Access
  • Cite Icon11
  • https://doi.org/10.1109/ipdps53621.2022.00014Copy DOI Icon

Distributed-Memory Sparse Kernels for Machine Learning

  • May 1, 2022
  • Vivek Bharadwaj +2 more
Show More
  • Abstract
  • Literature Map
  • References
  • Citations
  • Similar Papers
Abstract

Sampled Dense Times Dense Matrix Multiplication (SDDMM) and Sparse Times Dense Matrix Multiplication (SpMM) appear in diverse settings, such as collaborative filtering, document clustering, and graph embedding. Frequently, the SDDMM output becomes the input sparse matrix for a sub-sequent SpMM operation. Existing work has focused on shared memory parallelization of these primitives. While there has been extensive analysis of communication-minimizing distributed 1.5D algorithms for SpMM, no such analysis exists for SDDMM or the back-to-back sequence of SDDMM and SpMM, termed FusedMM. We show that distributed memory 1.5D and 2.5D algorithms for SpMM can be converted to algorithms for SD-DMM with identical communication costs and input / output data layouts. Further, we give two communication-eliding strategies to reduce costs further for FusedMM kernels: either reusing the replication of an input dense matrix for the SDDMM and SpMM in sequence, or fusing the local SDDMM and SpMM kernels. We benchmark FusedMM algorithms on Cori, a Cray XC40 at LBNL, using Erdos-Renyi random matrices and large real-world sparse matrices. On 256 nodes with 68 cores each, 1.5D FusedMM algorithms using either communication eliding approach can save at least 30% of time spent exclusively in communication compared to executing a distributed-memory SpMM and SDDMM kernel in sequence. Our 2.5D communication-eliding algorithms save 21 % of communication time compared to the unoptimized sequence. On real-world matrices with hundreds of millions of edges, all of our algorithms exhibit at least a 10x speedup over the SpMM algorithm in PETSc. On these matrices, our communication-eliding techniques exhibit runtimes up to 1.6 times faster than an unoptimized sequence of SDDMM and SpMM. We embed and test the scaling of our algorithms in real-world applications, including collaborative filtering via alternating-least-squares and inference for attention-based graph neural networks.

Similar Papers
  • Research Article
  • Citations430

Large-Scale Sparse Singular Value Computations

  • Apr 01, 1992
  • The International Journal of Supercomputing Applications
  • Michael W Berry
  • Research Article

Exploring Hybrid Techniques for Enhanced Pitch Estimation in Speech Processing

  • Oct 01, 2025
  • Recent Patents on Engineering
  • S.K.B Sangeetha +4
  • Research Article
  • Citations19

A systematic comparison of runtime load balancing algorithms for massively parallel rigid particle dynamics

  • Jul 05, 2019
  • Computer Physics Communications
  • Sebastian Eibl +1
  • Research Article
  • Citations6

Navigating prevalence shifts in image analysis algorithm deployment.

  • May 01, 2025
  • Medical image analysis
  • Patrick Godau +7
  • Conference Article
  • Citations63

BalanceFL: Addressing Class Imbalance in Long-Tail Federated Learning

  • May 01, 2022
  • Xian Shuai +5
  • Conference Article
  • Citations57

Multi-view Clustering with Graph Embedding for Connectome Analysis

  • Nov 06, 2017
  • Guixiang Ma +6
  • Research Article
  • Citations2

Parallel direct method for linear equations

  • Sep 01, 1999
  • Advances in Engineering Software
  • M.H.B.M Shariff +1
  • Research Article
  • Citations131

Real-Time Stochastic Optimization of Complex Energy Systems on High-Performance Computers

  • Sep 01, 2014
  • Computing in Science & Engineering
  • Cosmin G Petra +2
  • Conference Article
  • Citations47

Real-time neural network inference on extremely weak devices

  • Oct 14, 2022
  • Kai Huang +1
  • Research Article
  • Citations5

Estimating the largest singular values of large sparse matrices via modified moments

  • Oct 01, 1991
  • Numerical Algorithms
  • Michael Berry +1
  • Conference Article
  • Citations14

DiMES

  • Jan 01, 2005
  • Dipanjan Gope +2
  • Research Article
  • Citations4

An Integrated Collaborative Filtering Framework With Location-Aware Graph Embedding in Intelligent Internet of Things Systems

  • Dec 01, 2022
  • IEEE Transactions on Aerospace and Electronic Systems
  • Yong Sun +5
  • Conference Article
  • Citations12

Issues in parallelizing multiobjective evolutionary algorithms for real world applications

  • Mar 11, 2002
  • David A Van Veldhuizen +2
  • Conference Article

Swift: High-Performance Sparse-Dense Matrix Multiplication on GPUs

  • Jan 31, 2026
  • Jinyu Hu +5
  • Book Chapter
  • Citations11

Enhanced SVD for Collaborative Filtering

  • Jan 01, 2016
  • Xin Guan +2
Cactus Communications logo

Copyright 2026 Cactus Communications. All rights reserved.