一种模型训练方法及装置

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种模型训练方法及装置
申请号:CN202411946235
申请日期:2024-12-27
公开号:CN120047765A
公开日期:2025-05-27
类型:发明专利
摘要
本申请实施例公开了一种模型训练方法及装置。该方法包括:将训练图像输入至预训练自编码器中的编码器得到第一压缩图像编码,将训练图像输入至目标视觉基础模型得到第二压缩图像编码;其中,预训练自编码器包括编码器和解码器;根据第一压缩图像编码和第二压缩图像编码,确定对比损失值,并将对比损失值和至少一个其他损失值作为目标损失值;其中,至少一个其他损失值为反映预训练自编码器性能的损失值;基于目标损失值对预训练自编码器进行训练优化。上述方案能够以目标视觉基础模型进行图像压缩的结果为参考,对预训练自编码器进行训练优化,提高了自编码器的编解码效果,能够通过自编码器获取更多、更精准的原始图像信息。
技术关键词
图像编码 编码器 视觉 模型训练方法 基础 解码图像 图像压缩 元素 偏差 模型训练装置 输入解码器 生成对抗网络 线性 数据 输入模块 编解码 数值