按 CanvasForThinking《用 PyTorch 实现一个可微光学逆向设计系统》实现。对应 TorchOptics 论文 Figure 4 的问题设定:三块只调相位的调制器,把一束高斯光变成四束高斯光。传播是角谱法,损失是模重叠。
模重叠从 0.0424 升到 0.9964。计算设备是 cpu。
| 量 | 取值 |
|---|---|
| 网格 | 250 × 250 |
| 像素 | 10 μm |
| 波长 | 700 nm |
| 每段传播 | 0.2 m |
| 高斯腰斑 | 150 μm |
| 四束偏移 | 0.5 mm |
| 优化器 | Adam,学习率 0.1,400 步 |
| 设备 | cpu |
| 初始 η | 0.042449 |
| 最终 η | 0.996408 |
| 最终损失 1-η | 0.003592 |