Does Mixture-of-Experts Actually Help Inference on Consumer and Edge Hardware? An Empirical Study

· Source: cs.AI updates on arXiv.org · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Robotics & Autonomous Systems, Software Development & Engineering · Depth: Expert, extended

Summary

An empirical study investigated whether Mixture-of-Experts (MoE) language models offer inference advantages on consumer and edge hardware, specifically an Apple M2 Pro and an NVIDIA Jetson Orin Nano 8 GB. Benchmarking OLMoE-1B-7B (1.3 B active, 6.9 B total parameters) against three dense baselines (Llama-3.2-1B, Qwen2.5-1.5B, Gemma-2-2B) using llama.cpp, researchers measured throughput, peak memory, and energy consumption. On the M2 Pro, OLMoE achieved 114.5 tokens/sec, about 10% slower than Llama-3.2-1B (127.3 tokens/sec) but faster than other dense models, with 4.5 GB peak memory. However, on the Jetson Orin Nano, OLMoE performed significantly worse, reaching 22.9 tokens/sec (31% slower than Llama-3.2-1B's 33.2 tokens/sec) and consuming 0.65 Joules/token, 2.1 times higher than Llama-3.2-1B's 0.31 Joules/token. Its 7.9 GB peak memory pushed the 8 GB Jetson to its limit. Analysis revealed routing overhead was under 9% of MoE-block compute on Jetson, indicating memory bandwidth, total parameter footprint, and expert dispatch are the primary bottlenecks, not routing arithmetic.

Key takeaway

For Machine Learning Engineers deploying LLMs on edge devices with limited memory, you should prioritize dense models over Mixture-of-Experts (MoE) architectures at the OLMoE-1B-7B scale. MoE models like OLMoE-1B-7B exhibit significantly lower throughput and higher energy consumption on 8 GB Jetson-class hardware, primarily due to their large total parameter footprint and memory bandwidth demands, not routing overhead. Consider further quantization or cache-aware routing for MoE if quality gains are critical.

Key insights

On bandwidth-bound edge hardware, MoE inference cost tracks total parameters, not active ones, negating expected efficiency gains.

Principles

Method

The study empirically measured MoE and dense LLM inference on consumer and edge hardware, using llama.cpp, Q4_K_M quantization, and direct power rail readings for energy.

In practice

Topics

Code references

Best for: AI Engineer, NLP Engineer, Research Scientist, AI Scientist, Machine Learning Engineer, AI Hardware Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by cs.AI updates on arXiv.org.