华为推出改进版Transformer架构，盘古-π解决特征缺陷问题，带来超越LLaMA的同等规模性能

从现在开始，我们要努力学习啦！今天我给大家带来《华为推出改进版Transformer架构，盘古-π解决特征缺陷问题，带来超越LLaMA的同等规模性能》，感兴趣的朋友请继续看下去吧！下文中的内容我们主要会涉及到等等知识点，如果在阅读本文过程中有遇到不清楚的地方，欢迎留言呀！我们一起讨论，一起学习！

华为盘古系列是一个带来全新架构层面的产品系列。

量子位获悉，华为诺亚方舟实验室等联合推出新型大语言模型架构：盘古-π。

通过在传统Transformer架构上增强非线性性能，该方法做出了改进，从而有效地减轻了特征塌陷问题。

带来的直接结果是模型的输出表达能力更加强大。

在使用相同数据训练的情况下，盘古-π（7B）在多任务上超越LLaMA 2等同规模大模型，并能实现10%的推理加速。

在1B规模上可达SOTA。

基于这一架构，我们还开发了一个金融法律大模型，名为"云山"。

该任务由领袖级AI专家陶大程带领。

具体如何实现？一起来看。

利用非线性解决特征塌陷

目前常见大模型基本都采用Transformer架构，比如GPT、LLaMA等。

它的核心构成包括多头自注意力机制（MSA）和前馈网络（FFN）。

MSA的主要功能是计算输入序列中每个token和其他所有token之间的相关性，通过学习输入序列中的依赖关系，可以增强对语言的理解能力。FFN主要对输入进行非线性转换，增强模型表达能力，使其可以逼近更复杂的函数。

不过，华为诺亚方舟实验室发现，特征崩溃（feature collapse）会影响Transformer架构的表现，降低其表达能力，使模型难以区分不同输入。

以LLaMA为例，在更深层的神经网络上，特征等级显著降低，导致了所有token之间的相似性更强。

从机制上来看，自注意力模块可以看做在完全图上进行信息聚合，连续堆叠多层注意力就像连续多层图卷积一样，会产生过度特征平滑效应。

另一方面，多层感知器（MLP）中的激活函数提供的非线性还不够，抑制特征崩溃的作用有限。

由此，团队想要提高模型的非线性表达能力，避免特征崩溃，进而提出了本次工作盘古-π。

如下是盘古-π的结构示意：

在FFN中加入串联激活函数，在MSA中集成一种增强快捷连接（Aug-S)，可以更有效地在Transformer架构中引入更多非线性。

使用了增强快捷连接（Aug-S）的MSA，能将每个token的特征转换为不同表示形式。

基于这一新架构，通过大规模训练和微调，研究团队开发了一个盘古-π基础模型。

实验结果显示，该模型在多任务中表现超越其他同规模模型（分别测试了7B和1B规模）。

而且盘古-π-7B可以达到约10%的推理加速。

同时团队还以此为基础开发了一个金融法律领域大模型“云山”，它同样在多个benchmark中成绩超越其他模型。

通讯作者为陶大程

值得关注的是，本项研究的团队阵容也非常亮眼。

通讯作者为陶大程。

他是欧洲科学院外籍院士、澳大利亚科学院院士。本科就读于中科大，说是毕业于港中文MMLab、师从汤晓鸥。

2007年从博士毕业后，先后在中国香港理工大学、新加坡南洋理工大学，澳大利亚悉尼科技大学、悉尼大学任教。目前是清华大学智能产业研究院AIR团队卓越访问教授。

与此同时，他还先后加盟过优必选、京东，曾是京东最高级别AI科学家、担任京东探索研究院院长。

一作为王云鹤。

他是2012实验室诺亚方舟实验室高级研究员，现任算法应用部部长。

王云鹤在华为负责高效AI算法的创新研发以及在华为业务中的应用。他和团队开发了高效AI算法，其衍生应用在中国天眼FAST观测工作中，协助中科院国家天文台专家找到了数百个新的快速射电暴样本。

论文地址：http://arxiv.org/abs/2312.17276

本篇关于《华为推出改进版Transformer架构，盘古-π解决特征缺陷问题，带来超越LLaMA的同等规模性能》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于科技周边的相关知识，请关注golang学习网公众号！