A Nearly-Optimal Bound for Fast Regression with ℓ∞ Guarantee. - Zhao Song 002; Mingquan Ye; Junze Yin; Lichen Zhang 003 (ICML, 2023)

Federated Empirical Risk Minimization via Second-Order Method. - Song Bian 002; Zhao Song 002; Junze Yin (CoRR, 2023)

Faster Robust Tensor Power Method for Arbitrary Order. - Yichuan Deng 002; Zhao Song 002; Junze Yin (CoRR, 2023)

Query Complexity of Active Learning for Function Family With Nearly Orthogonal Basis. - Xiang Chen; Zhao Song 002; Baocheng Sun 002; Junze Yin; Danyang Zhuo (CoRR, 2023)

GradientCoin: A Peer-to-Peer Decentralized Large Language Models. - Yeqi Gao; Zhao Song 002; Junze Yin (CoRR, 2023)

A Unified Scheme of ResNet and Softmax. - Zhao Song 002; Weixin Wang 002; Junze Yin (CoRR, 2023)

The Expressibility of Polynomial based Attention Scheme. - Zhao Song 002; Guangyi Xu 002; Junze Yin (CoRR, 2023)

Revisiting Quantum Algorithms for Linear Regressions: Quadratic Speedups without Data-Dependent Parameters. - Zhao Song 002; Junze Yin; Ruizhe Zhang 001 (CoRR, 2023)

Local Convergence of Approximate Newton Method for Two Layer Nonlinear Regression. - Zhihang Li; Zhao Song 002; Zifan Wang 003; Junze Yin (CoRR, 2023)

Fast Dynamic Sampling for Determinantal Point Processes. - Zhao Song 002; Junze Yin; Lichen Zhang 003; Ruizhe Zhang 001 (AISTATS, 2024)

Solving Attention Kernel Regression Problem via Pre-conditioner. - Zhao Song 002; Junze Yin; Lichen Zhang 003 (AISTATS, 2024)

Low Rank Matrix Completion via Robust Alternating Minimization in Nearly Linear Time. - Yuzhou Gu; Zhao Song 002; Junze Yin; Lichen Zhang 003 (ICLR, 2024)

How to Inverting the Leverage Score Distribution? - Zhihang Li; Zhao Song 002; Weixin Wang 002; Junze Yin; Zheng Yu (CoRR, 2024)

Conv-Basis: A New Paradigm for Efficient Attention Inference and Gradient Computation in Transformers. - Jiuxiang Gu; Yingyu Liang; Heshan Liu; Zhenmei Shi; Zhao Song 002; Junze Yin (CoRR, 2024)

Inverting the Leverage Score Gradient: An Efficient Approximate Newton Method. - Chenyang Li; Zhao Song 002; Zhaoxing Xu; Junze Yin (CoRR, 2024)

CoVE: Compressed Vocabulary Expansion Makes Better LLM-based Recommender Systems. - Haochen Zhang; Tianyi Zhang 011; Junze Yin; Oren Gal; Anshumali Shrivastava; Vladimir Braverman (ACL (Findings), 2025)

An Iterative Algorithm for Rescaled Hyperbolic Functions Regression. - Yeqi Gao; Zhao Song 002; Junze Yin (AISTATS, 2025)

Fast and Efficient Matching Algorithm with Deadline Instances. - Zhao Song 002; Weixin Wang 002; Chenbo Yin 002; Junze Yin (CPAL, 2025)

Efficient Alternating Minimization with Applications to Weighted Low Rank Approximation. - Zhao Song 002; Mingquan Ye; Junze Yin; Lichen Zhang 003 (ICLR, 2025)

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 002; Weixin Wang 002; Junze Yin (UAI, 2025)

Dynamic Maintenance of Kernel Density Estimation Data Structure: From Practice to Theory. - Jiehao Liang; Zhao Song 002; Zhaozhuo Xu; Junze Yin; Danyang Zhuo (UAI, 2025)

I3S: Importance Sampling Subspace Selection for Low-Rank Optimization in LLM Pretraining. - Haochen Zhang; Junze Yin; Guanchu Wang; Zirui Liu 001; Tianyi Zhang 011; Anshumali Shrivastava; Lin Yang 011; Vladimir Braverman (CoRR, 2025)