🚀 4M: 大规模多模态掩码建模
一个用于训练任意到任意多模态基础模型的框架。
可扩展。开源。涵盖数十种模态和任务。
项目网站
| GitHub仓库
| BibTeX引用
以下论文的官方实现和预训练模型:
4M: Massively Multimodal Masked Modeling,NeurIPS 2023(亮点论文)
David Mizrahi*,Roman Bachmann*,Oğuzhan Fatih Kar,Teresa Yeo,Mingfei Gao,Afshin Dehghan,Amir Zamir
4M-21: An Any-to-Any Vision Model for Tens of Tasks and Modalities,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: Massively Multimodal Masked Modeling”(此处表示为4M - 7)以及 “4M - 21: An Any - to - Any Vision Model for Tens of Tasks and Modalities”(此处表示为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_XL')
请参阅 README_GENERATION.md 以获取更详细的说明,以及访问 https://github.com/apple/ml-4m 以获取其他4M模型和分词器的检查点。
📚 详细文档
📄 引用
如果您发现这个仓库很有帮助,请考虑引用我们的工作:
@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},
}
📄 许可证
此仓库中的模型权重根据 LICENSE 文件中的示例代码许可证发布。