🚀 4M:大规模多模态掩码建模
一个用于训练任意到任意多模态基础模型的框架。
可扩展。开源。涵盖数十种模态和任务。
项目网站
| GitHub仓库
| 引用格式
以下论文的官方实现和预训练模型:
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)的代码和模型。
🚀 快速开始
4M可用于训练“任意到任意”的基础模型,利用分词和掩码技术实现多模态扩展。使用4M训练的模型具备广泛的视觉任务处理能力,能灵活迁移至新的任务和模态。
📦 安装指南
有关安装说明,请参阅 https://github.com/apple/ml-4m。
💻 使用示例
基础用法
可以从Hugging Face Hub加载此模型,示例代码如下:
from fourm.models.fm import FM
fm = FM.from_pretrained('EPFL-VILAB/4M-21_B')
更多详细说明请参阅 README_GENERATION.md,其他4M模型和分词器的检查点请访问 https://github.com/apple/ml-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},
}
📄 许可证
此仓库中的模型权重根据许可文件中的示例代码许可证发布。