一句话定位
Apple Intelligence 基础模型技术报告:一个约 3B 的端侧模型 + 一个用于 Private Cloud Compute 的服务器模型,详述架构、数据、训练、推理优化与负责任 AI。
摘要
本文介绍为 Apple Intelligence 功能打造的基础语言模型:一个约 30 亿参数、为端侧高效运行设计的模型(AFM-on-device),以及一个为 Private Cloud Compute 设计的大服务器模型(AFM-server)。模型旨在高效、准确、负责任地完成广泛任务。报告描述模型架构、训练数据、训练过程、推理优化、评测结果,并强调负责任 AI 原则贯穿全程。
关键技术细节
- AFM-on-device:约 3B 参数;为内存/延迟优化(GQA、共享 input/output embedding、low-bit palettization 量化)。
- AFM-server:大模型,跑在 Private Cloud Compute(隐私保护推理)。
- 数据:授权出版数据 + AppleBot 网页 + 代码/数学/合成;强调不使用用户私有数据。
- 训练 infra:用 AXLearn(基于 JAX)在 TPU 上训练;AFM-server 用 8192 TPUv4,AFM-on-device 用 2048 TPUv5p。
- 后训练:SFT + RLHF,提出两种新算法 iTeC(迭代教学委员会)与 MDLOO(含 leave-one-out 优势估计的 online RL)。
- 适配:用 LoRA adapter 为不同功能动态切换(端侧按任务加载 adapter)。
原始链接
一手源存档(sources/)
- 2407.21075.pdf (arXiv 原文 PDF,不入 git)