vix.ing · top · new · best · stats · spec

Communication-Optimal Convolutional Neural Nets

2018/02/19 by James Demmel, Demmel, James, Grace Dinh +1 · 1 citation
Computer Science · Engineering · #Advanced Neural Network Applications #Computational Complexity (cs.CC) #Data Structures and Algorithms (cs.DS) #FOS: Computer and information sciences #Sparse and Compressive Sensing Techniques #Stochastic Gradient Optimization Techniques

paper · pdf · doi:10.48550/arxiv.1802.06905

openalex publication_date 2018/02/19 · openalex created_date 2025/10/10 · openalex updated_date 2026/07/28

Abstract

Efficiently executing convolutional neural nets (CNNs) is important in many machine-learning tasks. Since the cost of moving a word of data, either between levels of a memory hierarchy or between processors over a network, is much higher than the cost of an arithmetic operation, minimizing data movement is critical to performance optimization. In this paper, we present both new lower bounds on data movement needed for CNNs, and optimal sequential algorithms that attain these lower bounds. In most common cases, our optimal algorithms can attain significantly more data reuse than matrix multiplication.

Cited by

Related