English
Related papers

Related papers: Accurate GPU Memory Prediction for Deep Learning J…

200 papers

One of the most efficient methods to solve L2-regularized primal problems, such as logistic regression and linear support vector machine (SVM) classification, is the widely used trust region Newton algorithm, TRON. While TRON has recently…

Machine Learning · Computer Science 2020-10-16 John T. Halloran , David M. Rocke

Stochastic nonlinear dynamical systems are ubiquitous in modern, real-world applications. Yet, estimating the unknown parameters of stochastic, nonlinear dynamical models remains a challenging problem. The majority of existing methods…

Machine Learning · Statistics 2022-05-06 Anubhab Ghosh , Mohamed Abdalmoaty , Saikat Chatterjee , Håkan Hjalmarsson

With the increase in the complexity of chip designs, VLSI physical design has become a time-consuming task, which is an iterative design process. Power planning is that part of the floorplanning in VLSI physical design where power grid…

Machine Learning · Computer Science 2020-07-27 Sukanta Dey , Sukumar Nandi , Gaurav Trivedi

Deep learning training at scale is resource-intensive and time-consuming, often running across hundreds or thousands of GPUs for weeks or months. Efficient checkpointing is crucial for running these workloads, especially in multi-tenant…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-02-25 Radostin Stoyanov , Viktória Spišaková , Jesus Ramos , Steven Gurfinkel , Andrei Vagin , Adrian Reber , Wesley Armour , Rodrigo Bruno

Memory units have been widely used to enrich the capabilities of deep networks on capturing long-term dependencies in reasoning and prediction tasks, but little investigation exists on deep generative models (DGMs) which are good at…

Machine Learning · Computer Science 2016-05-31 Chongxuan Li , Jun Zhu , Bo Zhang

Security assessment is among the most fundamental functions of power system operator. The sheer complexity of power systems exceeding a few buses, however, makes it an extremely computationally demanding task. The emergence of deep learning…

Machine Learning · Computer Science 2019-04-22 José-María Hidalgo-Arteaga , Fiodar Hancharou , Florian Thams , Spyros Chatzivasileiadis

The ability to learn continually without forgetting the past tasks is a desired attribute for artificial learning systems. Existing approaches to enable such learning in artificial neural networks usually rely on network growth, importance…

Machine Learning · Computer Science 2021-03-18 Gobinda Saha , Isha Garg , Kaushik Roy

The performance of deep learning algorithms such as neural networks (NNs) has increased tremendously recently, and they can achieve state-of-the-art performance in many domains. However, due to memory and computation resource constraints,…

Machine Learning · Computer Science 2024-05-30 Soyed Tuhin Ahmed , Mehdi Tahoori

Fine-tuning large language models is a popular choice among users trying to adapt them for specific applications. However, fine-tuning these models is a demanding task because the user has to examine several factors, such as resource…

Machine Learning · Computer Science 2024-06-07 Arjun Singh , Nikhil Pandey , Anup Shirgaonkar , Pavan Manoj , Vijay Aski

Deep learning (DL) research yields accuracy and product improvements from both model architecture changes and scale: larger data sets and models, and more computation. For hardware design, it is difficult to predict DL model changes.…

Machine Learning · Computer Science 2019-09-05 Joel Hestness , Newsha Ardalani , Greg Diamos

In this research. we analyze the potential of Feature Density (HD) as a way to comparatively estimate machine learning (ML) classifier performance prior to training. The goal of the study is to aid in solving the problem of…

Computation and Language · Computer Science 2022-06-07 Juuso Eronen , Michal Ptaszynski , Fumito Masui , Gniewosz Leliwa , Michal Wroczynski

Kubernetes offers a powerful orchestration platform for machine learning training, but memory management can be challenging due to specialized needs and resource constraints. This paper outlines how Kubernetes handles memory requests,…

Distributed, Parallel, and Cluster Computing · Computer Science 2024-12-30 Jaideep Ray

Accommodating long-running deep learning (DL) training and inference jobs is challenging on GPU clusters that use traditional batch schedulers, such as Slurm. Given fixed wall clock time limits, DL researchers usually need to run a sequence…

Distributed, Parallel, and Cluster Computing · Computer Science 2023-06-27 Qiyang Ding , Pengfei Zheng , Shreyas Kudari , Shivaram Venkataraman , Zhao Zhang

Large language models (LLMs) are highly compute- and memory-intensive, posing significant demands on high-performance GPUs. At the same time, advances in GPU technology driven by shrinking transistor sizes and lower operating voltages have…

Hardware Architecture · Computer Science 2026-01-29 Duo Chai , Zizhen Liu , Shuhuai Wang , Songwei Pei , Cheng Liu , Huawei Li , Shangguang Wang

Deep learning (DL) frameworks take advantage of GPUs to improve the speed of DL inference and training. Ideally, DL frameworks should be able to fully utilize the computation power of GPUs such that the running time depends on the amount of…

Machine Learning · Computer Science 2020-12-07 Woosuk Kwon , Gyeong-In Yu , Eunji Jeong , Byung-Gon Chun

The training scale of large language models (LLMs) has reached tens of thousands of GPUs and is still continuously expanding, enabling faster learning of larger models. Accompanying the expansion of the resource scale is the prevalence of…

Training Large Language Models (LLMs) on long contexts is severely constrained by prohibitive GPU memory overhead, not training time. The primary culprits are the activations, whose memory footprints scale linearly with sequence length. We…

Computation and Language · Computer Science 2026-03-03 Wenhao Li , Daohai Yu , Gen Luo , Yuxin Zhang , Fei Chao , Rongrong Ji , Yifan Wu , Jiaxin Liu , Ziyang Gong , Zimu Liao

Edge intelligent applications like VR/AR and language model based chatbots have become widespread with the rapid expansion of IoT and mobile devices. However, constrained edge devices often cannot serve the increasingly large and complex…

Distributed, Parallel, and Cluster Computing · Computer Science 2025-10-28 Zongshun Zhang , Ibrahim Matta

One major obstacle towards AI is the poor ability of models to solve new problems quicker, and without forgetting previously acquired knowledge. To better understand this issue, we study the problem of continual learning, where the model…

Machine Learning · Computer Science 2022-09-14 David Lopez-Paz , Marc'Aurelio Ranzato

Deep learning (DL) defines a new data-driven programming paradigm that constructs the internal system logic of a crafted neuron network through a set of training data. We have seen wide adoption of DL in many safety-critical scenarios.…

Software Engineering · Computer Science 2018-08-16 Lei Ma , Felix Juefei-Xu , Fuyuan Zhang , Jiyuan Sun , Minhui Xue , Bo Li , Chunyang Chen , Ting Su , Li Li , Yang Liu , Jianjun Zhao , Yadong Wang