Magma：微软推出的多模态AI代理基础模型，可实现多场景代理

文章来源：08AI导航发布时间：2025-03-13 15:37:56

Magma是什么？

Magma 是微软推出的一款多模态 ai 代理基础模型，能够处理虚拟和现实环境中的复杂交互，实现图像字幕和问答、视频字幕和问答、UI导航、机器人操作等多种任务。

Magma功能特点

多模态能力：支持图像字幕和问答、视频字幕和问答、UI 导航、机器人操作等任务。

数字与物理世界的交互：能够处理虚拟和现实环境中的任务。

多功能性：单一模型具备通用的图像和视频理解能力，同时能生成目标驱动的视觉计划和动作。

先进性能：在多模态任务上表现出色，特别是在空间理解和推理方面。

可扩展的预训练策略：能够从未标记的视频中学习，具有很强的泛化能力。

Magma的技术原理

多模态预训练：结合图像、视频和动作数据，通过统一框架进行大规模预训练，学习跨模态的连接。

Set-of-Mark (SoM)：标记图像中的可操作对象，帮助模型实现动作落地。

Trace-of-Mark (ToM)：标记视频中物体的运动轨迹，增强时间动态理解能力。

视觉与语言结合：使用卷积网络将视觉信息编码为标记序列，与语言模型结合，生成动作或语言描述。

泛化与微调：预训练后的模型具备零样本泛化能力，可通过微调进一步提升性能。

跨任务适应：适用于多种任务（如UI导航、机器人操作、图像和视频理解），展现出强大的泛化能力。

Magma的技术原理.jpg

安装与使用

克隆项目：

gitclonehttps://github.com/microsoft/Magma.gitcdMagma

安装依赖：

condacreate-nmagmapython=3.10-ycondaactivatemagmapipinstall--upgradepippipinstall-e.

推理例子

fromPILimportImageimporttorchfromtransformersimportAutoModelForCausalLM,AutoProcessormodel=AutoModelForCausalLM.from_pretrained("microsoft/Magma-8B",trust_remote_code=True)processor=AutoProcessor.from_pretrained("microsoft/Magma-8B",trust_remote_code=True)model.to("cuda")image=Image.open("example.jpg").convert("RGB")convs=[{"role":"system","content":"Youareanagentthatcansee,talkandact."},{"role":"user","content":" Whatisintheimage?"}]prompt=processor.tokenizer.apply_chat_template(convs,tokenize=False,add_generation_prompt=True)inputs=processor(images=[image],texts=prompt,return_tensors="pt").to("cuda")withtorch.inference_mode():generate_ids=model.generate(**inputs)response=processor.decode(generate_ids[0],skip_special_tokens=True).strip()print(response)