测试小模型在边缘设备上运行的可行性与性能代价
Sometimes Painful but Certainly Promising: Feasibility and Trade-offs of Language Model Inference at the Edge
- 在典型边缘设备上评估小型语言模型的推理表现
- 发现量化可减内存但无法完全解决资源瓶颈
- 适合关注隐私、低延迟的边缘AI部署者参考
语言模型的快速发展拓展了自然语言处理的能力,从文本生成到复杂决策均有应用。尽管当前主流模型参数量达数百亿并集中部署于数据中心,但近年来小型模型(通常低于100亿参数)因量化等压缩技术的推动正逐步走向边缘设备,带来隐私增强、延迟降低和数据主权提升等潜力。然而,即使小型模型仍面临计算复杂性与边缘硬件算力有限之间的矛盾,引发对云外推理实际可行性的疑问。为此,本文系统评估了代表性基于CPU与GPU加速的边缘设备上生成式语言模型的推理表现,测量内存占用、推理速度、能耗等关键指标,并分析吞吐-能效权衡、成本与可用性,同时考察模型质量。结果表明,量化虽缓解内存压力,却未能彻底消除资源瓶颈,尤其对较大模型而言。研究量化了实际部署中必须考虑的内存与能耗约束,揭示了模型规模、推理性能与效率间的权衡关系。边缘语言模型探索尚处初期,本研究为未来模型优化、推理效率提升及以边缘为中心的AI系统发展提供基础参考。
原文摘要 · Abstract (English)
The rapid rise of Language Models (LMs) has expanded the capabilities of natural language processing, powering applications from text generation to complex decision-making. While state-of-the-art LMs often boast hundreds of billions of parameters and are primarily deployed in data centers, recent trends show a growing focus on compact models-typically under 10 billion parameters-enabled by techniques such as quantization and other model compression techniques. This shift paves the way for LMs on edge devices, offering potential benefits such as enhanced privacy, reduced latency, and improved data sovereignty. However, the inherent complexity of even these smaller models, combined with the limited computing resources of edge hardware, raises critical questions about the practical trade-offs in executing LM inference outside the cloud. To address these challenges, we present a comprehensive evaluation of generative LM inference on representative CPU-based and GPU-accelerated edge devices. Our study measures key performance indicators-including memory usage, inference speed, and energy consumption-across various device configurations. Additionally, we examine throughput-energy trade-offs, cost considerations, and usability, alongside an assessment of qualitative model performance. While quantization helps mitigate memory overhead, it does not fully eliminate resource bottlenecks, especially for larger models. Our findings quantify the memory and energy constraints that must be considered for practical real-world deployments, offering concrete insights into the trade-offs between model size, inference performance, and efficiency. The exploration of LMs at the edge is still in its early stages. We hope this study provides a foundation for future research, guiding the refinement of models, the enhancement of inference efficiency, and the advancement of edge-centric AI systems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。