skip to main content
10.1145/3647649.3647705acmotherconferencesArticle/Chapter ViewAbstractPublication PagesicigpConference Proceedingsconference-collections
research-article

Image Dehazing based on Iterative-Refining Diffusion Model

Published:03 May 2024Publication History

ABSTRACT

One of the hardest problems in computer vision is image dehazing. In the single picture dehazing job, the approaches based on pixel domain mapping and prior knowledge of physical models have achieved amazing results. Sadly, most deep dehazing algorithms now in use have limited generalization capabilities, which makes it challenging to apply them to data samples with foggy conditions that exhibit a wide range of variation. We suggest an Iterative-Refining Diffusion Model built on the U-Net architecture to solve the issue. We show that the suggested approach may be used to the dehazing problem. It is based on Denoising Diffusion Probabilistic Models (DDPM) [14] and the denoising score matching. An empirical data distribution is created from the conventional normal distribution by a sequence of repeated refining stages that are comparable to the Langevin dynamics process. The U-Net architecture [27], the model’s network architecture, is trained with dehazing targets to progressively eliminate different haze levels from the output. Extensive analyses demonstrate that the proposed model outperforms the state-of-the-art methods on multiple benchmarks.

References

  1. Cosmin Ancuti, Codruta O Ancuti, and Christophe De Vleeschouwer. 2016. D-hazy: A dataset to evaluate quantitatively dehazing algorithms. In 2016 IEEE international conference on image processing (ICIP). IEEE, 2226–2230.Google ScholarGoogle ScholarCross RefCross Ref
  2. Georgios Batzolis, Jan Stanczuk, Carola-Bibiane Schönlieb, and Christian Etmann. 2021. Conditional image generation with score-based diffusion models. arXiv preprint arXiv:2111.13606 (2021).Google ScholarGoogle Scholar
  3. Andrew Brock, Jeff Donahue, and Karen Simonyan. 2018. Large scale GAN training for high fidelity natural image synthesis. arXiv preprint arXiv:1809.11096 (2018).Google ScholarGoogle Scholar
  4. Bolun Cai, Xiangmin Xu, Kui Jia, Chunmei Qing, and Dacheng Tao. 2016. Dehazenet: An end-to-end system for single image haze removal. IEEE transactions on image processing 25, 11 (2016), 5187–5198.Google ScholarGoogle Scholar
  5. Dongdong Chen, Mingming He, Qingnan Fan, Jing Liao, Liheng Zhang, Dongdong Hou, Lu Yuan, and Gang Hua. 2019. Gated context aggregation network for image dehazing and deraining. In 2019 IEEE winter conference on applications of computer vision (WACV). IEEE, 1375–1383.Google ScholarGoogle Scholar
  6. Nanxin Chen, Yu Zhang, Heiga Zen, Ron J Weiss, Mohammad Norouzi, and William Chan. 2020. Wavegrad: Estimating gradients for waveform generation. arXiv preprint arXiv:2009.00713 (2020).Google ScholarGoogle Scholar
  7. Jia Deng, Wei Dong, Richard Socher, Li-Jia Li, Kai Li, and Li Fei-Fei. 2009. Imagenet: A large-scale hierarchical image database. In 2009 IEEE conference on computer vision and pattern recognition. Ieee, 248–255.Google ScholarGoogle ScholarCross RefCross Ref
  8. Zijun Deng, Lei Zhu, Xiaowei Hu, Chi-Wing Fu, Xuemiao Xu, Qing Zhang, Jing Qin, and Pheng-Ann Heng. 2019. Deep multi-model fusion for single-image dehazing. In Proceedings of the IEEE/CVF international conference on computer vision. 2453–2462.Google ScholarGoogle ScholarCross RefCross Ref
  9. Patrick Esser, Robin Rombach, and Bjorn Ommer. 2021. Taming transformers for high-resolution image synthesis. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 12873–12883.Google ScholarGoogle ScholarCross RefCross Ref
  10. Raanan Fattal. 2014. Dehazing using color-lines. ACM transactions on graphics (TOG) 34, 1 (2014), 1–14.Google ScholarGoogle Scholar
  11. Yosef Gandelsman, Assaf Shocher, and Michal Irani. 2019. " Double-DIP": unsupervised image decomposition via coupled deep-image-priors. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 11026–11035.Google ScholarGoogle ScholarCross RefCross Ref
  12. Alona Golts, Daniel Freedman, and Michael Elad. 2019. Unsupervised single image dehazing using dark channel prior loss. IEEE transactions on Image Processing 29 (2019), 2692–2701.Google ScholarGoogle Scholar
  13. Yuanbiao Gou, Boyun Li, Zitao Liu, Songfan Yang, and Xi Peng. 2020. Clearer: Multi-scale neural architecture search for image restoration. Advances in neural information processing systems 33 (2020), 17129–17140.Google ScholarGoogle Scholar
  14. Jonathan Ho, Ajay Jain, and Pieter Abbeel. 2020. Denoising diffusion probabilistic models. Advances in neural information processing systems 33 (2020), 6840–6851.Google ScholarGoogle Scholar
  15. Jonathan Ho, Chitwan Saharia, William Chan, David J Fleet, Mohammad Norouzi, and Tim Salimans. 2022. Cascaded diffusion models for high fidelity image generation. The Journal of Machine Learning Research 23, 1 (2022), 2249–2281.Google ScholarGoogle ScholarDigital LibraryDigital Library
  16. Tao Huang, Songjiang Li, Xu Jia, Huchuan Lu, and Jianzhuang Liu. 2021. Neighbor2neighbor: Self-supervised denoising from single noisy images. In Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 14781–14790.Google ScholarGoogle ScholarCross RefCross Ref
  17. Tong-yao JIA, Li ZHUO, Jia-feng LI, and Jing ZHANG. 2023. Research Advances on Deep Learning Based Single Image Dehazing. ACTA ELECTONICA SINICA 51, 1 (2023), 231.Google ScholarGoogle Scholar
  18. Zhi Jin, Muhammad Zafar Iqbal, Dmytro Bobkov, Wenbin Zou, Xia Li, and Eckehard Steinbach. 2019. A flexible deep CNN framework for image restoration. IEEE Transactions on Multimedia 22, 4 (2019), 1055–1068.Google ScholarGoogle ScholarDigital LibraryDigital Library
  19. Diederik P Kingma and Max Welling. 2013. Auto-encoding variational bayes. arXiv preprint arXiv:1312.6114 (2013).Google ScholarGoogle Scholar
  20. Boyi Li, Xiulian Peng, Zhangyang Wang, Jizheng Xu, and Dan Feng. 2017. Aod-net: All-in-one dehazing network. In Proceedings of the IEEE international conference on computer vision. 4770–4778.Google ScholarGoogle ScholarCross RefCross Ref
  21. Boyi Li, Wenqi Ren, Dengpan Fu, Dacheng Tao, Dan Feng, Wenjun Zeng, and Zhangyang Wang. 2018. Benchmarking single-image dehazing and beyond. IEEE Transactions on Image Processing 28, 1 (2018), 492–505.Google ScholarGoogle ScholarDigital LibraryDigital Library
  22. Wanwan Li. 2021. Image Synthesis and Editing with Generative Adversarial Networks (GANs): A Review. In 2021 Fifth World Conference on Smart Trends in Systems Security and Sustainability (WorldS4). IEEE, 65–70.Google ScholarGoogle ScholarCross RefCross Ref
  23. Wei-An Lin, Haofu Liao, Cheng Peng, Xiaohang Sun, Jingdan Zhang, Jiebo Luo, Rama Chellappa, and Shaohua Kevin Zhou. 2019. Dudonet: Dual domain network for ct metal artifact reduction. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. 10512–10521.Google ScholarGoogle ScholarCross RefCross Ref
  24. Xiaohong Liu, Yongrui Ma, Zhihao Shi, and Jun Chen. 2019. Griddehazenet: Attention-based multi-scale network for image dehazing. In Proceedings of the IEEE/CVF international conference on computer vision. 7314–7323.Google ScholarGoogle ScholarCross RefCross Ref
  25. Kangfu Mei, Aiwen Jiang, Juncheng Li, and Mingwen Wang. 2019. Progressive feature fusion network for realistic image dehazing. In Computer Vision–ACCV 2018: 14th Asian Conference on Computer Vision, Perth, Australia, December 2–6, 2018, Revised Selected Papers, Part I 14. Springer, 203–215.Google ScholarGoogle Scholar
  26. Xu Qin, Zhilin Wang, Yuanchao Bai, Xiaodong Xie, and Huizhu Jia. 2020. FFA-Net: Feature fusion attention network for single image dehazing. In Proceedings of the AAAI conference on artificial intelligence, Vol. 34. 11908–11915.Google ScholarGoogle ScholarCross RefCross Ref
  27. Olaf Ronneberger, Philipp Fischer, and Thomas Brox. 2015. U-net: Convolutional networks for biomedical image segmentation. In Medical Image Computing and Computer-Assisted Intervention–MICCAI 2015: 18th International Conference, Munich, Germany, October 5-9, 2015, Proceedings, Part III 18. Springer, 234–241.Google ScholarGoogle Scholar
  28. Chitwan Saharia, Jonathan Ho, William Chan, Tim Salimans, David J Fleet, and Mohammad Norouzi. 2022. Image super-resolution via iterative refinement. IEEE Transactions on Pattern Analysis and Machine Intelligence 45, 4 (2022), 4713–4726.Google ScholarGoogle Scholar
  29. Daniel Scharstein, Heiko Hirschmüller, York Kitajima, Greg Krathwohl, Nera Nešić, Xi Wang, and Porter Westling. 2014. High-resolution stereo datasets with subpixel-accurate ground truth. In Pattern Recognition: 36th German Conference, GCPR 2014, Münster, Germany, September 2-5, 2014, Proceedings 36. Springer, 31–42.Google ScholarGoogle ScholarCross RefCross Ref
  30. Nathan Silberman, Derek Hoiem, Pushmeet Kohli, and Rob Fergus. 2012. Indoor segmentation and support inference from rgbd images. In Computer Vision–ECCV 2012: 12th European Conference on Computer Vision, Florence, Italy, October 7-13, 2012, Proceedings, Part V 12. Springer, 746–760.Google ScholarGoogle ScholarDigital LibraryDigital Library
  31. Yang Song and Stefano Ermon. 2019. Generative modeling by estimating gradients of the data distribution. Advances in neural information processing systems 32 (2019).Google ScholarGoogle Scholar
  32. Hu Yu, Jie Huang, Kaiwen Zheng, Man Zhou, and Feng Zhao. 2023. High-quality Image Dehazing with Diffusion Model. arXiv preprint arXiv:2308.11949 (2023).Google ScholarGoogle Scholar
  33. He Zhang, Vishwanath Sindagi, and Vishal M Patel. 2018. Multi-scale single image dehazing using perceptual pyramid deep network. In Proceedings of the IEEE conference on computer vision and pattern recognition workshops. 902–911.Google ScholarGoogle ScholarCross RefCross Ref
  34. Shengdong Zhang, Fazhi He, and Wenqi Ren. 2020. NLDN: Non-local dehazing network for dense haze removal. Neurocomputing 410 (2020), 363–373.Google ScholarGoogle ScholarCross RefCross Ref

Index Terms

  1. Image Dehazing based on Iterative-Refining Diffusion Model

      Recommendations

      Comments

      Login options

      Check if you have access through your login credentials or your institution to get full access on this article.

      Sign in
      • Published in

        cover image ACM Other conferences
        ICIGP '24: Proceedings of the 2024 7th International Conference on Image and Graphics Processing
        January 2024
        480 pages
        ISBN:9798400716720
        DOI:10.1145/3647649

        Copyright © 2024 ACM

        Permission to make digital or hard copies of all or part of this work for personal or classroom use is granted without fee provided that copies are not made or distributed for profit or commercial advantage and that copies bear this notice and the full citation on the first page. Copyrights for components of this work owned by others than the author(s) must be honored. Abstracting with credit is permitted. To copy otherwise, or republish, to post on servers or to redistribute to lists, requires prior specific permission and/or a fee. Request permissions from [email protected].

        Publisher

        Association for Computing Machinery

        New York, NY, United States

        Publication History

        • Published: 3 May 2024

        Permissions

        Request permissions about this article.

        Request Permissions

        Check for updates

        Qualifiers

        • research-article
        • Research
        • Refereed limited
      • Article Metrics

        • Downloads (Last 12 months)11
        • Downloads (Last 6 weeks)11

        Other Metrics

      PDF Format

      View or Download as a PDF file.

      PDF

      eReader

      View online with eReader.

      eReader

      HTML Format

      View this article in HTML Format .

      View HTML Format