中文
相关论文

相关论文: NEO: Saving GPU Memory Crisis with CPU Offloading …

200 篇论文

As large language models (LLMs) continue to scale, the high power consumption of AI accelerators in datacenters presents significant challenges, substantially increasing the total cost of ownership (TCO) for cloud service providers (CSPs)…

机器学习 · 计算机科学 2025-08-26 Jiwoo Kim , Joonhyung Lee , Gunho Park , Byeongwook Kim , Se Jung Kwon , Dongsoo Lee , Youngjoo Lee

Fine-tuning large language models (LLMs) often exceeds GPU memory limits, prompting systems to offload model states to CPU memory. However, existing offloaded training frameworks like ZeRO-Offload treat all parameters equally and update the…

分布式、并行与集群计算 · 计算机科学 2025-08-06 Tingfeng Lan , Yusen Wu , Bin Ma , Zhaoyuan Su , Rui Yang , Tekin Bicer , Masahiro Tanaka , Olatunji Ruwase , Dong Li , Yue Cheng

Large Language Models (LLMs), despite their remarkable performance across a wide range of tasks, necessitate substantial GPU memory and consume significant computational resources. Beyond the memory taken up by model weights, the memory…

计算与语言 · 计算机科学 2024-06-24 Jincheng Dai , Zhuowei Huang , Haiyun Jiang , Chen Chen , Deng Cai , Wei Bi , Shuming Shi

Offloading large language models (LLMs) state to host memory during inference promises to reduce operational costs by supporting larger models, longer inputs, and larger batch sizes. However, the design of existing memory offloading…

分布式、并行与集群计算 · 计算机科学 2025-02-13 Chenxiang Ma , Zhisheng Ye , Hanyu Zhao , Zehua Yang , Tianhao Fu , Jiaxun Han , Jie Zhang , Yingwei Luo , Xiaolin Wang , Zhenlin Wang , Yong Li , Diyu Zhou

Fine-tuning pre-trained large language models (LLMs) with limited hardware presents challenges due to GPU memory constraints. Various distributed fine-tuning methods have been proposed to alleviate memory constraints on GPU. However,…

人工智能 · 计算机科学 2024-04-18 Taeho Kim , Yanming Wang , Vatshank Chaturvedi , Lokesh Gupta , Seyeon Kim , Yongin Kwon , Sangtae Ha

Mixture-of-Experts (MoE), while offering significant advantages as a Large Language Model (LLM) architecture, faces substantial challenges when deployed on low-cost edge devices with tight memory constraints. Expert offloading mitigates…

分布式、并行与集群计算 · 计算机科学 2025-12-04 Liujianfu Wang , Yuyang Du , Yuchen Pan , Soung Chang Liew , Jiacheng Liu , Kexin Chen

Energy is now a critical ML computing resource. While measuring energy consumption and observing trends is a valuable first step, accurately understanding and diagnosing why those differences occur is crucial for optimization. To that end,…

机器学习 · 计算机科学 2026-02-02 Jae-Won Chung , Ruofan Wu , Jeff J. Ma , Mosharaf Chowdhury

Large Language Models (LLMs), despite their recent impressive accomplishments, are notably cost-prohibitive to deploy, particularly for applications involving long-content generation, such as dialogue systems and story writing. Often, a…

This paper presents ServerlessLLM, a distributed system designed to support low-latency serverless inference for Large Language Models (LLMs). By harnessing the substantial near-GPU storage and memory capacities of inference servers,…

机器学习 · 计算机科学 2024-07-26 Yao Fu , Leyang Xue , Yeqi Huang , Andrei-Octavian Brabete , Dmitrii Ustiugov , Yuvraj Patel , Luo Mai

Deploying large language models (LLMs) on mobile devices increasingly relies on heterogeneous execution, yet no prior study has systematically characterized NPU effectiveness at the operator and pipeline level. We present the first…

硬件体系结构 · 计算机科学 2026-05-28 Pu Li , Jiawen Qi , Qinyu Chen

Large language models have demonstrated extraordinary performance in many AI tasks but are expensive to use, even after training, due to their requirement of high-end GPUs. Recently, a distributed system called PETALS was developed to lower…

分布式、并行与集群计算 · 计算机科学 2025-12-30 Tingyang Sun , Ting He , Bo Ji , Parimal Parag

RAPID-LLM is a unified performance modeling framework for large language model (LLM) training and inference on GPU clusters. It couples a DeepFlow-based frontend that generates hardware-aware, operator-level Chakra execution traces from an…

Throughout its lifecycle, a large language model (LLM) generates a substantially larger carbon footprint during inference than training. LLM inference requests vary in batch size, prompt length, and token generation number, while cloud…

机器学习 · 计算机科学 2024-10-07 Zhenxiao Fu , Fan Chen , Shan Zhou , Haitong Li , Lei Jiang

This paper introduces PowerInfer, a high-speed Large Language Model (LLM) inference engine on a personal computer (PC) equipped with a single consumer-grade GPU. The key principle underlying the design of PowerInfer is exploiting the high…

机器学习 · 计算机科学 2024-12-13 Yixin Song , Zeyu Mi , Haotong Xie , Haibo Chen

There is an urgent and pressing need to optimize usage of Graphical Processing Units (GPUs), which have arguably become one of the most expensive and sought after IT resources. To help with this goal, several of the current generation of…

分布式、并行与集群计算 · 计算机科学 2024-09-11 Bekir Turkkan , Pavankumar Murali , Pavithra Harsha , Rohan Arora , Gerard Vanloo , Chandra Narayanaswami

Large language models (LLMs) have demonstrated exceptional performance across a variety of tasks. However, their substantial scale leads to significant computational resource consumption during inference, resulting in high costs.…

机器学习 · 计算机科学 2025-06-13 Zhaode Wang , Jingbang Yang , Xinyu Qian , Shiwen Xing , Xiaotang Jiang , Chengfei Lv , Shengyu Zhang

The rapid adoption of large language models (LLMs) has led to significant advances in natural language processing and text generation. However, the energy consumed through LLM model inference remains a major challenge for sustainable AI…

分布式、并行与集群计算 · 计算机科学 2024-07-08 Grant Wilkins , Srinivasan Keshav , Richard Mortier

Ensembles of Deep Neural Networks (DNNs) have achieved qualitative predictions but they are computing and memory intensive. Therefore, the demand is growing to make them answer a heavy workload of requests with available computational…

分布式、并行与集群计算 · 计算机科学 2022-08-31 Pierrick Pochelu , Serge G. Petiton , Bruno Conche

Efficient LLM inference is critical for real-world applications, especially within heterogeneous GPU clusters commonly found in organizations and on-premise datacenters as GPU architecture rapidly evolves. Current disaggregated prefill…

分布式、并行与集群计算 · 计算机科学 2025-09-23 Yunzhao Liu , Qiang Xu , Y. Charlie Hu

By exploiting the superiority of non-orthogonal multiple access (NOMA), NOMA-aided mobile edge computing (MEC) can provide scalable and low-latency computing services for the Internet of Things. However, given the prevalent stochasticity of…

信息论 · 计算机科学 2021-07-01 Meihui Hua , Hui Tian , Xinchen Lyu , Wanli Ni , Gaofeng Nie