SSD Offloading for LLM Mixture-of-Experts Weights Considered Harmful in Energy Efficiency

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Kyung, Kwanhee, Yun, Sungmin, Ahn, Jung Ho
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913982741217280
author Kyung, Kwanhee
Yun, Sungmin
Ahn, Jung Ho
author_facet Kyung, Kwanhee
Yun, Sungmin
Ahn, Jung Ho
contents Large Language Models (LLMs) applying Mixture-of-Experts (MoE) scale to trillions of parameters but require vast memory, motivating a line of research to offload expert weights from fast-but-small DRAM (HBM) to denser Flash SSDs. While SSDs provide cost-effective capacity, their read energy per bit is substantially higher than that of DRAM. This paper quantitatively analyzes the energy implications of offloading MoE expert weights to SSDs during the critical decode stage of LLM inference. Our analysis, comparing SSD, CPU memory (DDR), and HBM storage scenarios for models like DeepSeek-R1, reveals that offloading MoE weights to current SSDs drastically increases per-token-generation energy consumption (e.g., by up to ~12x compared to the HBM baseline), dominating the total inference energy budget. Although techniques like prefetching effectively hide access latency, they cannot mitigate this fundamental energy penalty. We further explore future technological scaling, finding that the inherent sparsity of MoE models could potentially make SSDs energy-viable if Flash read energy improves significantly, roughly by an order of magnitude.
format Preprint
id arxiv_https___arxiv_org_abs_2508_06978
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle SSD Offloading for LLM Mixture-of-Experts Weights Considered Harmful in Energy Efficiency
Kyung, Kwanhee
Yun, Sungmin
Ahn, Jung Ho
Hardware Architecture
Large Language Models (LLMs) applying Mixture-of-Experts (MoE) scale to trillions of parameters but require vast memory, motivating a line of research to offload expert weights from fast-but-small DRAM (HBM) to denser Flash SSDs. While SSDs provide cost-effective capacity, their read energy per bit is substantially higher than that of DRAM. This paper quantitatively analyzes the energy implications of offloading MoE expert weights to SSDs during the critical decode stage of LLM inference. Our analysis, comparing SSD, CPU memory (DDR), and HBM storage scenarios for models like DeepSeek-R1, reveals that offloading MoE weights to current SSDs drastically increases per-token-generation energy consumption (e.g., by up to ~12x compared to the HBM baseline), dominating the total inference energy budget. Although techniques like prefetching effectively hide access latency, they cannot mitigate this fundamental energy penalty. We further explore future technological scaling, finding that the inherent sparsity of MoE models could potentially make SSDs energy-viable if Flash read energy improves significantly, roughly by an order of magnitude.
title SSD Offloading for LLM Mixture-of-Experts Weights Considered Harmful in Energy Efficiency
topic Hardware Architecture
url https://arxiv.org/abs/2508.06978