Invention Grant
- Patent Title: 基于对比学习的CTR预测模型训练方法、装置及电子设备
-
Application No.: CN202310834678.9Application Date: 2023-07-10
-
Publication No.: CN116562359BPublication Date: 2023-11-10
- Inventor: 董辉 , 王芳
- Applicant: 深圳须弥云图空间科技有限公司
- Applicant Address: 广东省深圳市南山区粤海街道海珠社区滨海大道3369号有线信息传输大厦25F2504
- Assignee: 深圳须弥云图空间科技有限公司
- Current Assignee: 深圳须弥云图空间科技有限公司
- Current Assignee Address: 广东省深圳市南山区粤海街道海珠社区滨海大道3369号有线信息传输大厦25F2504
- Agency: 北京嘉科知识产权代理事务所
- Agent 马瑞
- Main IPC: G06N3/0895
- IPC: G06N3/0895 ; G06Q30/0202
Abstract:
本申请提供一种基于对比学习的CTR预测模型训练方法、装置及电子设备。该方法包括:将用于CTR预测模型训练的离散特征映射到低维度的稠密向量中,得到原始特征向量;对原始特征向量进行正则化约束,得到特征对齐损失和特征一致性损失;对原始特征向量进行数据增强,得到第一特征向量和第二特征向量,对第一特征向量和第二特征向量进行特征交叉得到中间向量,计算中间向量之间的距离,得到对比学习损失;依据特征对齐损失、特征一致性损失、对比学习损失以及原始CTR预测任务的损失函数生成综合损失函数,利用综合损失函数反向更新模型参数,以便对CTR预测模型进行训练。本申请提高了CTR预测模型训练的泛化性能,从而提升CTR预测模型的预测性能及预测精度。
Public/Granted literature
- CN116562359A 基于对比学习的CTR预测模型训练方法、装置及电子设备 Public/Granted day:2023-08-08
Information query