English
Related papers

Related papers: Memory Analysis on the Training Course of DeepSeek…

200 papers

Addressing the so-called ``Red-AI'' trend of rising energy consumption by large-scale neural networks, this study investigates the actual energy consumption, as measured by node-level watt-meters, of training various fully connected neural…

Machine Learning · Computer Science 2024-03-14 Charles Edison Tripp , Jordan Perr-Sauer , Jamil Gafur , Amabarish Nag , Avi Purkayastha , Sagi Zisman , Erik A. Bensen

Distributed training frameworks, like TensorFlow, have been proposed as a means to reduce the training time of deep learning models by using a cluster of GPU servers. While such speedups are often desirable---e.g., for rapidly evaluating…

Performance · Computer Science 2019-05-07 Shijian Li , Robert J. Walls , Lijie Xu , Tian Guo

Distributed deep learning systems (DDLS) train deep neural network models by utilizing the distributed resources of a cluster. Developers of DDLS are required to make many decisions to process their particular workloads in their chosen…

Distributed, Parallel, and Cluster Computing · Computer Science 2020-07-09 Matthias Langer , Zhen He , Wenny Rahayu , Yanbo Xue

The advent of the transformer has sparked a quick growth in the size of language models, far outpacing hardware improvements. (Dense) transformers are expected to reach the trillion-parameter scale in the near future, for which training…

Machine Learning · Computer Science 2021-06-08 Joel Lamy-Poirier

Discrete optimization is a central problem in artificial intelligence. The optimization of the aggregated cost of a network of cost functions arises in a variety of problems including (W)CSP, DCOP, as well as optimization in stochastic…

Artificial Intelligence · Computer Science 2018-01-12 Ferdinando Fioretto , Enrico Pontelli , William Yeoh , Rina Dechter

GPUs have limited memory and it is difficult to train wide and/or deep models that cause the training process to go out of memory. It is shown in this paper how an open source tool called Large Model Support (LMS) can utilize a high…

Distributed, Parallel, and Cluster Computing · Computer Science 2018-11-30 Samuel Matzek , Max Grossman , Minsik Cho , Anar Yusifov , Bryant Nelson , Amit Juneja

In recommendation systems, practitioners observed that increase in the number of embedding tables and their sizes often leads to significant improvement in model performances. Given this and the business importance of these models to major…

Machine Learning · Computer Science 2020-10-26 Jie Amy Yang , Jianyu Huang , Jongsoo Park , Ping Tak Peter Tang , Andrew Tulloch

While larger neural models are pushing the boundaries of what deep learning can do, often more weights are needed to train models rather than to run inference for tasks. This paper seeks to understand this behavior using search spaces --…

Machine Learning · Computer Science 2021-05-28 Darko Stosic , Dusan Stosic

Efficient deployment of neural networks on resource-constrained hardware demands optimal use of on-chip memory. In event-based processors, this is particularly critical for routing architectures, where substantial memory is dedicated to…

Emerging Technologies · Computer Science 2024-12-03 Jimmy Weber , Theo Ballet , Melika Payvand

Generative models are increasingly used in 3D vision to synthesize novel shapes, yet it remains unclear whether their generation relies on memorizing training shapes. Understanding their memorization could help prevent training data leakage…

Computer Vision and Pattern Recognition · Computer Science 2025-12-30 Shu Pu , Boya Zeng , Kaichen Zhou , Mengyu Wang , Zhuang Liu

We present DeepSeek-V2, a strong Mixture-of-Experts (MoE) language model characterized by economical training and efficient inference. It comprises 236B total parameters, of which 21B are activated for each token, and supports a context…

Computation and Language · Computer Science 2024-06-21 DeepSeek-AI , Aixin Liu , Bei Feng , Bin Wang , Bingxuan Wang , Bo Liu , Chenggang Zhao , Chengqi Dengr , Chong Ruan , Damai Dai , Daya Guo , Dejian Yang , Deli Chen , Dongjie Ji , Erhang Li , Fangyun Lin , Fuli Luo , Guangbo Hao , Guanting Chen , Guowei Li , H. Zhang , Hanwei Xu , Hao Yang , Haowei Zhang , Honghui Ding , Huajian Xin , Huazuo Gao , Hui Li , Hui Qu , J. L. Cai , Jian Liang , Jianzhong Guo , Jiaqi Ni , Jiashi Li , Jin Chen , Jingyang Yuan , Junjie Qiu , Junxiao Song , Kai Dong , Kaige Gao , Kang Guan , Lean Wang , Lecong Zhang , Lei Xu , Leyi Xia , Liang Zhao , Liyue Zhang , Meng Li , Miaojun Wang , Mingchuan Zhang , Minghua Zhang , Minghui Tang , Mingming Li , Ning Tian , Panpan Huang , Peiyi Wang , Peng Zhang , Qihao Zhu , Qinyu Chen , Qiushi Du , R. J. Chen , R. L. Jin , Ruiqi Ge , Ruizhe Pan , Runxin Xu , Ruyi Chen , S. S. Li , Shanghao Lu , Shangyan Zhou , Shanhuang Chen , Shaoqing Wu , Shengfeng Ye , Shirong Ma , Shiyu Wang , Shuang Zhou , Shuiping Yu , Shunfeng Zhou , Size Zheng , T. Wang , Tian Pei , Tian Yuan , Tianyu Sun , W. L. Xiao , Wangding Zeng , Wei An , Wen Liu , Wenfeng Liang , Wenjun Gao , Wentao Zhang , X. Q. Li , Xiangyue Jin , Xianzu Wang , Xiao Bi , Xiaodong Liu , Xiaohan Wang , Xiaojin Shen , Xiaokang Chen , Xiaosha Chen , Xiaotao Nie , Xiaowen Sun , Xiaoxiang Wang , Xin Liu , Xin Xie , Xingkai Yu , Xinnan Song , Xinyi Zhou , Xinyu Yang , Xuan Lu , Xuecheng Su , Y. Wu , Y. K. Li , Y. X. Wei , Y. X. Zhu , Yanhong Xu , Yanping Huang , Yao Li , Yao Zhao , Yaofeng Sun , Yaohui Li , Yaohui Wang , Yi Zheng , Yichao Zhang , Yiliang Xiong , Yilong Zhao , Ying He , Ying Tang , Yishi Piao , Yixin Dong , Yixuan Tan , Yiyuan Liu , Yongji Wang , Yongqiang Guo , Yuchen Zhu , Yuduan Wang , Yuheng Zou , Yukun Zha , Yunxian Ma , Yuting Yan , Yuxiang You , Yuxuan Liu , Z. Z. Ren , Zehui Ren , Zhangli Sha , Zhe Fu , Zhen Huang , Zhen Zhang , Zhenda Xie , Zhewen Hao , Zhihong Shao , Zhiniu Wen , Zhipeng Xu , Zhongyu Zhang , Zhuoshu Li , Zihan Wang , Zihui Gu , Zilin Li , Ziwei Xie

Parallel training of neural networks at scale is challenging due to significant overheads arising from communication. Recently, deep learning researchers have developed a variety of pruning algorithms that are capable of pruning (i.e.…

Machine Learning · Computer Science 2023-05-16 Siddharth Singh , Abhinav Bhatele

Context: Along with developing Deep learning (DL) models, larger datasets and more complex model structures are applied, leading to rising computing resources and energy consumption, which is an alert that green DL models should receive…

Software Engineering · Computer Science 2026-03-09 Taoran Wang , Yanhui Li , Mingliang Ma , Lin Chen , Yuming Zhou

Large deep learning models offer significant accuracy gains, but training billions to trillions of parameters is challenging. Existing solutions such as data and model parallelisms exhibit fundamental limitations to fit these models into…

Machine Learning · Computer Science 2020-05-14 Samyam Rajbhandari , Jeff Rasley , Olatunji Ruwase , Yuxiong He

As machine learning spreads into more and more application areas, micro controllers and low power CPUs are increasingly being used to perform inference with machine learning models. The capability to deploy onto these limited hardware…

Distributed, Parallel, and Cluster Computing · Computer Science 2020-11-18 Peter Blacker , Christopher Paul Bridges , Simon Hadfield

The unprecedented growth of deep learning models has enabled remarkable advances but introduced substantial computational bottlenecks. A key factor contributing to training efficiency is batch-size and learning-rate scheduling in stochastic…

Machine Learning · Computer Science 2025-08-08 Hikaru Umeda , Hideaki Iiduka

Training Large Language Models(LLMs) is one of the most compute-intensive tasks in high-performance computing. Predicting end-to-end training time for multi-billion parameter models distributed across hundreds of GPUs remains challenging…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-09-30 Biyao Zhang , Mingkai Zheng , Debargha Ganguly , Xuecen Zhang , Vikash Singh , Vipin Chaudhary , Zhao Zhang

Large language models have led to state-of-the-art accuracies across a range of tasks. However, training these models efficiently is challenging for two reasons: a) GPU memory capacity is limited, making it impossible to fit large models on…

Recently, there has been a trend of shifting the execution of deep learning inference tasks toward the edge of the network, closer to the user, to reduce latency and preserve data privacy. At the same time, growing interest is being devoted…

Machine Learning · Computer Science 2023-06-07 Seyyidahmed Lahmer , Aria Khoshsirat , Michele Rossi , Andrea Zanella

Performance modelling of a deep learning application is essential to improve and quantify the efficiency of the model framework. However, existing performance models are mostly case-specific, with limited capability for the new deep…

Distributed, Parallel, and Cluster Computing · Computer Science 2023-05-22 Tulasi Kavarakuntla , Liangxiu Han , Huw Lloyd , Annabel Latham , Anthony Kleerekoper , Samson B. Akintoye