{"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/capacity-control-is-an-effective-memorization","title":"Capacity Control is an Effective Memorization Mitigation Mechanism in Text-Conditional Diffusion Models","arxiv_id":"2410.22149","date":"2024-10-29","proceeding":null,"authors":["Raman Dutt","Pedro Sanchez","Ondrej Bohdal","Sotirios A. Tsaftaris","Timothy Hospedales"],"abstract":"In this work, we present compelling evidence that controlling model capacity during fine-tuning can effectively mitigate memorization in diffusion models. Specifically, we demonstrate that adopting Parameter-Efficient Fine-Tuning (PEFT) within the pre-train fine-tune paradigm significantly reduces memorization compared to traditional full fine-tuning approaches. Our experiments utilize the MIMIC dataset, which comprises image-text pairs of chest X-rays and their corresponding reports. The results, evaluated through a range of memorization and generation quality metrics, indicate that PEFT not only diminishes memorization but also enhances downstream generation quality. Additionally, PEFT methods can be seamlessly combined with existing memorization mitigation techniques for further improvement. The code for our experiments is available at: https://github.com/Raman1121/Diffusion_Memorization_HPO","url_abs":"https://arxiv.org/abs/2410.22149v1","url_pdf":"https://arxiv.org/pdf/2410.22149v1.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":"capacity-control-is-an-effective-memorization","repo_url":"https://github.com/raman1121/diffusion_memorization_hpo","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":0,"framework":"pytorch","reach":{"status":"ok"}}],"tasks":[{"task_slug":"memorization","task_name":"Memorization"},{"task_slug":"parameter-efficient-fine-tuning","task_name":"parameter-efficient fine-tuning"}],"methods":[{"method_slug":"diffusion","method_name":"Diffusion"}],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2410.22149","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2410.22149"}},"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/raman1121/diffusion_memorization_hpo","reach":{"status":"ok"}}],"summary":{"ran_draft_wrong":1,"unverified":9},"by_repo_kind":{"official":{"samples":10,"ran":1,"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":10,"samples":[{"code_sha256_prefix":"4675779aec92d6c0","entry":"contrastive_loss","repo":"raman1121/diffusion_memorization_hpo","repo_kind":"official","path":"difffit/modeling_clip.py","file_url":"https://github.com/raman1121/diffusion_memorization_hpo/blob/HEAD/difffit/modeling_clip.py","link_basis":"harvester_set","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"4675779aec92d6c0"}},{"code_sha256_prefix":"f2ca34eca05d40cd","entry":"clip_loss","repo":"raman1121/diffusion_memorization_hpo","repo_kind":"official","path":"difffit/modeling_clip.py","file_url":"https://github.com/raman1121/diffusion_memorization_hpo/blob/HEAD/difffit/modeling_clip.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"f2ca34eca05d40cd"}},{"code_sha256_prefix":"80b0901668ac3b4b","entry":"construct_graph","repo":"raman1121/diffusion_memorization_hpo","repo_kind":"official","path":"extraction_attack.py","file_url":"https://github.com/raman1121/diffusion_memorization_hpo/blob/HEAD/extraction_attack.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"80b0901668ac3b4b"}},{"code_sha256_prefix":"02589606db2535b0","entry":"find_cliques","repo":"raman1121/diffusion_memorization_hpo","repo_kind":"official","path":"extraction_attack.py","file_url":"https://github.com/raman1121/diffusion_memorization_hpo/blob/HEAD/extraction_attack.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"02589606db2535b0"}},{"code_sha256_prefix":"64daf97f4f932ef7","entry":"insert_rand_word","repo":"raman1121/diffusion_memorization_hpo","repo_kind":"official","path":"imagenette_dataset.py","file_url":"https://github.com/raman1121/diffusion_memorization_hpo/blob/HEAD/imagenette_dataset.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"64daf97f4f932ef7"}},{"code_sha256_prefix":"35c1d6e65f13ee7c","entry":"prompt_augmentation","repo":"raman1121/diffusion_memorization_hpo","repo_kind":"official","path":"imagenette_dataset.py","file_url":"https://github.com/raman1121/diffusion_memorization_hpo/blob/HEAD/imagenette_dataset.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"35c1d6e65f13ee7c"}},{"code_sha256_prefix":"b4aa69078465b08e","entry":"return_blockwise_param_sum","repo":"raman1121/diffusion_memorization_hpo","repo_kind":"official","path":"adaptors.py","file_url":"https://github.com/raman1121/diffusion_memorization_hpo/blob/HEAD/adaptors.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"b4aa69078465b08e"}},{"code_sha256_prefix":"e5ef16b9b63da748","entry":"return_norm_param_sum","repo":"raman1121/diffusion_memorization_hpo","repo_kind":"official","path":"adaptors.py","file_url":"https://github.com/raman1121/diffusion_memorization_hpo/blob/HEAD/adaptors.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"e5ef16b9b63da748"}},{"code_sha256_prefix":"5b81be558f23d729","entry":"return_param_sum","repo":"raman1121/diffusion_memorization_hpo","repo_kind":"official","path":"adaptors.py","file_url":"https://github.com/raman1121/diffusion_memorization_hpo/blob/HEAD/adaptors.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"5b81be558f23d729"}},{"code_sha256_prefix":"c13432a556516ef9","entry":"similarity","repo":"raman1121/diffusion_memorization_hpo","repo_kind":"official","path":"extraction_attack.py","file_url":"https://github.com/raman1121/diffusion_memorization_hpo/blob/HEAD/extraction_attack.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"c13432a556516ef9"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}