一、做的什么问题
- 主要任务:完善目前的联邦学习网络
- 次要任务:跑其他代码进行对比
二、上周工作回顾
搭建了大致的FL网络框架,同时将U-Net网络转换为FCNVMB,并进行实验。分别跑了基于FCNVMB的FL网络,以及单独对FCNVMB进行训练。
三、本周进展
再次研究DD-Net,对DD-Net中的FCNVMB网络进行手搓。对FCNVMB的网络架构有了更深的认识。
运行此FCNVMB,数据集为CurveFaultA,参数如下
| Learning Rate | 0.001 |
| Epochs | 50 |
| BatchSize | 10 |
| TrainSize | 1600 |
与之前FL工作进行对比:
与之前的结果做对比:
MSE: 0.015033
MAE: 0.084872
LPIPS : 0.107840
UIQ: 0.063430
只有UIQ表现更好。
四、下周工作计划
继续手搓网络。手搓网络对网络的理解与代码水平提升巨大,下一步手搓InversionNet以及DDNet,搓出来之后进行运行并调参。这之后尝试将FL架构中的网络替换为手搓的代码,比对效果。