Kaidi Cao

According to our database, Kaidi Cao authored at least 30 papers between 2018 and 2024.

Collaborative distances:

• Dijkstra number of four.
• Erdős number of four.

2024

AvaTaR: Optimizing LLM Agents for Tool-Assisted Knowledge Retrieval.

Shirley Wu, Shiyu Zhao, Qian Huang, Kexin Huang, Michihiro Yasunaga, Kaidi Cao, Vassilis N. Ioannidis, Karthik Subbian, Jure Leskovec, James Zou

CoRR, 2024

STaRK: Benchmarking LLM Retrieval on Textual and Relational Knowledge Bases.

Shirley Wu, Shiyu Zhao, Michihiro Yasunaga, Kexin Huang, Kaidi Cao, Qian Huang, Vassilis N. Ioannidis, Karthik Subbian, James Zou, Jure Leskovec

CoRR, 2024

PyTorch Frame: A Modular Framework for Multi-Modal Tabular Learning.

Weihua Hu, Yiwen Yuan, Zecheng Zhang, Akihiro Nitta, Kaidi Cao, Vid Kocijan, Jure Leskovec, Matthias Fey

CoRR, 2024

2023

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

Shirley Wu, Kaidi Cao, Bruno Ribeiro, James Zou, Jure Leskovec

CoRR, 2023

TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs.

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

CoRR, 2023

Communication-Free Distributed GNN Training with Vertex Cut.

Kaidi Cao, Rui Deng, Shirley Wu, Edward W. Huang, Karthik Subbian, Jure Leskovec

CoRR, 2023

TpuGraphs: A Performance Prediction Dataset on Large Tensor Computational Graphs.

Phitchaya Mangpo Phothilimthana, Sami Abu-El-Haija, Kaidi Cao, Bahare Fatemi, Michael Burrows, Charith Mendis, Bryan Perozzi

Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023

Learning Large Graph Property Prediction via Graph Segment Training.

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

Proceedings of the Advances in Neural Information Processing Systems 36: Annual Conference on Neural Information Processing Systems 2023, 2023

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

Kaidi Cao, Jiaxuan You, Jiaju Liu, Jure Leskovec

Proceedings of the Eleventh International Conference on Learning Representations, 2023

2022

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

Weihua Hu, Kaidi Cao, Kexin Huang, Edward W. Huang, Karthik Subbian, Jure Leskovec

CoRR, 2022

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

Tailin Wu, Qinchen Wang, Yinan Zhang, Rex Ying, Kaidi Cao, Rok Sosic, Ridwan Jalali, Hassan Hamam, Marko Maucec, Jure Leskovec

Proceedings of the KDD '22: The 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, Washington, DC, USA, August 14, 2022

Learning Backward Compatible Embeddings.

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

Proceedings of the KDD '22: The 28th ACM SIGKDD Conference on Knowledge Discovery and Data Mining, Washington, DC, USA, August 14, 2022

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

Kaidi Cao, Jiaxuan You, Jure Leskovec

Proceedings of the Tenth International Conference on Learning Representations, 2022

Open-World Semi-Supervised Learning.

Kaidi Cao, Maria Brbic, Jure Leskovec

Proceedings of the Tenth International Conference on Learning Representations, 2022

2021

Heteroskedastic and Imbalanced Deep Learning with Adaptive Regularization.

Kaidi Cao, Yining Chen, Junwei Lu, Nikos Aréchiga, Adrien Gaidon, Tengyu Ma

Proceedings of the 9th International Conference on Learning Representations, 2021

Concept Learners for Few-Shot Learning.

Kaidi Cao, Maria Brbic, Jure Leskovec

Proceedings of the 9th International Conference on Learning Representations, 2021

2020

Coresets for Robust Training of Neural Networks against Noisy Labels.

Baharan Mirzasoleiman, Kaidi Cao, Jure Leskovec

CoRR, 2020

Concept Learners for Generalizable Few-Shot Learning.

Kaidi Cao, Maria Brbic, Jure Leskovec

CoRR, 2020

Coresets for Robust Training of Deep Neural Networks against Noisy Labels.

Baharan Mirzasoleiman, Kaidi Cao, Jure Leskovec

Proceedings of the Advances in Neural Information Processing Systems 33: Annual Conference on Neural Information Processing Systems 2020, 2020

Few-Shot Video Classification via Temporal Alignment.

Kaidi Cao, Jingwei Ji, Zhangjie Cao, Chien-Yi Chang, Juan Carlos Niebles

Proceedings of the 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2020

Interstellar: Using Halide's Scheduling Language to Analyze DNN Accelerators.

Xuan Yang, Mingyu Gao, Qiaoyi Liu, Jeff Setter, Jing Pu, Ankita Nayak, Steven Bell, Kaidi Cao, Heonjae Ha, Priyanka Raina, Christos Kozyrakis, Mark Horowitz

Proceedings of the ASPLOS '20: Architectural Support for Programming Languages and Operating Systems, 2020

2019

Learning Imbalanced Datasets with Label-Distribution-Aware Margin Loss.

Kaidi Cao, Colin Wei, Adrien Gaidon, Nikos Aréchiga, Tengyu Ma

Proceedings of the Advances in Neural Information Processing Systems 32: Annual Conference on Neural Information Processing Systems 2019, 2019

Disentangling Content and Style via Unsupervised Geometry Distillation.

Wayne Wu, Kaidi Cao, Cheng Li, Chen Qian, Chen Change Loy

Proceedings of the Deep Generative Models for Highly Structured Data, 2019

Delving Deep Into Hybrid Annotations for 3D Human Recovery in the Wild.

Yu Rong, Ziwei Liu, Cheng Li, Kaidi Cao, Chen Change Loy

Proceedings of the 2019 IEEE/CVF International Conference on Computer Vision, 2019

Learning Temporal Action Proposals With Fewer Labels.

Jingwei Ji, Kaidi Cao, Juan Carlos Niebles

Proceedings of the 2019 IEEE/CVF International Conference on Computer Vision, 2019

TransGaGa: Geometry-Aware Unsupervised Image-To-Image Translation.

Wayne Wu, Kaidi Cao, Cheng Li, Chen Qian, Chen Change Loy

Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, 2019

2018

CariGANs: unpaired photo-to-caricature translation.

Kaidi Cao, Jing Liao, Lu Yuan

ACM Trans. Graph., 2018

DNN Dataflow Choice Is Overrated.

Xuan Yang, Mingyu Gao, Jing Pu, Ankita Nayak, Qiaoyi Liu, Steven Bell, Jeff Setter, Kaidi Cao, Heonjae Ha, Christos Kozyrakis, Mark Horowitz

CoRR, 2018

Pose-Robust Face Recognition via Deep Residual Equivariant Mapping.

Kaidi Cao, Yu Rong, Cheng Li, Xiaoou Tang, Chen Change Loy

Proceedings of the 2018 IEEE Conference on Computer Vision and Pattern Recognition, 2018

Merge or Not? Learning to Group Faces via Imitation Learning.

Yue He, Kaidi Cao, Cheng Li, Chen Change Loy

Proceedings of the Thirty-Second AAAI Conference on Artificial Intelligence, 2018