Shuai Zhang

Shuai Zhang

Amazon

Brief Biography

I am currently employed at Amazon Web Services AI, focusing on foundation models related research, opensource projects, and products. Before this role, I was a postdoctoral researcher in the Department of Computer Science at ETH Zurich, under the guidance of Prof. Ce Zhang. I received a PhD degree from the school of computer science and engineering of UNSW Sydney, advised by Prof. Lina Yao, and my PhD study was also supported by CSIRO’s Data61 PhD Scholarship. I obtained a Bachelor degree from Nanjing University.

I am a recipient of the

Selected Publications

Prompt Pre-Training with Twenty-Thousand Classes for Open-Vocabulary Visual Recognition
Shuhuai Ren, Aston Zhang, Yi Zhu, Shuai Zhang, Shuai Zheng, Mu Li, Alex Smola, Xu Sun.
NeurIPS 2023 . Thirty-seventh Conference on Neural Information Processing Systems. code

Data-Informed Geometric Space Selection
Shuai Zhang, Wenqi Jiang.
NeurIPS 2023 . Thirty-seventh Conference on Neural Information Processing Systems.

xFraud: Explainable Fraud Transaction Detection
Susie Xi Rao, Shuai Zhang, Zhichao Han, Zitao Zhang, Wei Min, Zhiyao Chen, Yinan Shan, Yang Zhao, Ce Zhang.
VLDB 2022 . The Proceedings of the VLDB Endowment. code

Neural Methods for Logical Reasoning over Knowledge Graphs
Alfonso Amayuelas, Shuai Zhang, Susie Xi Rao, Ce Zhang.
ICLR 2022 . The International Conference on Learning Representations. code

Beyond Fully-Connected Layers with Quaternions: Parameterization of Hypercomplex Multiplications with 1/n Parameters
Aston Zhang, Yi Tay, Shuai Zhang, Alvin Chan, Anh Tuan Luu, Siu Hui, Jie Fu.
ICLR 2021. The International Conference on Learning Representations. code
Outstanding Paper Award.

Self-Instantiated Recurrent Units with Dynamic Soft Recursion
Aston Zhang, Yi Tay, Yikang Shen, Alvin Chan, Shuai Zhang
NeurIPS 2021 . Thirty-fifth Conference on Neural Information Processing Systems. code

HyperML: A Boosting Metric Learning Approach in Hyperbolic Space for Recommender Systems
Lucas Vinh Tran, Yi Tay, Shuai Zhang, Gao Cong, Xiaoli Li.
WSDM 2020. The 13th ACM International Conference on Web Search and Data Mining.
Best Paper Award Runner-up.

Quaternion Knowledge Graph Embeddings
Shuai Zhang, Yi Tay, Lina Yao, Qi Liu.
NeurIPS 2019. Thirty-third Conference on Neural Information Processing Systems code.

Lightweight and Efficient Neural Natural Language Processing with Quaternion Networks
Yi Tay, Aston Zhang, Anh Tuan Luu, Jinfeng Rao, Shuai Zhang, Shuohang Wang, Jie Fu and Siu Cheung Hui.
ACL 2019. The 57th Annual Meeting of the Association for Computational Linguistics code.

Mentoring and Services

  • Zitian Chen, applied scientist intern, PhD student from University of Massachusetts Amherst.
  • Pei Chen, applied scientist intern, PhD student from Texas A&M University.
  • Yixin Chen, applied scientist intern, PhD student from the Chinese University of Hong Kong.
  • Bingqing Song, applied scientist intern, PhD student from University of Minnesota.
  • Shikai Qiu, applied scientist intern, PhD student from New York University.
  • Shuhuai Ren, applied scientist intern, PhD student from Peking University.
  • Alfonso Amayuelas (Now PhD student at UCSB), master thesis (results are published at ICLR 2022), EPFL, co-advisor with Prof. Ce Zhang
  • Maksimov Anton, master thesis, ETH zurich, co-advisor with Prof. Ce Zhang
  • Yannick Wattenberg, bachelor thesis, ETH zurich, co-advisor with Prof. Ce Zhang
  • Aurelio Negri, bachelor thesis, ETH zurich, co-advisor with Prof. Ce Zhang
  • Clémence Lanfranchi, semester project, ETH zurich, co-advisor with Prof. Ce Zhang

I am actively serving as reviewer/program committe member for AI conferences (NeurIPS, ICML, ICLR, ACL, etc.) and journals.