5 papers
NeutronSparse: Coordinating Heterogeneous Engines for Sparse Matrix Multiplication on NPUs
Xin Ai, Zeyu Ling, Hao Yuan +4
Sparse matrix-matrix multiplication (SpMM) is a fundamental data operation for large-scale sparse data processing. With NPUs increasingly deployed in data centers for their perform…
AcOrch: Accelerating Sampling-based GNN Training under CPU-NPU Heterogeneous Environments
Kefu Chen, Xin Ai, Qiange Wang +2
Graph Neural Networks (GNNs) have achieved remarkable success in various applications. Sampling-based GNN training, which conducts mini-batch training on sampled subgraphs, has bec…
EvoRAG: Making Knowledge Graph-based RAG Automatically Evolve through Feedback-driven Backpropagation
Zhenbo Fu, Yuanzhe Zhang, Qiange Wang +5
Knowledge Graph-based Retrieval-Augmented Generation (KG-RAG) has emerged as a promising paradigm for enhancing LLM reasoning by retrieving multi-hop paths from KGs. However, exist…
Towards Autonomous Graph Data Analytics with Analytics-Augmented Generation
Qiange Wang, Chaoyi Chen, Jingqi Gao +3
This paper argues that reliable end-to-end graph data analytics cannot be achieved by retrieval- or code-generation-centric LLM agents alone. Although large language models (LLMs)…
NeutronTP: Load-Balanced Distributed Full-Graph GNN Training with Tensor Parallelism
Xin Ai, Hao Yuan, Zeyu Ling +6
Graph neural networks (GNNs) have emerged as a promising direction. Training large-scale graphs that relies on distributed computing power poses new challenges. Existing distribute…