公开(公告)号：CN117216546A

公开(公告)日：2023-12-12

申请号：CN202310638818.5

申请日：2023-05-31

Applicant: 腾讯科技(深圳)有限公司

Inventor： 白云鹏 ,  王鑫涛 ,  曹炎培 ,  葛艺潇 ,  袁春 ,  单瀛

IPC: G06F18/214 ,  G06F18/22 ,  G06F18/15 ,  G06F18/213 ,  G06T11/00 ,  G06T5/00 ,  G06T9/00

Abstract: 本申请提供了一种脑电生成图像模型训练方法、装置、设备及计算机可读存储介质；方法包括：获取脑电信号样本以及第一图像样本；通过脑电编码器对脑电信号样本进行编码，得到脑电编码，通过图像编码器对第一图像样本进行编码，得到图像编码；通过图像生成网络对噪声图像进行以脑电信号样本的脑电编码为条件的去噪，得到预测去噪结果；基于预测去噪结果与第一真实去噪结果的误差构建第一去噪损失，基于脑电编码与图像编码之间的距离构建脑电图像对齐损失；基于第一去噪损失以及脑电图像对齐损失，对脑电编码器以及图像生成网络进行参数更新。通过本申请，能够通过对齐脑电信号维度以及图像维度，从而优化模型的图像生成能力。