Jiankai Sun is a machine learning engineer at Pinterest. Previously, he worked as a research scientist at Bytedance Inc. Seattle, focusing on large language models and privacy-preserving techniques such as federated learning. He got his Ph.D. from the Department of Computer Science at the Ohio State University, advised by Prof. Srinivasan Parthasarathy.
His research mainly focuses on large language models, privacy-preserving techniques such as federated learning and some other machine learning related areas such as graph embedding, recommender systems, and question routing in online CQAs.
Please check his Google Scholar page to see his full list of publications.
Please check his LinkedIn to see his most recent working experiences.
Selected Publications on Federated Learning:
DPAUC: Differentially Private AUC Computation in Federated Learning, Jiankai Sun, Xin Yang, Kevin Yao, Junyuan Xie, Di Wu, Chong Wang. AAAI 2023. [paper][code]
Label Leakage and Protection in Two-party Split Learning, Oscar Li, Jiankai Sun, Xin Yang, Weihao Gao, Hongyi Zhang, Junyuan Xie, Virginia Smith, Chong Wang, ICLR 2022. [paper][code]
Defending against Reconstruction Attack in Vertical Federated Learning, Jiankai Sun, Yuanshun Yao, Weihao Gao, Junyuan Xie, Chong Wang, FL-ICML 2021, [paper][code]
Selected Publications on Graph Learning and Recommender Systems:
Privacy Preserving Conversion Modeling in Data Clean Room, RecSys 2024 [paper]
DGRec: Graph Neural Network for Recommendation with Diversified Embedding Generation, Liangwei Yang, Shengjie Wang, Yunzhe Tao, Jiankai Sun, Xiaolong Liu, Philip S. Yu, Taiqing Wang. WSDM 2023. [paper][code]
ATP: Directed Graph Embedding with Asymmetric Transitivity Preservation, Jiankai Sun, Bortik Bandyopadhyay, Armin Bashizade, Jiongqian Liang, P. Sadayappan, Srinivasan Parthasarathy. AAAI 2019. [paper][code]
Semi-supervised Embedding in Attributed Networks with Outliers. Jiongqian Liang, Peter Jacobs, Jiankai Sun, Srinivasan Parthasarathy, SDM'18. [paper][code]
Breaking Cycles in Noisy Hierarchies, Jiankai Sun, Deepak Ajwani, Patrick K. Nicholson, Alessandra Sala, Srinivasan Parthasarathy. WebSci'17. [paper][code]