arXiv:2608.00916cs.IRcs.LG2026-08被引 2

用学术预算训练300亿参数专家模型,效率更高、推理更快。

Tevatron Meets Megatron: Expert-Parallel LLM Reranker Training on an Academic Budget

论文配图:Tevatron Meets Megatron: Expert-Parallel LLM Reranker Training on an Academic Budget
图 1 · 摘自论文原文
  • 将Megatron-Core整合进Tevatron,支持大规模专家并行训练
  • 30B参数MoE模型在BEIR-15上达到8B密集模型性能,激活参数少于一半
  • 支持LoRA与全参数微调,适合资源有限的学术团队使用

当前重排序方法——如百亿级交叉编码器、专家混合(MoE)主干网络及强教师蒸馏——已超越多数学术机构的训练基础设施。现有Tevatron训练依赖Hugging Face Trainer搭配DeepSpeed或PyTorch FSDP1,但后者缺乏对大规模MoE训练的有效支持。本文提出Tevatron 3.0,将Megatron-Core训练后端集成至Tevatron,同时保留其数据管道、评估流程和兼容Hugging Face的检查点格式。我们在多个分布式配置下对比新后端,结果显示,在相似数据并行设置下,Megatron在重排序质量和训练效率上与FSDP相当,单节点配置下最多快22%,且支持LoRA与全参数微调。关键的是,专家并行性使300亿参数的Qwen3-30B-A3B MoE重排序器得以训练,而PyTorch FSDP1无法实现。基于该框架,我们对MoE与密集模型、LoRA与全参数微调、蒸馏与对比学习在BEIR-15上进行受控对比,并报告了Hugging Face与vLLM的推理吞吐量。结果表明,MoE重排序器在保持8B密集模型性能的同时,激活参数不足一半,推理吞吐量显著更高。我们将开源该框架及训练好的检查点。

原文摘要 · Abstract (English)

Modern reranking recipes---billion-scale cross-encoders, mixture-of-experts (MoE) backbones, and distillation against strong teachers---have outpaced the training infrastructure available to most academic groups. Existing Tevatron reranker training relies on the Hugging Face Trainer with DeepSpeed or PyTorch FSDP1, but these backends lack efficient support for large-scale MoE training. We present Tevatron 3.0, which integrates a Megatron-Core training backend into Tevatron while preserving its data pipeline, evaluation workflow, and Hugging Face-compatible checkpoints. We benchmark existing distributed training configurations against the new backend, showing that Megatron matches FSDP reranker quality and training efficiency under comparable data-parallel settings, is up to 22% faster in the recommended single-node configuration, and supports both LoRA and full-parameter fine-tuning. Crucially, expert parallelism enables training a 30B-parameter Qwen3-30B-A3B MoE reranker, which is infeasible with PyTorch FSDP1. Using this framework, we conduct a controlled comparison of MoE versus dense models, LoRA versus full-parameter tuning, and distillation versus contrastive training on BEIR-15 with three first-stage retrievers, and report serving throughput for Hugging Face and vLLM. We find that the MoE reranker matches dense 8B quality while activating less than half as many parameters and achieving substantially higher inference throughput. We will release the framework and trained checkpoints.

大模型训练专家混合重排序学术计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。