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

Lightweight In-Context Tuning for Multimodal Unified Models
Yixin Chen, Shuai Zhang, Boran Han, Jiaya Jia.
Preprint.

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.