Tianyi Zhou 0002

dblp:88/8205-2 · DBLP profile ↗
← Back
2ranked-venue papers in the field
0as first author
2since 2021 · last 2025
—ORCID · unresolved

Domains — venue-derived; a paper can count in several

Big Data, Cloud & Distributed Data Systems · 2
YearPublicationVenuePosition
2025 Why Softmax Attention Outperforms Linear Attention
Yichuan Deng 0002, Zhao Song 0002, Kaijun Yuan, Tianyi Zhou 0002
IEEE Big Data4
2023 Fast Heavy Inner Product Identification Between Weights and Inputs in Neural Network Training
abstract
In this paper, we consider a heavy inner product identification problem, which generalizes the Light Bulb problem ([1]): Given two sets $A \subset\{-1,+1\}^{d}$ and $B \subset\{-1,+1\}^{d}$ with $|A|=|B|=n$, if there are exact k pairs whose inner product passes a certain threshold, i.e., $\{\left(a_{1}, b_{1}\right), \cdots,\left(a_{k}, b_{k}\right)\} \subset A \times B$ such that $\forall i \in[k],\left\langle a_{i}, b_{i}\right\rangle \geq \rho \cdot d$, for a threshold $\rho \in(0,1)$, the goal is to identify those k heavy inner products. We provide an algorithm that runs in $O(n^{2 \omega / 3+o(1)})$ time to find the k inner product pairs that surpass $\rho \cdot d$ threshold with high probability, where $\omega$ is the current matrix multiplication exponent. By solving this problem, our method speed up the training of neural networks with ReLU activation function.
Lianke Qin, Saayan Mitra, Zhao Song 0002, Yuanyuan Yang 0005, Tianyi Zhou 0002
IEEE Big Data5