: Reconstruction, Raw, and Rain: Deraining Directly in the Bayer Domain
Organizations: Department of ECE, Technion, Israel · Department of CS, Technion, Israel · School of ECE, Ben-Gurion University
Abstract
Image reconstruction from corrupted images is crucial across many domains. Most reconstruction networks are trained on post-ISP sRGB images, even though the image-signal-processing pipeline irreversibly mixes colors, clips dynamic range, and blurs fine detail. This paper uses the rain degradation problem as a use case to show that these losses are avoidable, and demonstrates that learning directly on raw Bayer mosaics yields superior reconstructions. To substantiate the claim, we (i) evaluate post-ISP and Bayer reconstruction pipelines, (ii) curate Raw-Rain, the first public benchmark of real rainy scenes captured in both 12-bit Bayer and bit-depth-matched sRGB, and (iii) introduce Information Conservation Score (ICS), a color-invariant metric that aligns more closely with human opinion than PSNR or SSIM. On the test split, our raw-domain model improves sRGB results by up to +0.99 dB PSNR and +1.2% ICS, while running faster with half of the GFLOPs. The results advocate an ISP-last paradigm for low-level vision and open the door to end-to-end learnable camera pipelines.
Figures & tables
| Subject | Bayer | RGB | ICS | SSIM | PSNR |
|---|---|---|---|---|---|
| Avg. | 76.4% | 23.6% | 77.2 % | 73.4% | 71.7% |
| Split | Scenes | Frames | Rain dens. | Illum. |
|---|---|---|---|---|
| Train | 86 | 48 000 | light–heavy | day+evening |
| Identity | 30 | 18 000 | None | day |
| Val | 15 | 9 000 | light–heavy. | day+evening |
| Test | 10 | 6 000 | light–heavy | day+evening |
| Model | Domain | Scene 1 | Scene 2 | Scene 3 | ||||||
|---|---|---|---|---|---|---|---|---|---|---|
| PSNR | SSIM | ICS↑ | PSNR | SSIM | ICS↑ | PSNR | SSIM | ICS↑ | ||
| CBAM | Bayer | 46.742 | 0.965 | 0.945 | 51.781 | 0.993 | 0.964 | 41.472 | 0.942 | 0.864 |
| RGB | 44.87 | 0.957 | 0.925 | 47.867 | 0.986 | 0.948 | 40.257 | 0.917 | 0.852 | |
| UNET | Bayer | 45.91 | 0.962 | 0.939 | 52.084 | 0.992 | 0.963 | 41.562 | 0.943 | 0.874 |
| RGB | 45.911 | 0.973 | 0.937 | 50.663 | 0.992 | 0.96 | 41.01 | 0.934 | 0.876 | |
| MONO | Bayer | 47.053 | 0.972 | 0.948 | 51.973 | 0.993 | 0.965 | 41.945 | 0.94 | 0.879 |