中文
相关论文

相关论文: TinyKG: Memory-Efficient Training Framework for Kn…

200 篇论文

Recently, graph neural networks (GNNs), as the backbone of graph-based machine learning, demonstrate great success in various domains (e.g., e-commerce). However, the performance of GNNs is usually unsatisfactory due to the highly sparse…

机器学习 · 计算机科学 2023-06-02 Yuke Wang , Boyuan Feng , Zheng Wang , Guyue Huang , Yufei Ding

Graph Neural Networks (GNNs) offer a compact and computationally efficient way to learn embeddings and classifications on graph data. GNN models are frequently large, making distributed minibatch training necessary. The primary contribution…

机器学习 · 计算机科学 2024-04-22 Alok Tripathy , Katherine Yelick , Aydin Buluc

Graph convolutional networks (GCNs) have been employed as a kind of significant tool on many graph-based applications recently. Inspired by convolutional neural networks (CNNs), GCNs generate the embeddings of nodes by aggregating the…

机器学习 · 计算机科学 2020-11-20 Tao Huang , Yihan Zhang , Jiajing Wu , Junyuan Fang , Zibin Zheng

Designing accurate deep learning models for molecular property prediction plays an increasingly essential role in drug and material discovery. Recently, due to the scarcity of labeled molecules, self-supervised learning methods for learning…

生物大分子 · 定量生物学 2022-06-08 Han Li , Dan Zhao , Jianyang Zeng

Graph kernels are historically the most widely-used technique for graph classification tasks. However, these methods suffer from limited performance because of the hand-crafted combinatorial features of graphs. In recent years, graph neural…

机器学习 · 计算机科学 2022-02-28 Aosong Feng , Chenyu You , Shiqiang Wang , Leandros Tassiulas

Training deep graph neural networks (GNNs) is notoriously hard. Besides the standard plights in training deep architectures such as vanishing gradients and overfitting, it also uniquely suffers from over-smoothing, information squashing,…

机器学习 · 计算机科学 2022-05-10 Tianlong Chen , Kaixiong Zhou , Keyu Duan , Wenqing Zheng , Peihao Wang , Xia Hu , Zhangyang Wang

Recent approaches of computer vision utilize deep learning methods as they perform quite well if training and testing domains follow the same underlying data distribution. However, it has been shown that minor variations in the images that…

计算机视觉与模式识别 · 计算机科学 2022-01-31 Sebastian Monka , Lavdim Halilaj , Achim Rettinger

Continual Knowledge Graph Embedding (CKGE) seeks to integrate new knowledge while preserving past information. However, existing methods struggle with efficiency and scalability due to two key limitations: (1) suboptimal knowledge…

计算与语言 · 计算机科学 2025-06-11 Lijing Zhu , Qizhen Lan , Qing Tian , Wenbo Sun , Li Yang , Lu Xia , Yixin Xie , Xi Xiao , Tiehang Duan , Cui Tao , Shuteng Niu

Graph neural network (GNN) inference faces significant bottlenecks in preprocessing, which often dominate overall inference latency. We introduce AutoGNN, an FPGA-based accelerator designed to address these challenges by leveraging FPGA's…

Incorporating Knowledge Graphs into Recommendation has attracted growing attention in industry, due to the great potential of KG in providing abundant supplementary information and interpretability for the underlying models. However, simply…

信息检索 · 计算机科学 2024-06-03 Ding Zou , Wei Wei , Feida Zhu , Chuanyu Xu , Tao Zhang , Chengfu Huo

Neural Processes (NPs) are a rapidly evolving class of models designed to directly model the posterior predictive distribution of stochastic processes. Originally developed as a scalable alternative to Gaussian Processes (GPs), which are…

机器学习 · 计算机科学 2026-04-20 Daniel Jenson , Jhonathan Navott , Mengyan Zhang , Makkunda Sharma , Elizaveta Semenova , Seth Flaxman

This paper presents GraphAGILE, a domain-specific FPGA-based overlay accelerator for graph neural network (GNN) inference. GraphAGILE consists of (1) \emph{a novel unified architecture design} with an \emph{instruction set}, and (2) \emph{a…

分布式、并行与集群计算 · 计算机科学 2023-04-25 Bingyi Zhang , Hanqing Zeng , Viktor Prasanna

Communication is a key bottleneck for distributed graph neural network (GNN) training. This paper proposes GNNPipe, a new approach that scales the distributed full-graph deep GNN training. Being the first to use layer-level model…

分布式、并行与集群计算 · 计算机科学 2023-09-26 Jingji Chen , Zhuoming Chen , Xuehai Qian

Temporal Graph Neural Networks (TGNNs) are powerful models to capture temporal, structural, and contextual information on temporal graphs. The generated temporal node embeddings outperform other methods in many downstream tasks. Real-world…

硬件体系结构 · 计算机科学 2022-03-11 Hongkuan Zhou , Bingyi Zhang , Rajgopal Kannan , Viktor Prasanna , Carl Busart

Graph Convolutional Networks (GCNs) are increasingly adopted in large-scale graph-based recommender systems. Training GCN requires the minibatch generator traversing graphs and sampling the sparsely located neighboring nodes to obtain their…

机器学习 · 计算机科学 2021-08-17 Seung Won Min , Kun Wu , Sitao Huang , Mert Hidayetoğlu , Jinjun Xiong , Eiman Ebrahimi , Deming Chen , Wen-mei Hwu

Deploying neural networks on the edge has become increasingly important as deep learning is being applied in an increasing amount of applications. At the edge computing hardware typically has limited resources disallowing to run neural…

机器学习 · 计算机科学 2025-09-15 Quinten Van Baelen , Peter Karsmakers

Knowledge Graphs (KGs) represent human-crafted factual knowledge in the form of triplets (head, relation, tail), which collectively form a graph. Question Answering over KGs (KGQA) is the task of answering natural questions grounding the…

计算与语言 · 计算机科学 2024-05-31 Costas Mavromatis , George Karypis

The Long-Short-Term-Memory Recurrent Neural Networks (LSTM RNNs) are a popular class of machine learning models for analyzing sequential data. Their training on modern GPUs, however, is limited by the GPU memory capacity. Our profiling…

机器学习 · 计算机科学 2019-12-02 Bojian Zheng , Abhishek Tiwari , Nandita Vijaykumar , Gennady Pekhimenko

Temporal Knowledge Graph (TKG) completion models traditionally assume access to the entire graph during training. This overlooks challenges stemming from the evolving nature of TKGs, such as: (i) the model's requirement to generalize and…

人工智能 · 计算机科学 2025-07-28 Mehrnoosh Mirtaheri , Ryan A. Rossi , Sungchul Kim , Kanak Mahadik , Tong Yu , Xiang Chen , Mohammad Rostami

Training deep convolutional neural networks such as VGG and ResNet by gradient descent is an expensive exercise requiring specialized hardware such as GPUs. Recent works have examined the possibility of approximating the gradient…

计算机视觉与模式识别 · 计算机科学 2019-08-16 Ziheng Wang , Sree Harsha Nelaturu