游乐游手机版
首页/科技数码/文章详情

InfLLM开源二代模型提速3倍,支持稀疏注意力训练

时间:2025-12-08 21:41
InfLLM-V2是一种可高效处理长文本的稀疏注意力模型,仅需少量长文本数据即可训练,且性能接近传统稠密模型。通过动态切换短长文本处理模式,显著提升长上下文任务的效率与质量。从短到长低成本「无缝切换

InfLLM-V2突破性地解决了大模型长文本处理难题,这种创新的稀疏注意力机制仅需极少量长文本数据就能完成训练,性能表现却与传统稠密模型不相上下。通过动态切换长短文本处理模式,在保证任务质量的前提下显著提升了长上下文场景的处理效率。该技术实现了从短文本到长文本的低成本平滑过渡,同时优化了预填充和解码阶段的速度,彻底释放了长上下文模型的潜能。

长序列处理:大模型应用的突围关键

当序列长度增加时,传统稠密注意力机制的计算开销会呈指数级增长,严重限制了产品的实用性和成本控制能力。

为了攻克这一技术难关,清华大学、OpenBMB和哈尔滨工业大学的研究团队联合研发了InfLLM-V2解决方案。这一创新框架完全基于原生稀疏注意力原理,不需要额外增加任何参数,同时保持着极高的训练效率。

InfLLM-V2在短文本场景保留原始处理效率,在长文本场景则智能切换至稀疏模式,实现了端到端的显著加速。令人惊叹的是,该方法仅需50亿个长文本token就能完成训练,而DeepSeek-V3.2-Exp模型则需要近万亿token的数据量才能实现类似效果。

实测数据显示,InfLLM-V2相比传统稠密注意力机制实现了4倍的速度提升,长文本理解任务保持98.1%的原有性能,深度思考任务更是保持了99.7%的性能水平。

InfLLM三大核心技术优势

1. 经济高效的训练成本:仅需50亿长文本token即可完成训练,大幅降低训练成本和时间周期。

2. 智能化的长短文本切换:不引入额外参数,短文本使用稠密模式,长文本自动切换为稀疏模式,完美契合"短序列预训练-长序列微调"的主流实践。

3. 优化的算子实现:针对稀疏注意力计算中的瓶颈问题进行系统性优化,提出面向硬件的创新方案,显著降低内存I/O和计算开销。

\

InfLLM-V2高效运行的实现原理

传统Transformer的自注意力机制需要对每个查询token与所有历史token进行全量计算,这在处理超长上下文时会导致无法承受的延迟和成本。

研究发现,长序列中的注意力矩阵具有显著的稀疏特性,绝大多数远距离交互并不重要。InfLLM-V2创新性地将全量计算转变为选择性计算,仅保留关键的上下文交互,从而大幅降低了计算开销。

\

该技术的核心创新分为两个阶段:首先智能识别关键的上下文块,随后仅在选定区块上进行注意力计算。这种方法在训练阶段就引入了稀疏机制,系统性提升了长文本处理效率。

InfLLM-V2摒弃了原有NSA架构的多分支设计,提出了全新的单分支方案,实现了参数共享和运算模式统一,彻底解决了长短文本切换时的兼容性问题。

\

与DeepSeek Sparse Attention的技术对比

值得注意的是,DeepSeek在其V3.2-Exp模型中提出的DSA架构已经抛弃了NSA的设计思路,在后训练阶段引入稀疏注意力算法。

实验结果与分析

研究团队基于MiniCPM4模型在多组测试中对比了不同稀疏算法的表现。

长文本理解性能

在RULER、LongBench等标准测试中,InfLLM-V2展现了与传统稠密模型几乎相同的性能表现。

\

\

深度推理能力

在数学推导和代码生成等高难度任务中,InfLLM-V2依然保持着优异的性能表现,为解决大模型深度思考效率问题提供了新思路。

\

运算效率评测

在A100和RTX4090平台上的测试表明,InfLLM-V2在128K长文本处理中可获得4-9倍的算子级加速。细致的分析证实,优化的块选择设计是实现高性能的关键。

\

\

开源生态系统

研究团队已在Hugging Face平台发布了MiniCPM4/MiniCPM4.1系列模型,这些开源原生稀疏注意力模型在多个专业测试中展现出卓越的性能表现。

未来,研究团队将持续优化InfLLM-V2的训练和推理算子,并将其集成到主流推理框架中。为推动稀疏注意力技术发展,计划逐步开源基础模型和训练数据集。

来源:https://36kr.com/p/3501893279014024
上一篇暗黑4 PTR更新:战利品与名望系统全面优化 下一篇阿里通义千问发力机器人,抢占具身AI赛道
本站内容用于信息整理与展示,如有侵权或内容问题请及时联系处理。

相关推荐

补充同频道和同主题内容,方便继续浏览更多相关内容。

同类最新

继续查看同栏目最近更新的文章。

更多
国内首批智能体国标发布,AI应用从验证迈向规模化
科技数码 · 2026-07-06

国内首批智能体国标发布,AI应用从验证迈向规模化

国内首批智能体国家标准正式发布,这一消息在人工智能领域引发广泛关注。简单来说,这份标准为智能体技术确立了 "定义 ",明确了分类体系、技术指标与测试方法,相当于为整个行业制定了统一规范。这意味着什么?这意味着智能体技术将告别 "野蛮生长 ",迎来标准化发展,加速其在工业制造、智慧服务、智能家居等场景的落地应

微星40周年限量游戏本开售 5090+96GB 55999元起
科技数码 · 2026-07-06

微星40周年限量游戏本开售 5090+96GB 55999元起

值得关注的是,微星在成立40周年之际,特别推出了限量典藏版机型——泰坦18 Ultra龙魂典藏版2026款游戏本,已于昨日零点正式开售,官方定价为55999元。此外,部分地区还可叠加国家补贴,实际到手价可低至54499元。 作为40周年专属纪念款,其外观设计自然独具匠心。机身正面采用金属蚀刻与阳极氧

墨刀原型强调交互高保真真机演示,产品流程从草图到协作评审
科技数码 · 2026-07-06

墨刀原型强调交互高保真真机演示,产品流程从草图到协作评审

原型工具究竟在解决什么问题?这个问题其实很值得探讨。不少人听到“原型设计”,第一反应往往是绘制几张静态页面、添加几个页面跳转链接,但实际的产品流程远比这复杂。墨刀在“墨刀原型”的官方说明中,将重心放在原型设计、交互、高保真和真机演示这几个核心能力上。换句话说,它并不满足于让团队只输出页面静态图,而是

保时捷Taycan最后两款旅行车宣告停产
科技数码 · 2026-07-06

保时捷Taycan最后两款旅行车宣告停产

保时捷正式为两款纯电旅行车系列画上终止符。据海外汽车媒体motor1报道,Taycan Sport Turismo与Taycan Cross Turismo已经停止生产。随着2027款全新纯电Taycan的推出,这两款衍生车型将从产品阵容中完全移除。保时捷官方确认,此次停产的根本原因非常明确——实际

墨刀白板助力市场洞察需求梳理,多工具看板服务产品评审共创
科技数码 · 2026-07-06

墨刀白板助力市场洞察需求梳理,多工具看板服务产品评审共创

首先提出一个关键判断:在产品经理的工作流程中,原型工具通常是最为熟悉的环节。此次,墨刀将“墨刀白板”功能的定位明确聚焦在市场洞察、产品规划与需求梳理三大领域——即进入具体原型设计之前的上游协作空间。 与原型工具不同,白板不侧重页面与交互细节。它更适合承载早期“发散—整理—讨论—共创—评审”的过程。简