Efficient streaming language models with attention sinks G Xiao, Y Tian, B Chen, S Han, M Lewis
arXiv preprint arXiv:2309.17453, 2023
405 2023 FlexGen: High-Throughput Generative Inference of Large Language Models with a Single GPU Y Sheng, L Zheng, B Yuan, Z Li, M Ryabinin, B Chen, P Liang, C Re, ...
International Conference on Machine Learning, 2023
339 2023 H Z Zhang, Y Sheng, T Zhou, T Chen, L Zheng, R Cai, Z Song, Y Tian, C Ré, ...
International Conference on Machine Learning, 2023
267 2023 Deja vu: Contextual sparsity for efficient llms at inference time Z Liu, J Wang, T Dao, T Zhou, B Yuan, Z Song, A Shrivastava, C Zhang, ...
International Conference on Machine Learning, 22137-22176, 2023
254 2023 SLIDE: In Defense of Smart Algorithms over Hardware Acceleration for Large-scale Deep Learning Systems B Chen, T Medini, J Farwell, S Gobriel, C Tai, A Shrivastava
Proceedings of Machine Learning and System 2, 291--306, 2020
144 2020 Scatterbrain: Unifying sparse and low-rank attention B Chen, T Dao, E Winsor, Z Song, A Rudra, C Ré
Advances in Neural Information Processing Systems 34, 17413-17426, 2021
139 2021 Monarch: Expressive structured matrices for efficient and accurate training T Dao, B Chen, NS Sohoni, A Desai, M Poli, J Grogan, A Liu, A Rao, ...
International Conference on Machine Learning, 4690-4721, 2022
99 2022 Decentralized training of foundation models in heterogeneous environments B Yuan, Y He, JQ Davis, T Zhang, T Dao, B Chen, P Liang, C Re, C Zhang
Neural Information Processing Systems., 2022
86 2022 Pixelated butterfly: Simple and efficient sparse training for neural network models B Chen, T Dao, K Liang, J Yang, Z Song, A Rudra, C Re
International Conference on Learning Representations, 2022
82 * 2022 MONGOOSE: A learnable LSH framework for efficient neural network training B Chen, Z Liu, B Peng, Z Xu, JL Li, T Dao, Z Song, A Shrivastava, C Re
International Conference on Learning Representations, 2021
81 2021 Scan and Snap: Understanding Training Dynamics and Token Composition in 1-layer Transformer Y Tian, Y Wang, B Chen, S Du
International Conference on Machine Learning, 2023
79 2023 Analyzing log analysis: An empirical study of user log mining S Alspaugh, B Chen, J Lin, A Ganapathi, M Hearst, R Katz
28th Large Installation System Administration Conference (LISA14), 62-77, 2014
71 2014 Llm inference unveiled: Survey and roofline model insights Z Yuan, Y Shang, Y Zhou, Z Dong, Z Zhou, C Xue, B Wu, Z Li, Q Gu, ...
arXiv preprint arXiv:2402.16363, 2024
56 2024 Fast and accurate stochastic gradient estimation B Chen, Y Xu, A Shrivastava
Advances in Neural Information Processing Systems 32, 2019
54 * 2019 Angular visual hardness B Chen, W Liu, Z Yu, J Kautz, A Shrivastava, A Garg, A Anandkumar
International Conference on Machine Learning, 1637-1648, 2020
53 2020 Layer skip: Enabling early exit inference and self-speculative decoding M Elhoushi, A Shrivastava, D Liskovich, B Hosmer, B Wasti, L Lai, ...
arXiv preprint arXiv:2404.16710, 2024
52 2024 Anima Anandkumar, and Yuandong Tian. Galore: Memory-efficient llm training by gradient low-rank projection J Zhao, Z Zhang, B Chen, Z Wang
arXiv preprint arXiv:2403.03507, 2024
52 2024 KIVI: Plug-and-play 2bit KV Cache Quantization with Streaming Asymmetric Quantization Z Liu, J Yuan, H Jin, S Zhong, Z Xu, V Braverman, B Chen, X Hu
52 * 2024 Galore: Memory-efficient llm training by gradient low-rank projection J Zhao, Z Zhang, B Chen, Z Wang, A Anandkumar, Y Tian
arXiv preprint arXiv:2403.03507, 2024
46 2024 A tuning-free asymmetric 2bit quantization for kv cache Z Liu, J Yuan, H Jin, S Zhong, Z Xu, V Braverman, B Chen, XH Kivi
arXiv preprint arXiv:2402.02750, 2024
46 2024 
Anshumali ShrivastavaRice University, ThirdAI Corp.Email verificata su rice.edu