{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/enhanced-control-for-diffusion-bridge-in","title":"Enhanced Control for Diffusion Bridge in Image Restoration","arxiv_id":"2408.16303","date":"2024-08-29","proceeding":null,"authors":["Conghan Yue","Zhengwei Peng","Junlong Ma","Dongyu Zhang"],"abstract":"Image restoration refers to the process of restoring a damaged low-quality image back to its corresponding high-quality image. Typically, we use convolutional neural networks to directly learn the mapping from low-quality images to high-quality images achieving image restoration. Recently, a special type of diffusion bridge model has achieved more advanced results in image restoration. It can transform the direct mapping from low-quality to high-quality images into a diffusion process, restoring low-quality images through a reverse process. However, the current diffusion bridge restoration models do not emphasize the idea of conditional control, which may affect performance. This paper introduces the ECDB model enhancing the control of the diffusion bridge with low-quality images as conditions. Moreover, in response to the characteristic of diffusion models having low denoising level at larger values of \\(\\bm t \\), we also propose a Conditional Fusion Schedule, which more effectively handles the conditional feature information of various modules. Experimental results prove that the ECDB model has achieved state-of-the-art results in many image restoration tasks, including deraining, inpainting and super-resolution. Code is avaliable at https://github.com/Hammour-steak/ECDB.","url_abs":"https://arxiv.org/abs/2408.16303v1","url_pdf":"https://arxiv.org/pdf/2408.16303v1.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"enhanced-control-for-diffusion-bridge-in","repo_url":"https://github.com/Hammour-steak/ECDB","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok"}}],"tasks":[{"task_slug":"denoising","task_name":"Denoising"},{"task_slug":"image-restoration","task_name":"Image Restoration"},{"task_slug":"rain-removal","task_name":"Rain Removal"},{"task_slug":"super-resolution","task_name":"Super-Resolution"}],"methods":[{"method_slug":"diffusion","method_name":"Diffusion"},{"method_slug":"pixel-prediction","method_name":"Inpainting"}],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2408.16303","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2408.16303"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-24T18:15:14+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/Hammour-steak/ECDB","reach":{"status":"ok"}}],"summary":{"ran":4,"ran_violates":2},"by_repo_kind":{"official":{"samples":6,"ran":6,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":6,"samples":[{"code_sha256_prefix":"5b4b368d735cbee9","entry":"NonLinearity","repo":"Hammour-steak/ECDB","repo_kind":"official","path":"codes/models/modules/module_util.py","file_url":"https://github.com/Hammour-steak/ECDB/blob/HEAD/codes/models/modules/module_util.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"5b4b368d735cbee9"}},{"code_sha256_prefix":"85bf2da666cd3d07","entry":"add_noise","repo":"Hammour-steak/ECDB","repo_kind":"official","path":"codes/utils/deg_utils.py","file_url":"https://github.com/Hammour-steak/ECDB/blob/HEAD/codes/utils/deg_utils.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"85bf2da666cd3d07"}},{"code_sha256_prefix":"d1ef6b8cb9a28a53","entry":"default","repo":"Hammour-steak/ECDB","repo_kind":"official","path":"codes/models/modules/module_util.py","file_url":"https://github.com/Hammour-steak/ECDB/blob/HEAD/codes/models/modules/module_util.py","link_basis":"harvester_set","language":"python","status":"ran_violates","verification_level":1,"contract_check":"VIOLATES","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"d1ef6b8cb9a28a53"}},{"code_sha256_prefix":"608e364a9d2376a3","entry":"exists","repo":"Hammour-steak/ECDB","repo_kind":"official","path":"codes/models/modules/module_util.py","file_url":"https://github.com/Hammour-steak/ECDB/blob/HEAD/codes/models/modules/module_util.py","link_basis":"harvester_set","language":"python","status":"ran_violates","verification_level":1,"contract_check":"VIOLATES","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"608e364a9d2376a3"}},{"code_sha256_prefix":"0a0f56e2ac5b29cf","entry":"mask_to","repo":"Hammour-steak/ECDB","repo_kind":"official","path":"codes/utils/deg_utils.py","file_url":"https://github.com/Hammour-steak/ECDB/blob/HEAD/codes/utils/deg_utils.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"0a0f56e2ac5b29cf"}},{"code_sha256_prefix":"b3ccfc519d222aca","entry":"upscale","repo":"Hammour-steak/ECDB","repo_kind":"official","path":"codes/utils/deg_utils.py","file_url":"https://github.com/Hammour-steak/ECDB/blob/HEAD/codes/utils/deg_utils.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"b3ccfc519d222aca"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}