4 papers
KGCache: Amortized Subgraph Retrieval for KG Reasoning with LLMs
Uros Stanic, Changcheng Yuan, Sabuj Laskar +1
Large language models can answer knowledge-intensive questions more reliably when they are grounded with knowledge graphs, but systems such as Think-on-Graph and Reasoning-on-Graph…
Sparsity-Aware Roofline Models for Sparse Matrix-Matrix Multiplication
Matthew Qian, Yahia Ramadan, Suhita Anubha +1
Sparse matrix-dense matrix multiplication (SpMM) is a critical kernel in scientific computing, graph analytics, and machine learning, whose performance is often constrained by memo…
Distributed-memory Algorithms for Sparse Matrix Permutation, Extraction, and Assignment
Elaheh Hassani, Md Taufique Hussain, Ariful Azad
We present scalable distributed-memory algorithms for sparse matrix permutation, extraction, and assignment. Our methods follow an Identify-Exchange-Build (IEB) strategy where each…
SparseTransX: Efficient Training of Translation-Based Knowledge Graph Embeddings Using Sparse Matrix Operations
Md Saidul Hoque Anik, Ariful Azad
Knowledge graph (KG) learning offers a powerful framework for generating new knowledge and making inferences. Training KG embedding can take a significantly long time, especially f…