
阶跃星辰的AI手机:端侧大模型推理的工程赌局
这篇文章最有价值的信息是:阶跃星辰选择以自研硬件+操作系统的重资产模式切入AI终端,而非单纯做APP或云API。对于关注端侧大模型落地的工程师来说,这标志着从“能不能跑”到“跑得值不值”的工程阶段切换——印奇赌的是,在手机这个功耗和散热极度受限的TDP(热设计功耗)窗口内,大模型推理的实时性和可用性可以通过软硬协同优化达到用户可接受的阈值。
端侧推理的硬约束:内存带宽与算子融合
阶跃STEPX Neo手机的核心卖点是“大模型原生智能体”,这意味着它需要在本地持续运行一个7B至13B参数规模的transformer模型。当前手机端侧推理的最大瓶颈并非算力(TOPS),而是内存带宽。以骁龙8 Gen 3为例,其内存带宽约80GB/s,而一个7B模型在FP16精度下单次前向传播需要加载约14GB参数,仅参数传输就需要约175ms。加上KV cache和激活值,单次推理延迟轻松超过200ms,这对交互式对话(如语音助手)是致命的。
因此,工程落地必须做三件事:
- 量化:从FP16压缩到INT4甚至INT2,将参数体积降至1/4到1/8,但精度损失需要校准。
- 算子融合:将attention中的QKV投影、softmax、输出投影等连续小算子合并为更大的kernel,减少内存访问次数。例如,fused attention kernel可以将一次推理的带宽需求降低30%-40%。
- 模型剪枝/蒸馏:去除冗余注意力头或层,但阶跃若想保持“原生”能力,剪枝空间有限。
[!note]
实操层面,阶跃需要自研一套端侧推理引擎,类似Apple的CoreML或高通的SNPE,但必须针对自家的模型架构做定制化算子库。这比写一个通用的LLM推理框架(如llama.cpp)要复杂一个数量级,因为需要和手机SoC的NPU/DSP/GPU做底层调度竞争。
从IR优化到硬件适配:AOS的工程重心
阶跃发布的“Step AOS”智能体原生操作系统,本质上是一个大模型推理中间件。作为编译器研发者,我关注的是它的IR(中间表示)设计。传统的AI编译器(如TVM、MLIR)通常以计算图级别的IR作为优化入口,但大模型推理的特点是动态shape(输入token长度变化)和条件分支(如if语句触发的不同推理路径)。Step AOS如果能在IR层支持动态shape的自动调度,比如根据当前KV cache长度自动选择最优的矩阵分块策略,就算有了真正的技术壁垒。
配图插入位置:
另一个关键点是异构计算调度。手机上有CPU、GPU、NPU,每个单元的内存带宽和延迟特性不同。例如,注意力计算适合NPU的矩阵乘法加速,而softmax和layer norm操作在CPU上延迟更低。Step AOS需要实现一个运行时调度器,根据当前模型层和系统负载动态分配计算单元。这需要深入芯片底层,获取每个加速器的实时状态——没有芯片厂商的深度合作,很难做到。
一个编译器工程师的评估:护城河在哪?
从技术可行性看,阶跃的赌局并非天方夜谭。当前端侧大模型推理已进入可用阶段(例如Google的Gemini Nano在Pixel 8上运行)。但阶跃面临的挑战更严峻:
- 硬件不确定性:印奇选择与手机厂商合作(传闻是OPPO或一加),而非自研芯片。这意味着他们必须适配不同SoC的NPU指令集,无法像苹果那样软硬一体优化。每个算子在不同NPU上的实现差异极大,维护成本高。
- 模型迭代速度:大模型参数每几个月翻倍,而手机硬件迭代周期是18个月。端侧模型被迫在固定算力下不断更新,这要求推理引擎能快速适配新模型结构(如GQA、MOE),而不仅仅是参数压缩。
- 用户预期管理:用户习惯云端大模型的接近实时响应,而端侧推理哪怕做到200ms延迟,与云端50ms相比仍有差距。如果阶跃的智能体频繁出现“思考中”卡顿,用户会立刻放弃。
我的判断:阶跃STEPX Neo短期内不会成为主流,但可以作为技术验证机。真正的价值在于Step AOS的工程积累——如果它能将端侧推理延迟降低到100ms以内,并支持主流模型格式,那么这套系统可以授权给其他手机厂商,成为AI手机时代的“Android”。而印奇真正在赌的,不是卖手机,而是通过终端反哺算力,闭环智能体场景的数据飞轮。
行动建议
给关注端侧AI的工程师:建议重点研究阶跃的量化策略和动态调度算法。去GitHub上找他们开源的端侧推理代码(如果有的话),关注其如何在资源受限下保持模型“智能”而非“智障”。如果你正在做手机端AI应用
原文链接:https://www.tmtpost.com/8071509.html
物界前沿