基于加速扩散模型的缺失值插补算法
|
王圣举,张赞
|
Missing value imputation algorithm based on accelerated diffusion model
|
Shengju WANG,Zan ZHANG
|
|
表 1 基于加速扩散模型的数据插补方法的网络参数 |
Tab.1 Network parameter of accelerated diffusion model-based imputation method |
|
模块阶段 | 模块名称 | K | S | P | Cout | H | 初始层 | 卷积层 | 3×3 | 1 | 1×1 | 16 | — | 编码器 | 残差块 | 3×3 | 1 | 1×1 | 16 | — | 3×3 | 1 | 1×1 | 16 | — | 残差块 | 3×3 | 1 | 1×1 | 16 | — | 3×3 | 1 | 1×1 | 16 | — | 自注意力块 | — | — | — | — | 4 | 下采样块(非最后一个层级) | 5×5 | 2 | 1×1 | 128 | — | 下采样块(最后一个层级) | 3×3 | 1 | 1×1 | 128 | — | 中间层 | 残差块 | 3×3 | 1 | 1×1 | 128 | — | 3×3 | 1 | 1×1 | 128 | — | 自注意力块 | — | — | — | — | 4 | 残差块 | 3×3 | 1 | 1×1 | 128 | — | 3×3 | 1 | 1×1 | 128 | — | 解码器 | 残差块 | 3×3 | 1 | 1×1 | 128 | — | 3×3 | 1 | 1×1 | 128 | — | 3×3 | 1 | 1×1 | 128 | — | 残差块 | 3×3 | 1 | 1×1 | 128 | — | 3×3 | 1 | 1×1 | 128 | — | 3×3 | 1 | 1×1 | 128 | — | 自注意力块 | — | — | — | — | 4 | 上采样块(非最后一个层级) | 2×2 | 1 | 1×1 | 128 | — | 上采样块(最后一个层级) | 3×3 | 1 | 1×1 | 16 | — | nn.Upsample(scale_factor = 2) | 输出层 | 卷积层 | 1×1 | 1 | 0 | 1 | — |
|
|
|