Improved Operator Learning by Orthogonal Attention. - Zipeng Xiao; Zhongkai Hao; Bokai Lin; Zhijie Deng; Hang Su 006 (ICML, 2024)

Amortized Fourier Neural Operators. - Zipeng Xiao; Siqi Kou; Zhongkai Hao; Bokai Lin; Zhijie Deng (NeurIPS, 2024)

MatryoshkaKV: Adaptive KV Compression via Trainable Orthogonal Projection. - Bokai Lin; Zihao Zeng; Zipeng Xiao; Siqi Kou; TianQi Hou; Xiaofeng Gao 001; Hao Zhang; Zhijie Deng (ICLR, 2025)