activity
20222025
most citedAttention Scheme Inspired Softmax Regression

6 citations · 18 across the 13 of their papers we have counts for

collaborators
Showing cs.LGShow all

6 papers · 1 filter

cs.LG2025★ 1 cited

OpenThoughts: Data Recipes for Reasoning Models

Etash Guha, Ryan Marten, Sedrick Keh +47

Reasoning models have made rapid progress on many benchmarks involving math, code, and science. Yet, there are still many open questions about the best training recipes for reasoni…

cs.LG2024★ 1 cited

How Sparse Attention Approximates Exact Attention? Your Attention is Naturally -Sparse

Yichuan Deng, Zhao Song, Jing Xiong +1

Sparse Attention is a technique that approximates standard attention computation with sub-quadratic complexity. This is achieved by selectively ignoring smaller entries in the atte…

cs.LG2024

Enhancing Stochastic Gradient Descent: A Unified Framework and Novel Acceleration Methods for Faster Convergence

Yichuan Deng, Zhao Song, Chiwun Yang

Based on SGD, previous works have proposed many algorithms that have improved convergence speed and generalization in stochastic optimization, such as SGDm, AdaGrad, Adam, etc. How…

cs.LG2023★ 1 cited

Unmasking Transformers: A Theoretical Approach to Data Recovery via Attention Weights

Yichuan Deng, Zhao Song, Shenghao Xie +1

In the realm of deep learning, transformers have emerged as a dominant architecture, particularly in natural language processing tasks. However, with their widespread adoption, con…

cs.LG2023★ 6 cited

Attention Scheme Inspired Softmax Regression

Yichuan Deng, Zhihang Li, Zhao Song

Large language models (LLMs) have made transformed changes for human society. One of the key computation in LLMs is the softmax unit. This operation is important in LLMs because it…

cs.LG2022★ 2 cited

Training Overparametrized Neural Networks in Sublinear Time

Yichuan Deng, Hang Hu, Zhao Song +2

The success of deep learning comes at a tremendous computational and energy cost, and the scalability of training massively overparametrized neural networks is becoming a real barr…