| 2025 | ACL | CoVE: Compressed Vocabulary Expansion Makes Better LLM-based Recommender Systems. | Haochen Zhang, Tianyi Zhang, Junze Yin, Oren Gal, Anshumali Shrivastava, Vladimir Braverman |
| 2025 | AISTATS | An Iterative Algorithm for Rescaled Hyperbolic Functions Regression. | Yeqi Gao, Zhao Song, Junze Yin |
| 2025 | ICLR | Efficient Alternating Minimization with Applications to Weighted Low Rank Approximation. | Zhao Song, Mingquan Ye, Junze Yin, Lichen Zhang |
| 2025 | ICML | Binary Hypothesis Testing for Softmax Models and Leverage Score Models. | Yuzhou Gu, Zhao Song, Junze Yin |
| 2025 | KDD | The Expressibility of Polynomial based Attention Scheme. | Zhao Song, Chongxi Wang, Guangyi Xu, Junze Yin |
| 2025 | UAI | A Fast Optimization View: Reformulating Single Layer Attention in LLM Based on Tensor and SVM Trick, and Solving It in Matrix Multiplication Time. | Yeqi Gao, Zhao Song, Weixin Wang, Junze Yin |
| 2025 | UAI | Dynamic Maintenance of Kernel Density Estimation Data Structure: From Practice to Theory. | Jiehao Liang, Zhao Song, Zhaozhuo Xu, Junze Yin, Danyang Zhuo |
| 2024 | AISTATS | Fast Dynamic Sampling for Determinantal Point Processes. | Zhao Song, Junze Yin, Lichen Zhang, Ruizhe Zhang |
| 2024 | AISTATS | Solving Attention Kernel Regression Problem via Pre-conditioner. | Zhao Song, Junze Yin, Lichen Zhang |
| 2024 | ICLR | Low Rank Matrix Completion via Robust Alternating Minimization in Nearly Linear Time. | Yuzhou Gu, Zhao Song, Junze Yin, Lichen Zhang |
| 2023 | ICML | A Nearly-Optimal Bound for Fast Regression with ℓ | Zhao Song, Mingquan Ye, Junze Yin, Lichen Zhang |