R2Disk2

Checkpoints of R2D2 (5 iterations) for radio-interferometric image reconstruction of synthetic protoplanetary disks.

• Architecture: R2D2 (iterative residual U-Net, 5 iterations)
• Training: 4000 synthetic disks + 600 validation
• Test results (n=360): PSNR 37.72 dB, SSIM 0.949, 1.5 s/image
• Best epoch per iteration: N1=242, N2=143, N3=79, N4=137, N5=141

Each file epoch=*_N<i>.ckpt is the checkpoint of iteration i of the chain; inference with the full chain uses all five.

Code and benchmark: https://github.com/MauroRitacco/RI-Benchmark R2D2 GitHub (BASP group): https://github.com/basp-group/R2D2-RI/