🚀 4M:大规模多模态掩码建模
一个用于训练任意到任意多模态基础模型的框架。
可扩展。开源。涵盖数十种模态和任务。
项目网站
| GitHub仓库
| 引用格式
以下论文的官方实现和预训练模型:
4M:大规模多模态掩码建模,NeurIPS 2023(亮点论文)
David Mizrahi*,Roman Bachmann*,Oğuzhan Fatih Kar,Teresa Yeo,Mingfei Gao,Afshin Dehghan,Amir Zamir
4M - 21:适用于数十种任务和模态的任意到任意视觉模型,arXiv 2024
Roman Bachmann*,Oğuzhan Fatih Kar*,David Mizrahi*,Ali Garjani,Mingfei Gao,David Griffiths,Jiaming Hu,Afshin Dehghan,Amir Zamir
4M是一个用于训练“任意到任意”基础模型的框架,它通过分词和掩码技术来扩展到多种不同的模态。
使用4M训练的模型可以执行广泛的视觉任务,能够很好地迁移到未见的任务和模态,并且是灵活且可引导的多模态生成模型。
我们发布了“4M:大规模多模态掩码建模”(此处表示为4M - 7)以及“4M - 21:适用于数十种任务和模态的任意到任意视觉模型”(此处表示为4M - 21)的代码和模型。
🚀 快速开始
✨ 主要特性
- 支持训练“任意到任意”的多模态基础模型。
- 可扩展到多种不同的模态和任务。
- 模型具有良好的迁移能力和生成能力。
📦 安装指南
安装说明请参考:https://github.com/apple/ml-4m 。
💻 使用示例
基础用法
该模型可以从Hugging Face Hub加载,示例代码如下:
from fourm.models.fm import FM
fm = FM.from_pretrained('EPFL-VILAB/4M-21_L')
更多详细说明请参考 README_GENERATION.md,其他4M模型和分词器的检查点请参考:https://github.com/apple/ml-4m 。
📚 详细文档
目前文档主要包含安装和使用说明,更多详细文档可参考上述链接。
📄 许可证
本仓库中的模型权重遵循 LICENSE 文件中的示例代码许可证。
📄 引用
如果您觉得本仓库有帮助,请考虑引用我们的工作:
@inproceedings{4m,
title={{4M}: Massively Multimodal Masked Modeling},
author={David Mizrahi and Roman Bachmann and O{\u{g}}uzhan Fatih Kar and Teresa Yeo and Mingfei Gao and Afshin Dehghan and Amir Zamir},
booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
year={2023},
}
@article{4m21,
title={{4M-21}: An Any-to-Any Vision Model for Tens of Tasks and Modalities},
author={Roman Bachmann and O{\u{g}}uzhan Fatih Kar and David Mizrahi and Ali Garjani and Mingfei Gao and David Griffiths and Jiaming Hu and Afshin Dehghan and Amir Zamir},
journal={arXiv 2024},
year={2024},
}