一种基于预训练模型微调的相似视频检索方法及存储介质

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种基于预训练模型微调的相似视频检索方法及存储介质
申请号:CN202410907960
申请日期:2024-07-08
公开号:CN118885635A
公开日期:2024-11-01
类型:发明专利
摘要
本申请实施例提供了一种基于预训练模型微调的相似视频检索方法,包括S10,构建微调样本,使微调样本同时满足分类监督学习目标及对比学习非监督目标的训练;S20,使用微调样本对SBERT模型进行分类监督学习目标和对比学习非监督目标的训练,得到相似度模型;S30,获取媒资库中视频所对应的文本信息特征,通过文本信息特征构建用于表达视频信息的视频文本特征;S40,通过训练得到的相似度模型进行预测,输出视频对应的高阶特征向量;S50,将需要寻找相似视频的视频作为种子视频,对种子视频对应的高阶特征向量与其他视频的高阶特征向量进行余弦相似度计算,得到相似视频;具有检索效率和准确度高的有益效果,适用于大数据处理的技术领域。
技术关键词
视频检索方法 预训练模型 样本 非监督 种子 读取标签信息 文本特征值 BERT模型 更新模型参数 中文文本 列表 标签类别 表达式 可读存储介质 指数 成品 定义