分享好友 资讯首页 频道列表

Salesforce团队CoDA模型:17亿参数“小个子”挑战代码生成大模型

2025-11-06 01:3811010

在人工智能代码生成领域,Salesforce AI Research团队的一项新研究引发了广泛关注。该团队开发的CoDA模型以17亿参数的“轻量级”身躯,在代码生成任务中展现出与70亿参数大模型相媲美的实力,打破了“参数规模决定性能”的传统认知。

传统代码生成模型多采用自回归架构,其工作模式类似于逐字书写的笔迹——必须按顺序生成内容,一旦某个环节出错,后续内容可能全盘偏离。这种模式在处理代码填空、上下文补全等任务时显得尤为吃力,就像要求作家必须从第一个字写到最后一个字,而不能中途修改段落逻辑。

CoDA的创新之处在于引入了扩散模型架构。该模型的工作流程类似雕塑创作:从模糊的“噪声”开始,通过多轮迭代逐步雕琢出清晰的代码结构。这种非顺序生成的方式使模型能够同时参考前后文信息,就像雕塑家可以从任意角度调整作品细节,确保整体逻辑的连贯性。

研究团队为CoDA设计了独特的三阶段训练体系。在预训练阶段,模型接触包含网页文本、多语言代码、数学公式等在内的1800亿token数据,通过“渐进式掩码”技术逐步提升对不完整信息的处理能力——从最初随机遮挡单个词汇,到后期遮挡完整段落甚至要求续写开头。

中期训练阶段聚焦编程专业能力。研究团队精选210亿token的专项数据,包含学术论文、编程教材和优质代码库。此阶段强化了三种掩码策略:保留开头提示的“不可掩码前缀”、删除结尾部分的“截断后缀”,以及模拟真实编程场景的“块状掩码”。这些策略使模型更擅长处理函数补全、逻辑插入等实际任务。

后训练阶段则着重提升模型与人类的交互能力。通过OpenCoder数据集的指令微调,CoDA学会了理解自然语言需求并生成符合要求的代码。研究团队采用小学习率策略,确保模型在提升交互能力的同时不丢失编程技能。

技术实现层面,CoDA采用“分类扩散”方法处理离散代码token。每个位置的token生成被转化为分类问题,模型通过预测被掩码位置的原始内容完成代码构建。配合“置信度引导采样”技术,模型能根据预测可靠性动态调整决策速度——对确定内容快速生成,对不确定部分谨慎处理,既保证质量又提升效率。

在Google TPU v4-1024集群上,研究团队构建了高效的训练系统。该系统支持自动数据加载、分布式模型保存和动态掩码策略调整,实现了每秒10亿token的处理速度。更值得关注的是,整个训练流程、代码和模型权重均已开源,为社区提供了完整的复现指南。

实测数据显示,CoDA在Humaneval和MBPP两大编程基准测试中表现突出。在Humaneval测试中,其pass@1分数达54.3%,意味着能一次性正确解决半数以上编程问题;MBPP测试中则取得47.2%的成绩,显示出解决实际编程需求的能力。与同规模自回归模型相比,CoDA在需要上下文理解的复杂任务中优势明显。

效率方面,CoDA的推理延迟比同类扩散模型低约40%。研究发现,512步扩散迭代后性能提升趋于饱和,这为实际应用提供了优化方向——在保持性能的同时可大幅缩短推理时间。

渐进式掩码策略是CoDA的核心创新之一。该策略通过模拟真实使用场景,缓解了训练与推理的分布差异。例如,“不可掩码前缀”对应根据需求生成代码的场景,“截断后缀”训练处理不完整输入的能力,“块状掩码”则模拟函数中间逻辑的填补任务。这种训练方式显著提升了模型的泛化能力。

研究团队完全开源的决策具有深远影响。除了模型权重,他们还公开了TPU训练代码、数据处理流程和评估框架。这种透明度降低了领域研究门槛,促进了技术迭代。配套的评估工具确保了不同模型能在统一标准下比较性能,推动了行业规范化发展。

对于开发者而言,CoDA的开源特性意味着可直接获取应用资源。通过GitHub的“SalesforceAIResearch/CoDA”项目或HuggingFace平台的“Salesforce/CoDA-v0-Instruct”模型,开发者能快速集成代码生成能力。完整的训练推理代码更支持定制化开发。

这项研究证明,通过架构创新和训练策略优化,轻量级模型同样能实现高性能。CoDA为AI代码生成领域提供了新的发展范式——在资源约束下,通过智能设计实现效率与效果的平衡。其开源生态也将持续推动相关技术的进步与应用。

反对 0
举报 0
收藏 0
打赏 0
评论 0
太空“鼠”趣:4只小鼠在中国空间站的“太空生活”实录曝光
太空“鼠”趣:4只小鼠在中国空间站的“太空生活”实录曝光

0评论2025-11-06742

AI手机端侧模型崛起:企业竞逐新赛道 移动终端生态悄然生变
AI手机端侧模型崛起:企业竞逐新赛道 移动终端生态悄然生变

0评论2025-11-06730

太空“小住客”:中国空间站小鼠“航天员”生活日常大揭秘
太空“小住客”:中国空间站小鼠“航天员”生活日常大揭秘

0评论2025-11-06910

进博会见智元机器人:具身智能“全场景渗透”,开启实用新篇章
进博会见智元机器人:具身智能“全场景渗透”,开启实用新篇章

0评论2025-11-062876

快手科技携手清华提出ASPO策略:打破AI训练"偏心"困境,实现均衡提升
快手科技携手清华提出ASPO策略:打破AI训练"偏心"困境,实现均衡提升

0评论2025-11-061482

微算法科技推出QoS感知技术,赋能边缘大模型自适应拆分与智能推理
微算法科技推出QoS感知技术,赋能边缘大模型自适应拆分与智能推理

0评论2025-11-06829

注册仅俩月,对标特斯拉!星际动力人形机器人如何搅动行业风云?
注册仅俩月,对标特斯拉!星际动力人形机器人如何搅动行业风云?

0评论2025-11-061761

IBM 2025年四季度或裁员,AI浪潮下科技巨头优化结构聚焦核心业务
IBM 2025年四季度或裁员,AI浪潮下科技巨头优化结构聚焦核心业务

0评论2025-11-061093

UIUC团队研发TATTOO系统:用工具思维破解AI表格推理难题
UIUC团队研发TATTOO系统:用工具思维破解AI表格推理难题

0评论2025-11-061655