跳到正文
arXiv cs.AI· Jinane Bazzi, Mariam Rakka, Fadi Kurdahi, Mohammed E. Fouda, Ahmed Eltawil·· 5 小时前AI 评分28

EdgeCIM:面向小语言模型 CIM 加速的软硬件协同设计

EdgeCIM: A Hardware-Software Co-Design for CIM-Based Acceleration of Small Language Models

AI 导读

EdgeCIM 是一个用于在边缘设备上加速小语言模型(SLM)自回归解码的软硬件协同设计框架,核心是 65 nm 实现的 CIM macro 与基于 tile 的映射策略,支持最高 4B 参数模型的设计空间探索。

正文 · 原文

View PDF HTML (experimental)

Abstract:The growing demand for deploying Small Language Models (SLMs) on edge devices, including laptops, smartphones, automotive and robotic computing platforms, and high-performance embedded systems, has exposed fundamental inefficiencies in existing accelerators. While GPUs handle prefill workloads efficiently, the autoregressive decoding phase is dominated by GEMV operations that are inherently memory-bound, resulting in poor utilization and prohibitive energy costs at the edge. In this work, we present EdgeCIM, a hardware-software co-design framework for accelerating autoregressive decoding of SLMs on edge devices in latency-critical applications. At its core is a CIM macro, implemented in 65 nm, coupled with a tile-based mapping strategy that balances pipeline stages, maximizing parallelism while alleviating DRAM bandwidth bottlenecks. Our simulator enables design space exploration of SLMs up to 4B parameters, identifying hardware configurations that optimize latency and energy according to a configurable objective function. On LLaMA3.2 3B (INT4), EdgeCIM achieves 1.76x and 3.21x higher throughput than NVIDIA Jetson AGX Orin and Orin Nano Super operating in MAXN mode, respectively, and 7.07x higher throughput than Qualcomm's Snapdragon-X Elite. Extensive benchmarks on TinyLLaMA 1.1B, LLaMA3.2 (1B, 3B), Phi 3.5 mini 3.8B, Qwen2.5 (0.5B, 1.5B, 3B), SmolLM2 1.7B, SmolLM3 3B, and Qwen3 (0.6B, 1.7B, 4B) reveal that our accelerator, at INT4 precision, achieves on average 336.4 tokens/s and 173 tokens/J. These results establish EdgeCIM as a compelling solution towards real-time, energy-efficient, latency-critical SLM inference at the edge.
Comments: Accepted at IEEE TCAD
Subjects: Hardware Architecture (cs.AR); Artificial Intelligence (cs.AI)
Cite as: arXiv:2604.11512 [cs.AR]
  (or arXiv:2604.11512v2 [cs.AR] for this version)
  https://doi.org/10.48550/arXiv.2604.11512

arXiv-issued DOI via DataCite

Submission history

From: Mohammed Fouda [view email]
[v1] Mon, 13 Apr 2026 14:16:20 UTC (1,319 KB)
[v2] Tue, 27 Oct 2026 19:36:31 UTC (2,466 KB)

来源:arXiv cs.AI · arxiv.org