검색으로 돌아가기
기록
一种扩散模型去噪过程分块并行加速方法及系统
발명유효
10청구항 · 4 독립항
§ Ⅰ
개요
발명자
李开放; 谢人超; 唐琴琴; 游子越
IPC 분류
G06T 5/70 (2024.01)G06T 5/60 (2024.01)G06N 3/045 (2023.01)G06N 3/0442 (2023.01)
本发明提供一种扩散模型去噪过程分块并行加速方法及系统,所述方法包括:在预热阶段,将输入图像均匀分割并分配至各计算设备,利用各计算设备本地的扩散模型通过同步通信方式完成噪声预测与图像更新,按预设次数重复并记录资源利用时序数据;基于预训练的资源利用率预测模型预测各计算设备的未来资源利用率,生成资源利用率预测时序数据;据此计算每一轮的非均匀图像分割方案;在非均匀循环推理阶段,按该方案分割图像,各设备异步获取相邻子图边缘数据及通过滑动观察窗口处理提取全局激活数据,进行局部噪声预测,并通过广播同步通信聚合噪声更新图像,直至去噪完成。本发明实现了算力不相等节点下的高效并行推理,提升了资源利用率和加速效果。