Abstract
Graph Neural Networks (GCNs) have attracted wide attention and are applied to the real world. However, due to the ever-growing graph data with significant irregularities, off-chip communication with poor data locality has become the major bottleneck hurdling the development of GCNs. Fortunately, recent works demonstrate Resistive Random Access Memory (ReRAM) has the potential to perform inherently parallel in-situ computation of Matrix-Vector Multiplication (MVM) in the analog regime fundamentally breaking the communication bottleneck.Inspired by this observation, we propose a novel ReRAM-based GCN acceleration co-design (i.e. algorithm-hardware) framework, CoDG-ReRAM, that can deliver real-time GCN inference with high accuracy. On the algorithm side, we propose a novel model optimization pipeline that simultaneously and efficiently sparsifies and regularizes both graph and parameter matrices in GCNs and creates ReRAM-friendly models. On the hardware side, we take advantage of the software optimization results to provide a more systematic mapping scheme and in-crease computation efficiency to have an energy-efficient ReRAM-based GCN acceleration with low latency. Experimental results show that the proposed work improves performance and energy efficiency by 4× and 5.1 × respectively over SOTA ReRAM-based accelerators of GCNs with a negligible accuracy loss.
Original language | English (US) |
---|---|
Title of host publication | Proceedings - 2022 IEEE 40th International Conference on Computer Design, ICCD 2022 |
Publisher | Institute of Electrical and Electronics Engineers Inc. |
Pages | 280-289 |
Number of pages | 10 |
ISBN (Electronic) | 9781665461863 |
DOIs | |
State | Published - 2022 |
Externally published | Yes |
Event | 40th IEEE International Conference on Computer Design, ICCD 2022 - Olympic Valley, United States Duration: Oct 23 2022 → Oct 26 2022 |
Publication series
Name | Proceedings - IEEE International Conference on Computer Design: VLSI in Computers and Processors |
---|---|
Volume | 2022-October |
ISSN (Print) | 1063-6404 |
Conference
Conference | 40th IEEE International Conference on Computer Design, ICCD 2022 |
---|---|
Country/Territory | United States |
City | Olympic Valley |
Period | 10/23/22 → 10/26/22 |
Bibliographical note
Publisher Copyright:© 2022 IEEE.
Keywords
- Computer Architecture
- Graph Neural Network
- Processing In Memory
- Resistive Random Access Memory