He, Yihan

4 publications

NeurIPS 2024 Global Convergence in Training Large-Scale Transformers Cheng Gao, Yuan Cao, Zihao Li, Yihan He, Mengdi Wang, Han Liu, Jason M. Klusowski, Jianqing Fan

NeurIPS 2024 One-Layer Transformer Provably Learns One-Nearest Neighbor in Context Zihao Li, Yuan Cao, Cheng Gao, Yihan He, Han Liu, Jason M. Klusowski, Jianqing Fan, Mengdi Wang

NeurIPS 2022 SafeBench: A Benchmarking Platform for Safety Evaluation of Autonomous Vehicles Chejian Xu, Wenhao Ding, Weijie Lyu, Zuxin Liu, Shuai Wang, Yihan He, Hanjiang Hu, Ding Zhao, Bo Li

ICML 2020 Don’t Waste Your Bits! Squeeze Activations and Gradients for Deep Neural Networks via TinyScript Fangcheng Fu, Yuzheng Hu, Yihan He, Jiawei Jiang, Yingxia Shao, Ce Zhang, Bin Cui