3DUNet N2V FM2S
About
Summary
This method uses a 3-level 3D U-Net (32→64→128→128) with a large receptive field to separate noise from structure, combined with GroupNorm for stable small-batch training and residual prediction to reduce DC-shift bias. It is trained with L1 loss for sharper outputs and 1.5% uniform masking for faster N2V convergence. Mixed precision (AMP) and a cosine LR schedule improve speed and stability. The denoised output is then passed to an FM2S model for further enhancement.
Mechanism
Masked input (1.5% N2V masking) → 3D U-Net (multi-scale context, GroupNorm) → Residual prediction (input + correction) → L1 loss on masked pixels (sharp reconstruction) → AMP + cosine LR (efficient training) → FM2S model (final refinement & enhancement)
Interfaces
This algorithm implements all of the following input-output combinations:
| Inputs | Outputs | |
|---|---|---|
| 1 |
Validation and Performance
Challenge Performance
| Date | Challenge | Phase | Rank |
|---|---|---|---|
| April 8, 2026 | AI4LIFE-CIDC25 | Final Submission Phase: Noise Level Generalization | 4 |
| April 8, 2026 | AI4LIFE-CIDC25 | Final Submission Phase: Content Generalization | 4 |
Uses and Directions
This algorithm was developed for research purposes only.