18 citations · 111 across the 34 of their papers we have counts for
6 papers · 1 filter
MolCPT: Molecule Continuous Prompt Tuning to Generalize Molecular Representation Learning
Cameron Diao, Kaixiong Zhou, Zirui Liu +2
Molecular representation learning is crucial for the problem of molecular property prediction, where graph neural networks (GNNs) serve as an effective solution due to their struct…
DreamShard: Generalizable Embedding Table Placement for Recommender Systems
Daochen Zha, Louis Feng, Qiaoyu Tan +6
We study embedding table placement for distributed recommender systems, which aims to partition and place the tables on multiple hardware devices (e.g., GPUs) to balance the comput…
RSC: Accelerating Graph Neural Networks Training via Randomized Sparse Computations
Zirui Liu, Shengyuan Chen, Kaixiong Zhou +3
The training of graph neural networks (GNNs) is extremely time consuming because sparse graph-based operations are hard to be accelerated by hardware. Prior art explores trading of…
A Comprehensive Study on Large-Scale Graph Training: Benchmarking and Rethinking
Keyu Duan, Zirui Liu, Peihao Wang +5
Large-scale graph training is a notoriously challenging problem for graph neural networks (GNNs). Due to the nature of evolving graph structures into the training process, vanilla…
DIVISION: Memory Efficient Training via Dual Activation Precision
Guanchu Wang, Zirui Liu, Zhimeng Jiang +3
Activation compressed training provides a solution towards reducing the memory cost of training deep neural networks~(DNNs). However, state-of-the-art work combines a search of qua…
FMP: Toward Fair Graph Message Passing against Topology Bias
Zhimeng Jiang, Xiaotian Han, Chao Fan +4
Despite recent advances in achieving fair representations and predictions through regularization, adversarial debiasing, and contrastive learning in graph neural networks (GNNs), t…