arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.26786cs.LGcs.AI
收藏

迈向多模态大语言模型联邦预训练的一小步

A Step Toward Federated Pretraining of Multimodal Large Language Models

MAIS, Institute of Automation, Chinese Academy of Sciences(中国科学院自动化研究所多模态人工智能系统实验室);King Abdullah University of Science and Technology (KAUST)(阿卜杜拉国王科技大学);Pengcheng Laboratory(鹏城实验室);School of Artificial Intelligence, University of Chinese Academy of Sciences (UCAS)(中国科学院大学人工智能学院)

Baochen Xiong, Yifan Xu, Xiaoshan Yang, Yaguang Song, Yaowei Wang, Changsheng Xu

AI总结:

本文提出Fed-MA任务,通过冻结视觉编码器和LLM,协同训练跨模态投影器,解决参数干扰和梯度震荡问题,提出Fed-CMP框架在联邦预训练中取得显著优势。

摘要:

多模态大语言模型(MLLMs)的快速发展受到高质量公开数据饱和的限制,而大量异构多模态数据在隐私敏感的数据孤岛中难以访问。联邦学习(FL)提供了解锁这些分布式资源的有希望的解决方案,但现有研究主要集中在微调上,忽略了基础预训练阶段。本文正式引入联邦MLLM对齐(Fed-MA)任务,一种轻量级预训练范式,冻结视觉编码器和LLM,协同训练跨模态投影器。我们识别出两种关键挑战:(i)在聚合本地投影器时的参数干扰;(ii)单次传递协作SGD中的梯度震荡。为了解决这些挑战,我们提出Fed-CMP,一种用于联邦MLLM预训练的开创性框架。Fed-CMP采用Canonical Reliability-Aware Aggregation,构建一个规范空间将客户端投影器分解为共享对齐基和客户端特定系数,然后进行可靠性加权融合以抑制参数干扰。此外,Fed-CMP引入Orthogonality-Preserved Momentum,通过正交投影对共享对齐基应用动量,积累历史优化方向同时保持几何结构。我们基于公开数据集构建了四个联邦预训练场景,大量实验验证Fed-CMP显著优于现有基线。

英文摘要:

The rapid evolution of Multimodal Large Language Models (MLLMs) is bottlenecked by the saturation of high-quality public data, while vast amounts of diverse multimodal data remain inaccessible in privacy-sensitive silos. Federated Learning (FL) offers a promising solution to unlock these distributed resources, but existing research focuses predominantly on fine-tuning, leaving the foundational pre-training phase largely unexplored. In this paper, we formally introduce the Federated MLLM Alignment (Fed-MA) task, a lightweight pre-training paradigm that freezes the vision encoder and LLM while collaboratively training the cross-modal projector. We identify two critical challenges in this setting: (i) parameter interference in aggregating local projectors; and (ii) gradient oscillations in one-pass collaborative SGD. To address these challenges, we propose Fed-CMP, a pioneering framework for federated MLLM pre-training. Fed-CMP employs Canonical Reliability-Aware Aggregation, which constructs a canonical space to decompose client projectors into a shared alignment basis and client-specific coefficients, then performs reliability-weighted fusion to suppress parameter interference. Furthermore, Fed-CMP introduces Orthogonality-Preserved Momentum, which applies momentum to the shared alignment basis via orthogonal projection, accumulating historical optimization directions while preserving geometric structure. We construct four federated pre-training scenarios based on public datasets, and extensive experiments validate that Fed-CMP significantly outperforms existing baselines.

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

↑