arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2407.21770cs.AIcs.LG
收藏

MoMa:基于模态感知专家混合的高效早期融合预训练

MoMa: Efficient Early-Fusion Pre-training with Mixture of Modality-Aware Experts

Meta FAIR

Xi Victoria Lin, Akshat Shrivastava, Liang Luo, Srinivasan Iyer, Mike Lewis, Gargi Ghosh, Luke Zettlemoyer, Armen Aghajanyan

AI总结:

本研究提出模态感知专家混合架构MoMa,通过模态特定专家组与组内路由处理任意图文序列,在1万亿token预训练中相比稠密基线实现3.7倍整体FLOPs节省,结合MoD可达4.2倍,显著提升多模态早期融合预训练效率。

摘要:

我们提出了MoMa,这是一种专为混合模态早期融合语言模型预训练设计的新型模态感知专家混合(MoE,mixture-of-experts)架构。MoMa通过将专家模块划分为模态特定组,能够处理任意序列的图像和文本。这些组专门处理指定的token,同时在每个组内采用可学习的路由机制,以保持语义层面的自适应能力。我们的实验结果表明,这种模态特定的参数分配方式能大幅提升预训练效率。在1万亿token的训练预算下,拥有4个文本专家和4个图像专家的MoMa 1.4B模型,以预训练损失为衡量指标,相比计算量相当的稠密基线模型,实现了可观的FLOPs节省:整体节省3.7倍,其中文本处理节省2.6倍,图像处理节省5.2倍。这一表现优于拥有8个混合模态专家的标准专家选择MoE,后者整体FLOPs节省为3倍(文本3倍,图像2.8倍)。将MoMa与深度混合(MoD,mixture-of-depths)结合后,预训练FLOPs整体节省进一步提升至4.2倍(文本3.4倍,图像5.3倍),但这种组合因对路由准确性的敏感性提升,会损害因果推理的性能。这些结果表明,MoMa有望显著提升混合模态早期融合语言模型的预训练效率,为构建资源效率更高、能力更强的多模态AI系统铺平道路。

英文摘要:

We introduce MoMa, a novel modality-aware mixture-of-experts (MoE) architecture designed for pre-training mixed-modal, early-fusion language models. MoMa processes images and text in arbitrary sequences by dividing expert modules into modality-specific groups. These groups exclusively process designated tokens while employing learned routing within each group to maintain semantically informed adaptivity. Our empirical results reveal substantial pre-training efficiency gains through this modality-specific parameter allocation. Under a 1-trillion-token training budget, the MoMa 1.4B model, featuring 4 text experts and 4 image experts, achieves impressive FLOPs savings: 3.7x overall, with 2.6x for text and 5.2x for image processing compared to a compute-equivalent dense baseline, measured by pre-training loss. This outperforms the standard expert-choice MoE with 8 mixed-modal experts, which achieves 3x overall FLOPs savings (3x for text, 2.8x for image). Combining MoMa with mixture-of-depths (MoD) further improves pre-training FLOPs savings to 4.2x overall (text: 3.4x, image: 5.3x), although this combination hurts performance in causal inference due to increased sensitivity to router accuracy. These results demonstrate MoMa's potential to significantly advance the efficiency of mixed-modal, early-fusion language model pre-training, paving the way for more resource-efficient and capable multimodal AI systems.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑