🚀 CLIP ViT - B - 16(基於DataComp - 1B訓練)模型卡片
本模型是使用OpenCLIP在DataComp - 1B數據集上訓練的CLIP ViT - B/16模型,可用於零樣本圖像分類等研究場景,助力研究人員探索圖像分類相關問題。
🚀 快速開始
若要開始使用此模型,請參考:https://github.com/mlfoundations/open_clip
✨ 主要特性
- 多用途:可用於零樣本圖像分類、圖像和文本檢索等直接任務,也能用於圖像分類微調、線性探針圖像分類、圖像生成引導和條件設定等下游任務。
- 研究導向:旨在作為研究成果供研究社區使用,幫助研究人員更好地理解和探索零樣本、任意圖像分類。
📦 安裝指南
文檔未提及具體安裝步驟,暫不提供。
💻 使用示例
文檔未提供代碼示例,暫不展示。
📚 詳細文檔
🔍 模型詳情
模型描述
這是一個使用OpenCLIP(https://github.com/mlfoundations/open_clip)在DataComp - 1B(https://github.com/mlfoundations/datacomp)數據集上訓練的CLIP ViT - B/16模型。模型訓練在stability.ai集群上完成。
📋 用途
直接用途
可用於零樣本圖像分類、圖像和文本檢索等。
下游用途
可用於圖像分類及其他圖像任務的微調、線性探針圖像分類、圖像生成引導和條件設定等。
非適用場景
與OpenAI模型類似,目前該模型的任何部署用例(無論是否商用)都不在適用範圍內。非部署用例,如在受限環境中的圖像搜索,除非對模型進行了特定固定類別分類法的全面領域測試,否則也不建議使用。因為安全評估表明,由於CLIP在不同類別分類法下的性能存在差異,所以非常需要進行特定任務測試。這使得在任何用例中未經測試和不受約束地部署該模型目前都可能有害。此外,某些屬於監控和人臉識別領域的用例,無論模型性能如何,始終不在適用範圍內。因為目前缺乏確保其公平使用的測試規範和檢查,使用人工智能進行此類任務還為時過早。
🛠️ 訓練詳情
訓練數據
該模型使用DataComp - 1B數據集的14億個樣本進行訓練(https://arxiv.org/abs/2304.14108)。
⚠️ 重要提示
創建此數據集的動機是為了推動大規模多模態模型訓練以及處理從公開互聯網爬取的未整理大規模數據集的研究和實驗。因此,建議僅將該數據集用於研究目的。需注意,這個大規模數據集是未整理的,收集的鏈接可能會指向讓人類觀看者極為不適和不安的內容。所以,請謹慎使用演示鏈接,並自行承擔風險。可以通過基於安全標籤過濾樣本(使用我們構建的自定義訓練的NSFW分類器)來提取一個“安全”子集。雖然這大大降低了查看時遇到潛在有害內容的可能性,但我們不能完全排除安全模式下仍存在有害內容的可能性,因此警告仍然適用。我們認為,向廣泛的研究和其他感興趣的社區公開提供該數據集,將有助於透明地研究訓練大規模模型帶來的好處,以及在使用僅限於小社區的封閉大型數據集時可能未被報告或注意到的陷阱和危險。然而,我們不建議使用該數據集來創建現成的工業產品,因為關於此類大規模模型的一般特性和安全性的基礎研究仍在進行中,我們希望通過此次發佈來鼓勵這方面的研究。
訓練過程
請參考:https://arxiv.org/abs/2304.14108
📊 評估
評估在38個數據集上進行,使用了DataComp倉庫和[LAION CLIP基準測試](https://github.com/LAION - AI/CLIP_benchmark)。
測試數據、因素和指標
測試數據
測試在38個數據集上進行。更多詳細信息請參閱我們的論文(https://arxiv.org/abs/2304.14108)。
結果
該模型在ImageNet - 1k上實現了73.5%的零樣本top - 1準確率。更多詳細信息和結果請參閱我們的論文(https://arxiv.org/abs/2304.14108)。
🙏 致謝
感謝stability.ai提供訓練此模型的計算資源。
📑 引用
DataComp
@article{datacomp,
title={DataComp: In search of the next generation of multimodal datasets},
author={Samir Yitzhak Gadre, Gabriel Ilharco, Alex Fang, Jonathan Hayase, Georgios Smyrnis, Thao Nguyen, Ryan Marten, Mitchell Wortsman, Dhruba Ghosh, Jieyu Zhang, Eyal Orgad, Rahim Entezari, Giannis Daras, Sarah Pratt, Vivek Ramanujan, Yonatan Bitton, Kalyani Marathe, Stephen Mussmann, Richard Vencu, Mehdi Cherti, Ranjay Krishna, Pang Wei Koh, Olga Saukh, Alexander Ratner, Shuran Song, Hannaneh Hajishirzi, Ali Farhadi, Romain Beaumont, Sewoong Oh, Alex Dimakis, Jenia Jitsev, Yair Carmon, Vaishaal Shankar, Ludwig Schmidt},
journal={arXiv preprint arXiv:2304.14108},
year={2023}
}
OpenAI CLIP論文
@inproceedings{Radford2021LearningTV,
title={Learning Transferable Visual Models From Natural Language Supervision},
author={Alec Radford and Jong Wook Kim and Chris Hallacy and A. Ramesh and Gabriel Goh and Sandhini Agarwal and Girish Sastry and Amanda Askell and Pamela Mishkin and Jack Clark and Gretchen Krueger and Ilya Sutskever},
booktitle={ICML},
year={2021}
}
OpenCLIP軟件
@software{ilharco_gabriel_2021_5143773,
author = {Ilharco, Gabriel and
Wortsman, Mitchell and
Wightman, Ross and
Gordon, Cade and
Carlini, Nicholas and
Taori, Rohan and
Dave, Achal and
Shankar, Vaishaal and
Namkoong, Hongseok and
Miller, John and
Hajishirzi, Hannaneh and
Farhadi, Ali and
Schmidt, Ludwig},
title = {OpenCLIP},
month = jul,
year = 2021,
note = {If you use this software, please cite it as below.},
publisher = {Zenodo},
version = {0.1},
doi = {10.5281/zenodo.5143773},
url = {https://doi.org/10.5281/zenodo.5143773}
}
📄 許可證
本項目採用MIT許可證。