网易首页 > 网易号 > 正文 申请入驻

Avey AI研究团队推出全新编码器架构:比传统BERT更快更准确

0
分享至


Avey AI的研究团队在2026年的ICLR(国际学习表征大会)上发表了一篇重要论文,介绍了他们开发的全新编码器架构Avey-B。这项研究的论文编号为arXiv:2602.15814v1,感兴趣的读者可以通过这个编号查找到完整的研究内容。

在人工智能领域,有一类被称为"编码器"的模型就像是专业的文本理解专家。这些模型的工作原理很像我们阅读文章时的思考过程:需要同时理解每个词的意思,还要把握整篇文章的脉络。传统的BERT模型就是这样的专家,它在过去几年里帮助我们完成了大量的文本分析工作,从邮件分类到信息检索都有它的身影。

然而,随着我们需要处理的文本越来越长,传统的BERT模型就像一个阅读速度固定的人,面对长篇大论时会变得力不从心。每当文本长度翻倍,它所需要的时间和内存会急剧增长,这就限制了它在实际应用中的效果。正是在这样的背景下,Avey AI的研究团队开发了Avey-B,这是一个能够更高效处理长文本的新型编码器。

一、重新设计的文本理解方式

传统的BERT模型处理文本的方式就像是一个非常细致的编辑,需要同时关注文章中的每一个词,并且要理解任意两个词之间的关系。这种全面的关注虽然很准确,但也意味着随着文本变长,需要处理的关系数量会呈几何级数增长。

Avey-B采用了一种全新的策略,更像是一个聪明的阅读者:它首先将长文本分割成若干个小段落,然后对每个段落,只选择最相关的其他段落来帮助理解,而不是同时考虑所有段落。这种方法的巧妙之处在于,它保持了理解的准确性,同时大大减少了计算的复杂度。

研究团队将这个新架构称为Avey-B,其中"B"代表双向(bidirectional),表示它能够像人类阅读一样,既考虑前文,也考虑后文。这个模型建立在他们之前开发的Avey架构基础上,但专门针对编码器的需求进行了优化。

二、三项核心技术创新

Avey-B的成功源于三项关键的技术创新,每一项都解决了传统方法中的特定问题。

第一项创新被称为"解耦参数化"。在传统的方法中,模型学习到的固定权重会与输入文本的相似度分数混合在一起,这就像是让一个有固定偏好的评委和一个根据具体情况变化的评委同时打分,有时会产生矛盾的结果。比如说,两个段落明明很相似,但固定权重可能会给它们很低的分数,这就违背了"相似的内容应该得到更多关注"的直觉。

Avey-B通过将这两种评分机制分离到不同的层中来解决这个问题。有些层专门负责静态的线性变换,就像有固定标准的评委;而另一些层则根据内容的相似度动态调整,就像根据具体情况灵活判断的评委。这样的设计确保了相似度高的内容总是能得到应有的重视。

第二项创新是"稳定性导向的归一化"。在动态层中,研究团队引入了一种新的归一化方法,将每个位置的相似度分数除以该位置所有分数的总和。这就像是确保每个评委给出的分数总和都是固定的,避免了某些评委给分过高或过低的问题。这种方法不仅提高了训练的稳定性,还一致性地改善了各种下游任务的性能。

第三项创新是"神经压缩模块"。在原始的Avey设计中,每个段落需要与它的前k个相关段落拼接在一起进行处理,这会使输入大小膨胀k倍。对于双向处理来说,这种膨胀会严重影响效率。Avey-B在选择相关段落后,使用一个学习到的线性投影将扩展后的内容压缩回原始大小,就像是用一个智能的摘要工具,保留最重要的信息而去除冗余。

三、全方位的性能评测

为了验证Avey-B的效果,研究团队进行了广泛的实验评测,涵盖了四个主要的应用场景:序列分类、标记分类、问答任务和信息检索。

在序列分类任务中,模型需要判断整个句子或段落的性质,比如判断一条评论是正面还是负面。研究团队在MNLI、QQP和SST-2等标准数据集上测试了Avey-B。结果显示,Avey-B在这类任务上表现出色,特别是在SST-2任务上达到了最高分。

标记分类任务要求模型识别文本中特定位置的信息,比如在一段文字中找出人名、地名等。这类任务对模型的精细理解能力要求很高。在CoNLL-2003、OntoNotes和UNER等数据集上,Avey-B展现出了显著的优势,超越了所有对比的Transformer模型。这个结果特别有意义,因为它表明Avey-B的分段处理策略不仅没有损失细节信息,反而增强了对局部特征的捕捉能力。

在问答任务中,模型需要根据给定的文本段落回答相关问题。研究团队在ReCoRD、SQuAD和SQuAD v2等数据集上进行了测试。虽然Avey-B在某些问答任务上的表现不如RoBERTa和ModernBERT,但考虑到它的训练数据量远少于这些模型(大约少了11倍),这样的结果已经相当不错。

信息检索任务考验模型在长文档中找到相关信息的能力。在MLDR、MS MARCO和Natural Questions等数据集上,Avey-B表现尤为突出,大幅超越了所有对比模型。这个结果很好地验证了Avey-B设计理念的正确性:通过选择性地关注最相关的内容段落,它能够更有效地处理长文档检索任务。

四、效率优势的深度分析

除了准确性,Avey-B在处理效率方面的优势更加引人注目。研究团队详细测试了不同模型在各种序列长度下的表现,从128个词汇到96,000个词汇的范围内进行了全面对比。

在这些测试中,研究团队使用了吞吐量(每秒处理的词汇数)和延迟(处理一次前向传播所需的时间)两个指标。结果显示,随着序列长度的增加,传统的Transformer模型的处理速度急剧下降,而Avey-B则保持了相对稳定的性能。

具体来说,当序列长度达到96,000个词汇时,Avey-B的处理速度比ModernBERT快3.38倍,比NeoBERT快11.63倍。更重要的是,通过数学建模分析,研究团队发现Avey-B的性能下降率(用幂律函数T(N) ∝ N^(-α)中的指数α表示)约为0.44,而ModernBERT和NeoBERT分别为0.77和0.81。这意味着随着序列长度的增加,Avey-B的性能下降速度大约是传统模型的一半。

这种效率优势的根本原因在于Avey-B的神经处理器只需要处理固定大小的分段,而不需要处理整个序列。每个分段的处理成本是固定的,因此总体成本与序列长度成线性关系,而不是传统模型的二次关系。

五、设计细节的深入探索

为了找到最优的设计方案,研究团队进行了大量的设计选择研究和消融实验。

在静态层和动态层的排列方式上,研究团队测试了多种模式:交替排列、单一动态层、两阶段堆叠以及完全静态或动态的配置。结果显示,以静态层开始的交替排列(静态→动态→静态→动态...)效果最佳。这种设计让静态层提供稳定的表示基础,然后动态层根据输入内容进行自适应调整。

在动态层的归一化方法选择上,研究团队比较了按和归一化、RMS归一化、softmax和缩放softmax等方法。结果表明,简单的按和归一化方法效果最好,它保持了相似度的相对顺序,同时确保了数值的稳定性。

在超参数设置方面,研究团队发现序列长度N、分段大小S和选择的前k个相关分段数量之间存在一个重要的关系:最佳性能通常出现在S×(k+1)≈N的配置下。这意味着有效的上下文覆盖应该接近整个训练序列的长度。

六、严格的消融实验验证

为了验证每个设计决策的重要性,研究团队进行了详尽的消融实验,即逐一移除某个组件来观察性能变化。

当移除行归一化时,模型性能出现了显著下降,在各个任务类型上的平均分数分别下降了3.55%、0.87%、7.65%和15.33%。这表明归一化对于稳定训练和保持良好性能至关重要。

当回到耦合的参数化设计(即将静态权重和动态相似度分数混合)时,各任务的性能分别下降了1.43%、2.12%、2.53%和7.40%。这证实了解耦设计的重要性。

神经压缩模块的移除导致了适度的性能下降,但同时带来了4.37倍的效率提升。考虑到这个权衡,研究团队认为压缩模块提供了良好的效率-效果平衡。

当移除残差连接(即压缩器输出和原始分段之间的直接连接)时,所有任务的性能都有所下降,平均降幅为3.38%。这说明保留局部信息的重要性。

最后,完全移除排序器会导致7.46%的严重性能下降,这证实了选择性关注机制是Avey-B成功的关键因素。

七、长文本处理能力的极限测试

为了测试Avey-B在极长文本上的表现,研究团队设计了一个"大海捞针"基准测试。这个测试模拟了在极长文档中寻找特定信息的场景,文档长度可达96,000个词汇。

测试包含两种类型的任务:单针任务和多针任务。单针任务要求模型在长文档中找到一个特定的键值对,主要测试语义定位能力。多针任务则更加复杂,文档中包含多个具有相同键的键值对,查询要求找到第n个出现的值,这需要模型具备位置推理能力。

测试结果令人印象深刻。在单针任务(NIAH-1)中,Avey-B基础版本和大型版本在从1,000到96,000词汇的范围内都保持了稳定的性能,准确率只下降了3-4个百分点。相比之下,ModernBERT和NeoBERT无法处理超过其训练窗口长度的文本,分别限制在8,000和4,000词汇以内。

在更具挑战性的多针任务(NIAH-2)中,Avey-B同样表现出色。基础版本从78.3%(1,000词汇)下降到71.9%(96,000词汇),大型版本从78.9%下降到74.5%,展现了良好的长文本推理能力。

这些结果特别有意义,因为Avey-B只在2,048词汇的窗口上进行了训练,却能够处理比训练长度长47倍的文本,这证明了其架构设计的优越性。

八、统计稳定性的深入分析

除了性能和效率,研究团队还分析了不同模型在多次独立运行中的稳定性。通过在每个基准测试上进行10次独立的随机种子实验,他们计算了结果的标准差,以评估模型对初始化的敏感性。

结果显示,在基础规模的模型中,RoBERTa表现出最低的方差,这与其在学术界享有的稳定性声誉一致。Avey-B排名第二,显示出良好的训练稳定性。在大型模型中,这种稳定性差异变得更加明显。ModernBERT尽管有强劲的中位数性能,但在某些基准测试上表现出较大的不稳定性,特别是在ReCoRD、UNER和Natural Questions任务上。

Avey-B在各种规模下都保持了较低的标准差,通常低于1.06,很少出现病态的不稳定性。研究团队将这种稳定性归因于三个核心设计原则:解耦的静态和动态层防止了固定参数与相似度分数之间的破坏性交互;行归一化的相似度矩阵稳定了激活幅度并确保了良好的梯度流;神经压缩过滤了检索上下文中的无关信号。

九、深层机制的可视化分析

为了更好地理解Avey-B的工作机制,研究团队对模型学习到的参数矩阵进行了详细分析。他们比较了耦合和解耦版本中静态层学习到的交叉嵌入投影矩阵,发现了显著的差异。

在耦合版本中,由于相似度矩阵是非负的,模型为了避免破坏性的符号翻转,倾向于学习正权重。这导致了一种"正性偏置",特别是在较深的层中,正权重的比例接近100%。然而,仍然有一些负权重残留,这正是导致相关性单调性违反的根源。

相反,在解耦版本中,动态层单独产生混合权重,这些权重通过构造是归一化和非负的,因此在相似度操作层面保证了单调性。静态层则独立学习,不再需要被迫趋向非负性以保持单调性。结果显示,解耦版本的权重分布接近零均值,正负权重大致平衡,保留了抑制模式的能力。

在权重分布的其他统计特性上,两种版本也表现出不同的特征。耦合版本的矩阵表现出较小的标准差,趋向于平滑和同质的模式。而解耦版本维持了较大的波动,允许更强的正值和负值,这可能反映了更大的表示灵活性。

十、实际应用的广阔前景

Avey-B的设计特点使其在多个实际应用场景中具有独特的优势。在信息检索领域,其选择性注意机制天然适合在大量文档中快速定位相关内容。企业可以利用这一特性来构建更高效的内部知识库搜索系统,帮助员工快速找到所需信息。

在文档分析和处理方面,Avey-B能够处理长达数万词的文档,而不会像传统模型那样遭遇内存限制。这对于法律文档分析、学术论文处理、技术手册理解等需要处理长文本的场景具有重要意义。

在对话系统和客服机器人领域,Avey-B的效率优势可以显著降低响应时间和计算成本。特别是在需要理解长对话历史的场景中,其线性扩展特性比传统的二次扩展模型更具实用价值。

对于内容审核和分类任务,Avey-B在标记分类上的优异表现使其成为自动化内容管理的理想选择。无论是社交媒体内容审核、新闻文章分类,还是电子邮件的自动标签,都可以从中受益。

说到底,Avey-B代表了自然语言处理领域的一个重要进展。它不是简单地修补现有技术的缺陷,而是从根本上重新思考了如何高效处理长文本的问题。通过巧妙的架构设计,它在保持甚至提升准确性的同时,大幅改善了计算效率,为人工智能在更多实际场景中的应用开辟了新的可能性。

这项研究的意义不仅在于提出了一个新的模型,更在于它展示了一种新的思路:在追求更强大的AI能力时,我们不一定需要更多的计算资源,有时候更聪明的设计能够事半功倍。随着我们需要处理的信息量越来越大,像Avey-B这样兼具效率和效果的技术将变得越来越重要。

当然,这项技术还处于研究阶段,距离大规模商业应用还需要时间。但它为整个领域指出了一个值得探索的方向,相信会启发更多类似的创新。对于那些对这项技术感兴趣并希望深入了解技术细节的读者,可以通过论文编号arXiv:2602.15814v1查找完整的研究报告。

Q&A

Q1:Avey-B比传统BERT模型有什么优势?

A:Avey-B在处理长文本时具有显著优势。当文本长度达到96,000词时,它的处理速度比ModernBERT快3.38倍,比NeoBERT快11.63倍。同时在准确性方面,它在标记分类和信息检索任务上超越了所有传统Transformer模型,而且随着文本变长,性能下降速度只有传统模型的一半。

Q2:Avey-B是如何做到既快又准确的?

A:Avey-B采用了三项关键创新:首先是解耦参数化,将固定权重和动态相似度分数分离到不同层中,避免冲突;其次是稳定性归一化,确保相似内容得到应有重视;最后是神经压缩模块,在保留重要信息的同时减少计算量。这些设计让它能选择性关注最相关的内容段落,而不是同时处理所有信息。

Q3:普通用户什么时候能使用到Avey-B技术?

A:目前Avey-B还处于研究阶段,研究团队已经公开了完整的实现代码和预训练模型。虽然距离大规模商业应用还需要时间,但它的设计思路已经为整个AI领域指出了新方向。未来可能会在文档分析、信息检索、对话系统等需要处理长文本的应用中率先看到类似技术的应用。

特别声明:以上内容(如有图片或视频亦包括在内)为自媒体平台“网易号”用户上传并发布,本平台仅提供信息存储服务。

Notice: The content above (including the pictures and videos if any) is uploaded and posted by a user of NetEase Hao, which is a social media platform and only provides information storage services.

相关推荐
热点推荐
百姓躺平摆烂,食税群体怎么办?

百姓躺平摆烂,食税群体怎么办?

律法刑道
2026-06-03 09:30:48
石宇奇出局,国羽男单全军覆没,3单项仅剩独苗,印尼赛5日赛程

石宇奇出局,国羽男单全军覆没,3单项仅剩独苗,印尼赛5日赛程

佑铭羽球
2026-06-05 03:25:34
国家发改委新规!评标专家要失业了

国家发改委新规!评标专家要失业了

新浪财经
2026-06-04 01:43:35
为什么说闲鱼是中国最大的黑市?网友:我直接给跪了!

为什么说闲鱼是中国最大的黑市?网友:我直接给跪了!

另子维爱读史
2026-06-02 10:51:07
6月5日精选热点:光纤订单排至2027年 这些产能龙头要大涨

6月5日精选热点:光纤订单排至2027年 这些产能龙头要大涨

元芳说投资
2026-06-04 21:32:55
火箭发布下赛季球衣:西红柿炒蛋配色重现航天城

火箭发布下赛季球衣:西红柿炒蛋配色重现航天城

懂球帝
2026-06-05 01:22:10
76岁的万科创始人王石,最近彻底成了全网焦点。

76岁的万科创始人王石,最近彻底成了全网焦点。

梦录的西方史话
2026-04-23 14:36:39
庞峥麟两战10分+17分:连场上双证进攻能力 失误全场最多成短板

庞峥麟两战10分+17分:连场上双证进攻能力 失误全场最多成短板

醉卧浮生
2026-06-04 21:41:19
香奈儿盛典:周迅下垂、小腹隆起,金高银站C位,刘嘉玲头秃脸肿

香奈儿盛典:周迅下垂、小腹隆起,金高银站C位,刘嘉玲头秃脸肿

王瑄自驾
2026-06-05 00:13:11
八尺夫人莎拉波娃私服照惊艳全网!188cm超模比例,魅力十足

八尺夫人莎拉波娃私服照惊艳全网!188cm超模比例,魅力十足

科学发掘
2026-06-04 15:56:15
6年了,郭麒麟的反击几乎断送了朱亚文的演艺生涯

6年了,郭麒麟的反击几乎断送了朱亚文的演艺生涯

小熊侃史
2025-12-25 11:24:12
历史的谣言:苏美尔文明有始有终 根本不是什么外星产物

历史的谣言:苏美尔文明有始有终 根本不是什么外星产物

冷炮历史
2026-06-04 08:00:08
女老师穿“日系连衣裙”,家长们坐不住了:天气再热也不能这么穿

女老师穿“日系连衣裙”,家长们坐不住了:天气再热也不能这么穿

泽泽先生
2026-05-22 12:55:21
为啥中南海至今不挂"中央人民政府"牌子?取而代之的是一句口号

为啥中南海至今不挂"中央人民政府"牌子?取而代之的是一句口号

乐天闲聊
2026-06-05 02:13:08
中国男篮遭绝杀,最差球员上场21分钟致球队输18分

中国男篮遭绝杀,最差球员上场21分钟致球队输18分

赫岝乡村摄影
2026-06-05 01:16:36
伊朗最高领袖,罕见“亮相”了

伊朗最高领袖,罕见“亮相”了

中国新闻周刊
2026-06-04 15:49:02
9年斥资5800万带U17男足夺冠,董路坦言,不像孙继海卖孩子挣钱

9年斥资5800万带U17男足夺冠,董路坦言,不像孙继海卖孩子挣钱

小七说篮球
2026-06-04 16:53:07
越南人评价中国:环境很好,科技很发达,不过有一点比不上越南

越南人评价中国:环境很好,科技很发达,不过有一点比不上越南

无意争春
2026-06-03 15:14:11
保税仓直发的“海外专柜正品”竟然是国产假货!3人伪造授权书骗取“全球购”资质认证 卖了2800余万元

保税仓直发的“海外专柜正品”竟然是国产假货!3人伪造授权书骗取“全球购”资质认证 卖了2800余万元

闪电新闻
2026-06-05 00:17:18
广东姑娘的“阳台房”火了,因连廊劏房走红,过日子像被围观

广东姑娘的“阳台房”火了,因连廊劏房走红,过日子像被围观

林子说事
2026-06-04 12:34:22
2026-06-05 09:07:00
至顶AI实验室 incentive-icons
至顶AI实验室
一个专注于探索生成式AI前沿技术及其应用的实验室。
3476文章数 171关注度
往期回顾 全部

科技要闻

再度推迟发布,Meta继续在“AI泥潭”挣扎

头条要闻

男子11年前30万买手机二手价3000元 在售手机最贵69万

头条要闻

男子11年前30万买手机二手价3000元 在售手机最贵69万

体育要闻

欧冠决赛躺赢?他也曾是大巴黎的英雄

娱乐要闻

难怪奚梦瑶能嫁入豪门

财经要闻

卖“友好”,试点5天被骂停

汽车要闻

北京现代5月销量强势反弹:国内17065辆 出口环比翻倍

态度原创

数码
教育
本地
手机
公开课

数码要闻

戴尔新商务本,这配置你咋看?

教育要闻

加强保障!教育部启动实施“2026年高考护航行动”(来源:央视网)

本地新闻

用杨柳青年画的方式,打开天津

手机要闻

国产高端折叠屏手机再下一城!荣耀Magic V6马来西亚发布 1.29万元起

公开课

李玫瑾:为什么性格比能力更重要?

无障碍浏览 进入关怀版