Software-hardware co-design for fast and scalable training of deep learning recommendation models

Author:

Mudigere Dheevatsa1,Hao Yuchen1,Huang Jianyu1,Jia Zhihao2,Tulloch Andrew1,Sridharan Srinivas1,Liu Xing1,Ozdal Mustafa1,Nie Jade1,Park Jongsoo1,Luo Liang1,Yang Jie (Amy)1,Gao Leon1,Ivchenko Dmytro1,Basant Aarti1,Hu Yuxi1,Yang Jiyan1,Ardestani Ehsan K.1,Wang Xiaodong1,Komuravelli Rakesh1,Chu Ching-Hsiang1,Yilmaz Serhat1,Li Huayu1,Qian Jiyuan1,Feng Zhuobo1,Ma Yinbin1,Yang Junjie1,Wen Ellie1,Li Hong1,Yang Lin1,Sun Chonglin1,Zhao Whitney1,Melts Dimitry1,Dhulipala Krishna1,Kishore KR1,Graf Tyler1,Eisenman Assaf1,Matam Kiran Kumar1,Gangidi Adi1,Chen Guoqiang Jerry1,Krishnan Manoj1,Nayak Avinash1,Nair Krishnakumar1,Muthiah Bharath1,khorashadi Mahmoud1,Bhattacharya Pallab1,Lapukhov Petr1,Naumov Maxim1,Mathews Ajit1,Qiao Lin1,Smelyanskiy Mikhail1,Jia Bill1,Rao Vijay1

Affiliation:

1. Meta Platforms

2. Carnegie Mellon University

Publisher

ACM

Reference71 articles.

1. [n.d.]. NVIDIA Collective Communications Library (NCCL) https://developer.nvidia.com/nccl. [n.d.]. NVIDIA Collective Communications Library (NCCL) https://developer.nvidia.com/nccl.

2. [n.d.]. OCP Open rack standard (v2) https://www.opencompute.org/wiki/Open_Rack/SpecsAndDesigns#RACK_Standards. [n.d.]. OCP Open rack standard (v2) https://www.opencompute.org/wiki/Open_Rack/SpecsAndDesigns#RACK_Standards.

3. Martín Abadi Ashish Agarwal Paul Barham Eugene Brevdo Zhifeng Chen Craig Citro Greg S. Corrado Andy Davis Jeffrey Dean Matthieu Devin Sanjay Ghemawat Ian Goodfellow Andrew Harp Geoffrey Irving Michael Isard Yangqing Jia Rafal Jozefowicz Lukasz Kaiser Manjunath Kudlur Josh Levenberg Dandelion Mané Rajat Monga Sherry Moore Derek Murray Chris Olah Mike Schuster Jonathon Shlens Benoit Steiner Ilya Sutskever Kunal Talwar Paul Tucker Vincent Vanhoucke Vijay Vasudevan Fernanda Viégas Oriol Vinyals Pete Warden Martin Wattenberg Martin Wicke Yuan Yu and Xiaoqiang Zheng. 2015. TensorFlow: Large-Scale Machine Learning on Heterogeneous Systems. https://www.tensorflow.org/ Software available from tensorflow.org. Martín Abadi Ashish Agarwal Paul Barham Eugene Brevdo Zhifeng Chen Craig Citro Greg S. Corrado Andy Davis Jeffrey Dean Matthieu Devin Sanjay Ghemawat Ian Goodfellow Andrew Harp Geoffrey Irving Michael Isard Yangqing Jia Rafal Jozefowicz Lukasz Kaiser Manjunath Kudlur Josh Levenberg Dandelion Mané Rajat Monga Sherry Moore Derek Murray Chris Olah Mike Schuster Jonathon Shlens Benoit Steiner Ilya Sutskever Kunal Talwar Paul Tucker Vincent Vanhoucke Vijay Vasudevan Fernanda Viégas Oriol Vinyals Pete Warden Martin Wattenberg Martin Wicke Yuan Yu and Xiaoqiang Zheng. 2015. TensorFlow: Large-Scale Machine Learning on Heterogeneous Systems. https://www.tensorflow.org/ Software available from tensorflow.org.

4. Bilge Acun Matthew Murphy Xiaodong Wang Jade Nie Carole-Jean Wu and Kim Hazelwood. 2020. Understanding Training Efficiency of Deep Learning Recommendation Models at Scale. arXiv:2011.05497 [cs.AR] Bilge Acun Matthew Murphy Xiaodong Wang Jade Nie Carole-Jean Wu and Kim Hazelwood. 2020. Understanding Training Efficiency of Deep Learning Recommendation Models at Scale. arXiv:2011.05497 [cs.AR]

5. Tom B. Brown Benjamin Mann Nick Ryder Melanie Subbiah Jared Kaplan Prafulla Dhariwal Arvind Neelakantan Pranav Shyam Girish Sastry Amanda Askell Sandhini Agarwal Ariel Herbert-Voss Gretchen Krueger Tom Henighan Rewon Child Aditya Ramesh Daniel M. Ziegler Jeffrey Wu Clemens Winter Christopher Hesse Mark Chen Eric Sigler Mateusz Litwin Scott Gray Benjamin Chess Jack Clark Christopher Berner Sam McCandlish Alec Radford Ilya Sutskever and Dario Amodei. 2020. Language Models are Few-Shot Learners. arXiv:2005.14165 [cs.CL] Tom B. Brown Benjamin Mann Nick Ryder Melanie Subbiah Jared Kaplan Prafulla Dhariwal Arvind Neelakantan Pranav Shyam Girish Sastry Amanda Askell Sandhini Agarwal Ariel Herbert-Voss Gretchen Krueger Tom Henighan Rewon Child Aditya Ramesh Daniel M. Ziegler Jeffrey Wu Clemens Winter Christopher Hesse Mark Chen Eric Sigler Mateusz Litwin Scott Gray Benjamin Chess Jack Clark Christopher Berner Sam McCandlish Alec Radford Ilya Sutskever and Dario Amodei. 2020. Language Models are Few-Shot Learners. arXiv:2005.14165 [cs.CL]

Cited by 56 articles. 订阅此论文施引文献 订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献

1. Rail-only: A Low-Cost High-Performance Network for Training LLMs with Trillion Parameters;2024 IEEE Symposium on High-Performance Interconnects (HOTI);2024-08-21

2. RDMA over Ethernet for Distributed Training at Meta Scale;Proceedings of the ACM SIGCOMM 2024 Conference;2024-08-04

3. Breaking Barriers: Expanding GPU Memory with Sub-Two Digit Nanosecond Latency CXL Controller;Proceedings of the 16th ACM Workshop on Hot Topics in Storage and File Systems;2024-07-08

4. MAD-Max Beyond Single-Node: Enabling Large Machine Learning Model Acceleration on Distributed Systems;2024 ACM/IEEE 51st Annual International Symposium on Computer Architecture (ISCA);2024-06-29

5. ElasticRec: A Microservice-based Model Serving Architecture Enabling Elastic Resource Scaling for Recommendation Models;2024 ACM/IEEE 51st Annual International Symposium on Computer Architecture (ISCA);2024-06-29

同舟云学术

1.学者识别学者识别

2.学术分析学术分析

3.人才评估人才评估

"同舟云学术"是以全球学者为主线,采集、加工和组织学术论文而形成的新型学术文献查询和分析系统,可以对全球学者进行文献检索和人才价值评估。用户可以通过关注某些学科领域的顶尖人物而持续追踪该领域的学科进展和研究前沿。经过近期的数据扩容,当前同舟云学术共收录了国内外主流学术期刊6万余种,收集的期刊论文及会议论文总量共计约1.5亿篇,并以每天添加12000余篇中外论文的速度递增。我们也可以为用户提供个性化、定制化的学者数据。欢迎来电咨询!咨询电话:010-8811{复制后删除}0370

www.globalauthorid.com

TOP

Copyright © 2019-2024 北京同舟云网络信息技术有限公司
京公网安备11010802033243号  京ICP备18003416号-3