phone:18157199776 / E-mail: qisheng.chen@xiaohuodui.cn

AI 资讯

Your position:Homepage > AI 资讯 > OneLLM,将所有模态和 LLM Align 的统一框架

OneLLM,将所有模态和 LLM Align 的统一框架

Popularity:31 Add time:2024-01-09 14:44:30

引言

OneLLM使用通用编码器和统一的投影模块与LLM对齐多模式输入,它还利用modality tokens 实现了在模态之间的切换。

20240109142338_18068.

上图展现了OneLLM的四个重要的组件:不同模态的tokenizer,通用编码器,统一的投影模块,大语言模型。

  • 多模态token的tokenizer:将输入的各种模态的信号转换为token序列。

  • 通用编码器:通用编码器是在LAION上训练的CLIP VIT Large。

  • 统一的投影模块(UPM):统一投影模块将各个模块投影到LLM的embedding向量空间中。UPM有K=3个投影专家,每个专家有8个transformers块和88M个参数。

  • 大语言模型:OneLLM采用开源LLaMA2-7B。

OneLLM支持8种不同模态的数据的理解,包括:图像、音频、视频、点云、深度/法线图、IMU 和 fMRI 大脑活动。

1704781851377221.

OneLLM在视频-文本、音频-视频-文本、音频-文本等任务中优于现有方法,凸显了多模理解中的零样本能力。我们相信OneLLM会推动AI在语义理解上的进一步的发展。

20240109142338_38778.

模型使用和链接

OneLLM的模型权重已经开源到魔搭社区

论文链接:

https://arxiv.org/abs/2312.03700

模型权重链接:

https://modelscope.cn/models/csuhan/OneLLM-7B

模型下载代码:

#模型下载
from modelscope import snapshot_download
model_dir = snapshot_download('csuhan/OneLLM-7B')

模型创空间:

https://modelscope.cn/studios/csuhan/OneLLM

image understanding:

20240109142338_18364.

video understanding:

20240109142338_98402.

audio understanding:

20240109142338_20469.


Github代码链接:https://github.com/csuhan/OneLLM

点击跳转Github链接~

csuhan/OneLLM: OneLLM: One Framework to Align All Modalities with Language (github.com)



TAG:

Latest Information

Not search wanted articles

cache
Processed in 0.004487 Second.