基于加速扩散模型的缺失值插补算法
|
|
王圣举,张赞
|
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 | — |
|
|
|