基于CLIP的中文图文检索模型训练方法、装置、设备及介质

摘要

本发明涉及人工智能技术，揭露一种基于CLIP的中文图文检索模型训练方法，包括：利用预构建的中文编码器生成中文正样本的正样本向量及负样本的负样本向量集，计算正样本向量之间的第一相似度及正样本向量与负样本向量集之间的第二相似度，根据所述第一相似度及所述第二相似度调整该中文编码器的参数，直到该中文编码器符合第一预设训练条件，将完成训练的中文编码器替代预构建的CLIP检索模型中的英文编码器，得到中文图文CLIP检索模型，利用中文图文对训练集对所述中文图文CLIP检索模型进行图文匹配训练，得到目标中文图文CLIP检索模型。本发明还提出一种基于CLIP的中文图文检索模型训练装置、设备以及介质。本发明可以提升CLIP图文检索模型的中文图文检索性能。