苹果研究用AI 大模型让移动端CPU、GPU提高数倍至数十倍性能

与CPU和GPU的原始加载方法相比，推理速度分别提高了4-5倍和20-25倍

（映维网Nweon 2023年12月25日）苹果日前在一篇论文中介绍了一种可以在超出可用DRAM容量的设备运行大语言模型LLM方法。通过大幅提高内存使用效率，这项新研究将允许资源有限的设备运行2倍于DRAM的LLM。另外与CPU和GPU的原始加载方法相比，推理速度分别提高了4-5倍和20-25倍。

团队的方法包括构建一个与闪存行为相协调的推理成本模型，以在两个关键领域进行优化：减少从闪存传输的数据量和在更大，更连续的数据块中读取数据。在这个闪存信息框架中，苹果主要介绍了两种主要技术。

首先，windowing通过重用先前激活的神经元来策略性地减少数据传输；其次，针对闪存顺序数据访问强度量身定制的“row-column bundling”增加了从闪存读取的数据块的大小。

所述方法使得运行模型的大小达到可用DRAM的两倍。另外与CPU和GPU的原始加载方法相比，推理速度分别提高了4-5倍和20-25倍。同时，研究人员集成了稀疏感知、情景自适应加载和面向硬件的设计，从而为LLM在内存有限设备的有效推理铺平了道路。

相关论文：LLM in a flash: Efficient Large Language Model Inference with Limited Memory

https://paper.nweon.com/15070

尽管没有明确说明这项技术可用的设备，但随着苹果愈发重视全新的空间计算领域，我们或许可以想象这家公司未来将AI整合至诸如Vision Pro这样的资源有限设备。