Gao, Yihang

6 publications

TMLR 2025 AlgoFormer: An Efficient Transformer Framework with Algorithmic Structures Yihang Gao, Chuanyang Zheng, Enze Xie, Han Shi, Tianyang Hu, Yu Li, Michael Ng, Zhenguo Li, Zhaoqiang Liu

NeurIPS 2025 SAS: Simulated Attention Score Chuanyang Zheng, Jiankai Sun, Yihang Gao, Yuehao Wang, Peihao Wang, Jing Xiong, Liliang Ren, Hao Cheng, Janardhan Kulkarni, Yelong Shen, Zhangyang Wang, Mac Schwager, Anderson Schneider, Xiaodong Liu, Jianfeng Gao

ICML 2025 SepLLM: Accelerate Large Language Models by Compressing One Segment into One Separator Guoxuan Chen, Han Shi, Jiawei Li, Yihang Gao, Xiaozhe Ren, Yimeng Chen, Xin Jiang, Zhenguo Li, Weiyang Liu, Chao Huang

NeurIPS 2024 DAPE: Data-Adaptive Positional Encoding for Length Extrapolation Chuanyang Zheng, Yihang Gao, Han Shi, Minbin Huang, Jingyao Li, Jing Xiong, Xiaozhe Ren, Michael Ng, Xin Jiang, Zhenguo Li, Yu Li

ICML 2023 Gradient Descent Finds the Global Optima of Two-Layer Physics-Informed Neural Networks Yihang Gao, Yiqi Gu, Michael Ng

NeurIPS 2022 Approximate Secular Equations for the Cubic Regularization Subproblem Yihang Gao, Man-Chung Yue, Michael Ng