Alert button
Picture for Kaidi Cao

Kaidi Cao

Alert button

GraphMETRO: Mitigating Complex Distribution Shifts in GNNs via Mixture of Aligned Experts

Dec 07, 2023
Shirley Wu, Kaidi Cao, Bruno Ribeiro, James Zou, Jure Leskovec

Viaarxiv icon

TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs

Aug 25, 2023
Phitchaya Mangpo Phothilimthana, Sami Abu-El-Haija, Kaidi Cao, Bahare Fatemi, Charith Mendis, Bryan Perozzi

Figure 1 for TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs
Figure 2 for TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs
Figure 3 for TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs
Figure 4 for TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs
Viaarxiv icon

Communication-Free Distributed GNN Training with Vertex Cut

Aug 06, 2023
Kaidi Cao, Rui Deng, Shirley Wu, Edward W Huang, Karthik Subbian, Jure Leskovec

Figure 1 for Communication-Free Distributed GNN Training with Vertex Cut
Figure 2 for Communication-Free Distributed GNN Training with Vertex Cut
Figure 3 for Communication-Free Distributed GNN Training with Vertex Cut
Figure 4 for Communication-Free Distributed GNN Training with Vertex Cut
Viaarxiv icon

Learning Large Graph Property Prediction via Graph Segment Training

May 21, 2023
Kaidi Cao, Phitchaya Mangpo Phothilimthana, Sami Abu-El-Haija, Dustin Zelle, Yanqi Zhou, Charith Mendis, Jure Leskovec, Bryan Perozzi

Figure 1 for Learning Large Graph Property Prediction via Graph Segment Training
Figure 2 for Learning Large Graph Property Prediction via Graph Segment Training
Figure 3 for Learning Large Graph Property Prediction via Graph Segment Training
Figure 4 for Learning Large Graph Property Prediction via Graph Segment Training
Viaarxiv icon

AutoTransfer: AutoML with Knowledge Transfer -- An Application to Graph Neural Networks

Mar 14, 2023
Kaidi Cao, Jiaxuan You, Jiaju Liu, Jure Leskovec

Figure 1 for AutoTransfer: AutoML with Knowledge Transfer -- An Application to Graph Neural Networks
Figure 2 for AutoTransfer: AutoML with Knowledge Transfer -- An Application to Graph Neural Networks
Figure 3 for AutoTransfer: AutoML with Knowledge Transfer -- An Application to Graph Neural Networks
Figure 4 for AutoTransfer: AutoML with Knowledge Transfer -- An Application to Graph Neural Networks
Viaarxiv icon

Relational Multi-Task Learning: Modeling Relations between Data and Tasks

Mar 14, 2023
Kaidi Cao, Jiaxuan You, Jure Leskovec

Figure 1 for Relational Multi-Task Learning: Modeling Relations between Data and Tasks
Figure 2 for Relational Multi-Task Learning: Modeling Relations between Data and Tasks
Figure 3 for Relational Multi-Task Learning: Modeling Relations between Data and Tasks
Figure 4 for Relational Multi-Task Learning: Modeling Relations between Data and Tasks
Viaarxiv icon

TuneUp: A Training Strategy for Improving Generalization of Graph Neural Networks

Oct 26, 2022
Weihua Hu, Kaidi Cao, Kexin Huang, Edward W Huang, Karthik Subbian, Jure Leskovec

Figure 1 for TuneUp: A Training Strategy for Improving Generalization of Graph Neural Networks
Figure 2 for TuneUp: A Training Strategy for Improving Generalization of Graph Neural Networks
Figure 3 for TuneUp: A Training Strategy for Improving Generalization of Graph Neural Networks
Figure 4 for TuneUp: A Training Strategy for Improving Generalization of Graph Neural Networks
Viaarxiv icon

Learning Large-scale Subsurface Simulations with a Hybrid Graph Network Simulator

Jun 15, 2022
Tailin Wu, Qinchen Wang, Yinan Zhang, Rex Ying, Kaidi Cao, Rok Sosič, Ridwan Jalali, Hassan Hamam, Marko Maucec, Jure Leskovec

Figure 1 for Learning Large-scale Subsurface Simulations with a Hybrid Graph Network Simulator
Figure 2 for Learning Large-scale Subsurface Simulations with a Hybrid Graph Network Simulator
Figure 3 for Learning Large-scale Subsurface Simulations with a Hybrid Graph Network Simulator
Figure 4 for Learning Large-scale Subsurface Simulations with a Hybrid Graph Network Simulator
Viaarxiv icon

Learning Backward Compatible Embeddings

Jun 07, 2022
Weihua Hu, Rajas Bansal, Kaidi Cao, Nikhil Rao, Karthik Subbian, Jure Leskovec

Figure 1 for Learning Backward Compatible Embeddings
Figure 2 for Learning Backward Compatible Embeddings
Figure 3 for Learning Backward Compatible Embeddings
Figure 4 for Learning Backward Compatible Embeddings
Viaarxiv icon

Open-World Semi-Supervised Learning

Feb 06, 2021
Kaidi Cao, Maria Brbic, Jure Leskovec

Figure 1 for Open-World Semi-Supervised Learning
Figure 2 for Open-World Semi-Supervised Learning
Figure 3 for Open-World Semi-Supervised Learning
Figure 4 for Open-World Semi-Supervised Learning
Viaarxiv icon