NeoBERT用新架构和训练法让编码器模型达到顶尖表现
NeoBERT: A Next-Generation BERT
- 重构编码器结构,融合最新预训练技术与4096上下文长度
- 250M参数下超越BERT大模型等,在MTEB上达最先进水平
- 代码数据全开源,可直接替换现有模型快速部署
近期自回归大模型如LLaMA和DeepSeek在上下文学习与推理能力上取得显著进展,而以BERT和RoBERTa为代表的编码器模型虽广泛用于下游NLP任务,却未同步提升。为此,我们提出下一代编码器NeoBERT,通过整合前沿架构、现代数据与优化预训练方法,重新定义双向模型能力。NeoBERT设计为即插即用,采用最优深度-宽度比,支持4096令牌扩展上下文。尽管仅250M参数,其在大规模MTEB基准测试中仍优于BERT large、RoBERTa large、NomicBERT和ModernBERT,且在相同微调条件下表现领先。我们还系统评估各改进对GLUE的影响,并建立统一的MTEB微调与评估框架。所有代码、数据、检查点与训练脚本均已公开,以加速研究与实际应用。
原文摘要 · Abstract (English)
Recent innovations in architecture, pre-training, and fine-tuning have led to the remarkable in-context learning and reasoning abilities of large auto-regressive language models such as LLaMA and DeepSeek. In contrast, encoders like BERT and RoBERTa have not seen the same level of progress despite being foundational for many downstream NLP applications. To bridge this gap, we introduce NeoBERT, a next-generation encoder that redefines the capabilities of bidirectional models by integrating state-of-the-art advancements in architecture, modern data, and optimized pre-training methodologies. NeoBERT is designed for seamless adoption: it serves as a plug-and-play replacement for existing base models, relies on an optimal depth-to-width ratio, and leverages an extended context length of 4,096 tokens. Despite its compact 250M parameter footprint, it achieves state-of-the-art results on the massive MTEB benchmark, outperforming BERT large, RoBERTa large, NomicBERT, and ModernBERT under identical fine-tuning conditions. In addition, we rigorously evaluate the impact of each modification on GLUE and design a uniform fine-tuning and evaluation framework for MTEB. We release all code, data, checkpoints, and training scripts to accelerate research and real-world adoption.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。