技术深度解析
这一突破背后的核心架构简洁而优雅:一个特征事件嵌入器与一个仅解码器Transformer配对。嵌入器接收多模态用户事件——产品ID、价格、时间戳、设备类型——并将它们投影到一个共享的潜在空间中。这至关重要,因为用户行为数据本质上是异质的。一个点击事件可能有50个分类特征,而一个支付事件可能有200个数值特征。嵌入器必须处理缺失值、可变长度特征集和高基数的分类变量(例如,数百万个唯一的产品ID)。
解码器Transformer随后处理这一序列的嵌入向量,以预测下一个事件。研究人员发现,扩展模型大小(层数、隐藏维度、注意力头数)和训练计算量(处理的token数、批次大小、训练步数)与损失之间遵循幂律关系,类似于Kaplan等人在语言模型中观察到的结果。具体来说,测试损失L的缩放关系为L ≈ a * C^(-b),其中C是计算量,b是缩放指数。
| 计算量 (FLOPs) | 模型大小 (参数量) | 下一事件准确率 | 训练数据 (事件数) |
|---|---|---|---|
| 1e18 | 50M | 72.3% | 100M |
| 1e19 | 200M | 78.1% | 500M |
| 1e20 | 800M | 83.5% | 2B |
| 1e21 | 3.2B | 87.2% | 10B |
数据要点: 准确率的提升在较低计算量区间最为显著,超过1e20 FLOPs后收益递减。这表明大多数生产系统的最佳计算预算在1e19到1e20 FLOPs之间,此时每FLOP的边际增益最高。
论文的一个关键见解是,缩放指数b取决于事件分布的熵。在高熵环境(例如,拥有数百万种产品的电商平台)中,b较小,意味着需要更多计算量才能获得相同的准确率提升。在低熵领域(例如,事件类型有限的订阅流失预测)中,缩放效率更高。这对资源分配有直接影响:长尾市场的推荐系统比处理有限交易类型的支付欺诈模型需要更多计算量。
在工程方面,研究人员在GitHub上以仓库`behavior-scaling`开源了一个参考实现。该仓库提供了一个使用PyTorch和Hugging Face Transformers库的模块化训练流水线。它包含可配置的嵌入器架构(MLP、TabTransformer或自定义),并通过DeepSpeed支持分布式训练。截至本文撰写时,该仓库已获得超过1200颗星,并得到积极维护,社区已衍生出多个分支,将其适配到广告定向和医疗事件预测等特定垂直领域。
关键参与者与案例研究
这项研究由一家中国大型科技公司AI实验室的团队领导,但论文的原则是供应商中立的。几家公司已经在将这些发现投入运营:
- 阿里巴巴: 其驱动淘宝和天猫的推荐引擎每天处理超过100亿用户事件。早期内部测试表明,将缩放定律应用于其行为模型后,A/B测试周期减少了40%——他们现在可以在5%的误差范围内预测2倍计算量增加带来的性能提升。
- 蚂蚁集团: 其用于实时欺诈检测的支付风险评分系统采用了双组件架构。通过将模型从1亿参数扩展到5亿参数,他们在保持低于50毫秒的相同延迟预算的同时,将误报率降低了18%。
- 字节跳动: TikTok和抖音的母公司已将这种方法整合到其内容推荐流水线中。他们报告称,在根据缩放定律指导计算分配重新训练行为模型后,用户会话时长提升了12%。
| 公司 | 应用场景 | 之前模型大小 | 之后模型大小 | 性能提升 | 计算成本增加 |
|---|---|---|---|---|---|
| 阿里巴巴 | 电商推荐 | 200M | 800M | +8% 点击率 | 3.5倍 |
| 蚂蚁集团 | 支付欺诈 | 100M | 500M | -18% 误报率 | 4.0倍 |
| 字节跳动 | 内容推荐 | 150M | 600M | +12% 会话时长 | 3.8倍 |
数据要点: 计算量与性能比并非线性——4倍的计算量增加仅带来8-18%的改进。这凸显了使用缩放定律找到收益递减点的重要性,而非盲目扩展。
该领域的知名研究人员包括李伟博士,他2023年关于“神经事件嵌入”的论文为嵌入器设计奠定了基础;以及清华大学的陈宇教授,他一直积极倡导将缩放定律应用于非语言领域。他们的工作弥合了NLP社区对缩放的痴迷与行为AI实际需求之间的差距。
行业影响与市场动态
行为模型缩放定律的发现从多个方面重塑了竞争格局。首先,它使模型性能预测变得商品化。以前,构建推荐系统或欺诈检测模型就像一门艺术,依赖于直觉、反复试验和大量的A/B测试。现在,公司可以像语言模型团队那样,用数学精度规划计算预算。这降低了新进入者的门槛,他们现在可以更可靠地预测投资回报。
其次,它加速了向行为基础模型的转变。就像GPT和BERT成为NLP的基础模型一样,我们现在看到针对用户行为序列的预训练基础模型的出现。这些模型可以在大规模用户事件数据上进行预训练,然后针对特定任务(如推荐、欺诈检测或流失预测)进行微调。缩放定律使这种预训练策略在经济上可行,因为公司现在可以预测扩展预训练计算量带来的收益。
第三,它创造了新的市场动态。云服务提供商(如阿里云、AWS和Google Cloud)可能会推出针对行为AI工作负载优化的专用计算实例。同样,我们可能会看到专门的行为AI模型市场出现,公司可以在其中交易预训练的基础模型。
然而,也存在挑战。缩放定律依赖于高质量、大规模的数据集。对于许多公司来说,收集和标注数十亿用户事件在物流上很困难,而且在隐私方面存在风险。该论文承认,其发现适用于具有足够数据熵的环境;在数据稀疏或高度结构化的领域,缩放收益可能不那么明显。
展望未来,研究人员已经在探索多模态行为模型,这些模型可以处理文本、图像和事件序列。早期工作表明,缩放定律也适用于这些模型,尽管指数可能因模态而异。我们还可能看到行为AI与强化学习的融合,其中缩放定律指导探索-利用权衡。
对于从业者来说,信息很明确:行为AI正在成为一门计算科学。那些拥抱缩放定律的公司将能够更高效地分配资源,更快地迭代,并构建更好的用户体验。那些忽视它的公司则有可能在计算效率上落后。就像语言模型一样,赢家将是那些理解如何有效扩展的人。