mlsanityData to 14 Oct 2026

Experts-Guided Unbalanced Optimal Transport for ISP Learning from Unpaired and/or Paired Data

arXiv 2512.05635 ↗First posted 5 Dec 2025cs.CVcited by 1

Not on the map. Why?

Abstract (arXiv)

Learned Image Signal Processing (ISP) pipelines offer powerful end-to-end performance but are critically dependent on large-scale paired raw-to-sRGB datasets. This reliance on costly-to-acquire paired data remains a significant bottleneck. To address this challenge, we introduce a novel, unsupervised training framework based on Optimal Transport capable of training arbitrary ISP architectures in both unpaired and paired modes. We are the first to successfully apply Unbalanced Optimal Transport (UOT) for this complex, cross-domain translation task. Our UOT-based framework provides robustness to outliers in the target sRGB data, allowing it to discount atypical samples that would be prohibitively costly to map. A key component of our framework is a novel ``committee of expert discriminators,'' a hybrid adversarial regularizer. This committee guides the optimal transport mapping by providing specialized, targeted gradients to correct specific ISP failure modes, including color fidelity, structural artifacts, and frequency-domain realism. To demonstrate the superiority of our approach, we retrained existing state-of-the-art ISP architectures using our paired and unpaired setups. Our experiments show that while our framework, when trained in paired mode, exceeds the performance of the original paired methods across all metrics, our unpaired mode concurrently achieves quantitative and qualitative performance that rivals, and in some cases surpasses, the original paired-trained counterparts. The code and pre-trained models are available at: https://github.com/gosha20777/EGUOT-ISP.git.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 63 of its 85 references that are in the map. First posted on arXiv 5 Dec 2025.

Builds on

63 of 85 references.

References · 63

  1. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  2. 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,713
  3. 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,177
  4. 2022A ConvNet for the 2020scited by 3,579
  5. 2016Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Networkcited by 1,908
  6. 2021Restormer: Efficient Transformer for High-Resolution Image Restorationcited by 1,627
  7. 2019StarGAN v2: Diverse Image Synthesis for Multiple Domainscited by 1,098
  8. 2017StarGAN: Unified Generative Adversarial Networks for Multi-Domain Image-to-Image Translationcited by 1,039
  9. 2020Contrastive Learning for Unpaired Image-to-Image Translationcited by 639
  10. 2017Unsupervised Image-to-Image Translation Networkscited by 603
  11. 2018Learning to See in the Darkcited by 460
  12. 2019SEAN: Image Synthesis with Semantic Region-Adaptive Normalizationcited by 253
38 more
  1. 2020CycleISP: Real Image Restoration via Improved Data Synthesiscited by 213
  2. 2019Optimal transport mapping via input convex neural networkscited by 206
  3. 2019U-GAT-IT: Unsupervised Generative Attentional Networks with Adaptive Layer-Instance Normalization for Image-to-Image Translationcited by 186
  4. 2017DSLR-Quality Photos on Mobile Devices with Deep Convolutional Networkscited by 156
  5. 2024InstructIR: High-Quality Image Restoration Following Human Instructionscited by 140
  6. 2022Neural Optimal Transportcited by 126
  7. 2017Stabilizing Training of Generative Adversarial Networks through Regularizationcited by 118
  8. 2021Image-to-Image Translation: Methods and Applicationscited by 116
  9. 2020Replacing Mobile Camera ISP with a Single Deep Learning Modelcited by 108
  10. 2019Wasserstein-2 Generative Networkscited by 97
  11. 2021Invertible Image Signal Processingcited by 94
  12. 2018DeepISP: Towards Learning an End-to-End Image Processing Pipelinecited by 88
  13. 2019Handheld Multi-Frame Super-Resolutioncited by 85
  14. 2020HistoGAN: Controlling Colors of GAN-Generated and Real Images via Color Histogramscited by 64
  15. 2021Mobile Computational Photography: A Tourcited by 62
  16. 2020On Unbalanced Optimal Transport: An Analysis of Sinkhorn Algorithmcited by 61
  17. 2020Deep White-Balance Editingcited by 54
  18. 2022SepLUT: Separable Image-adaptive Lookup Tables for Real-time Image Enhancementcited by 50
  19. 2022Entropic Neural Optimal Transport via Diffusion Processescited by 50
  20. 2020AIM 2020 Challenge on Learned Image Signal Processing Pipelinecited by 44
  21. 2020Unpaired Image-to-Image Translation using Adversarial Consistency Losscited by 43
  22. 2021Learning RAW-to-sRGB Mappings with Inaccurately Aligned Supervisioncited by 43
  23. 2023Generative Modeling through the Semi-dual Formulation of Unbalanced Optimal Transportcited by 41
  24. 2020AWNet: Attentive Wavelet Network for Image ISPcited by 38
  25. 2016Fast Fourier Color Constancycited by 37
  26. 2023Neural Preset for Color Style Transfercited by 36
  27. 2021Learned Smartphone ISP on Mobile NPUs with Deep Learning, Mobile AI 2021 Challenge: Reportcited by 34
  28. 20224D LUT: Learnable Context-Aware 4D Lookup Table for Image Enhancementcited by 34
  29. 2018Sequential Attention GAN for Interactive Image Editingcited by 33
  30. 2023NILUT: Conditional Neural Implicit 3D Lookup Tables for Image Enhancementcited by 31
  31. 2024Regional Style and Color Transfercited by 30
  32. 2020Learning to Autofocuscited by 26
  33. 2021Beyond Joint Demosaicking and Denoising: An Image Processing Pipeline for a Pixel-bin Image Sensorcited by 22
  34. 2024LYT-NET: Lightweight YUV Transformer-based Network for Low-light Image Enhancementcited by 22
  35. 2022Kantorovich Strikes Back! Wasserstein GANs are not Optimal Transport?cited by 19
  36. 2022Perceptual Image Enhancement for Smartphone Real-Time Applicationscited by 18
  37. 2021Semi-Supervised Raw-to-Raw Mappingcited by 16
  38. 2024Neural Optimal Transport with Lagrangian Costscited by 16

50 most cited of 63 All 63 (JSON)

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.