Using Knowledge Graphs to harvest datasets for efficient CLIP model training
Using Knowledge Graphs to harvest datasets for efficient CLIP model training
Training high-quality CLIP models typically requires enormous datasets, which limits the development of domain-specific models -- especially in areas that even the largest CLIP models do not cover well -- and drives up training costs. This poses challenges for scientific research that needs fine-grained control over the training procedure of CLIP models. In this work, we show that by employing smart web search strategies enhanced with knowledge graphs, a robust CLIP model can be trained from scratch with considerably less data. Specifically, we demonstrate that an expert foundation model for living organisms can be built using just 10M images. Moreover, we introduce EntityNet, a dataset comprising 33M images paired with 46M text descriptions, which enables the training of a generic CLIP model in significantly reduced time.
Johannes Dienert、Hannah Bast、Simon Ging、Sebastian Walter、Jelena Bratuli?、Thomas Brox
生物科学研究方法、生物科学研究技术计算技术、计算机技术
Johannes Dienert,Hannah Bast,Simon Ging,Sebastian Walter,Jelena Bratuli?,Thomas Brox.Using Knowledge Graphs to harvest datasets for efficient CLIP model training[EB/OL].(2025-05-05)[2025-05-28].https://arxiv.org/abs/2505.02746.点此复制
评论