{"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/fast-sampling-of-diffusion-models-via","title":"Fast Sampling of Diffusion Models via Operator Learning","arxiv_id":"2211.13449","date":"2022-11-24","proceeding":null,"authors":["Hongkai Zheng","Weili Nie","Arash Vahdat","Kamyar Azizzadenesheli","Anima Anandkumar"],"abstract":"Diffusion models have found widespread adoption in various areas. However, their sampling process is slow because it requires hundreds to thousands of network evaluations to emulate a continuous process defined by differential equations. In this work, we use neural operators, an efficient method to solve the probability flow differential equations, to accelerate the sampling process of diffusion models. Compared to other fast sampling methods that have a sequential nature, we are the first to propose a parallel decoding method that generates images with only one model forward pass. We propose diffusion model sampling with neural operator (DSNO) that maps the initial condition, i.e., Gaussian distribution, to the continuous-time solution trajectory of the reverse diffusion process. To model the temporal correlations along the trajectory, we introduce temporal convolution layers that are parameterized in the Fourier space into the given diffusion model backbone. We show our method achieves state-of-the-art FID of 3.78 for CIFAR-10 and 7.83 for ImageNet-64 in the one-model-evaluation setting.","url_abs":"https://arxiv.org/abs/2211.13449v3","url_pdf":"https://arxiv.org/pdf/2211.13449v3.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":"fast-sampling-of-diffusion-models-via","repo_url":"https://github.com/devzhk/dsno-pytorch","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok","spdx":"MIT"}}],"tasks":[{"task_slug":"operator-learning","task_name":"Operator learning"}],"methods":[{"method_slug":"convolution","method_name":"Convolution"},{"method_slug":"diffusion","method_name":"Diffusion"}],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2211.13449","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2211.13449"}},"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/devzhk/dsno-pytorch","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"ran_draft_wrong":1,"ran":3,"unverified":15},"by_repo_kind":{"official":{"samples":19,"ran":4,"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":0,"samples":[{"code_sha256_prefix":"dbb756dcd778f52f","entry":"data_sampler","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"utils/data_helper.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/utils/data_helper.py","link_basis":"harvester_set","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"dbb756dcd778f52f"}},{"code_sha256_prefix":"04548c6c9fff2b3f","entry":"default_init","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"models/layers.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/models/layers.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"04548c6c9fff2b3f"}},{"code_sha256_prefix":"b663f5d919fd45e0","entry":"get_timestep_embedding","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"models/layers.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/models/layers.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"b663f5d919fd45e0"}},{"code_sha256_prefix":"f34882fce4b22d20","entry":"variance_scaling","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"models/layers.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/models/layers.py","link_basis":"harvester_set","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"f34882fce4b22d20"}},{"code_sha256_prefix":"4038cee0912727be","entry":"build_state_map","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"models/utils.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/models/utils.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"4038cee0912727be"}},{"code_sha256_prefix":"a57d49ef17dbed05","entry":"get_idx","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"generate_imagenet.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/generate_imagenet.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"a57d49ef17dbed05"}},{"code_sha256_prefix":"0e7775bc8f637675","entry":"get_key_map","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"models/utils.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/models/utils.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"0e7775bc8f637675"}},{"code_sha256_prefix":"3470bcd60748b7b6","entry":"get_logsnr_input","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"models/tddpmm.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/models/tddpmm.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"3470bcd60748b7b6"}},{"code_sha256_prefix":"347413fbd8917c6f","entry":"get_logsnr_schedule","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"models/tddpmm.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/models/tddpmm.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"347413fbd8917c6f"}},{"code_sha256_prefix":"71c9af712b11fc4b","entry":"get_random_label","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"generate_imagenet.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/generate_imagenet.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"71c9af712b11fc4b"}},{"code_sha256_prefix":"8db300b9f1afb2d3","entry":"get_weight","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"models/up_or_down_sampling.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/models/up_or_down_sampling.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"8db300b9f1afb2d3"}},{"code_sha256_prefix":"9375ef0a6c4c1787","entry":"naive_downsample_2d","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"models/up_or_down_sampling.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/models/up_or_down_sampling.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"9375ef0a6c4c1787"}},{"code_sha256_prefix":"a7d9bb56776bab6d","entry":"naive_upsample_2d","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"models/up_or_down_sampling.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/models/up_or_down_sampling.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"a7d9bb56776bab6d"}},{"code_sha256_prefix":"f28952b5b1c172f0","entry":"percepLoss","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"utils/loss.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/utils/loss.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"f28952b5b1c172f0"}},{"code_sha256_prefix":"48a5788c680cb78a","entry":"save2dir","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"generate_imagenet.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/generate_imagenet.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"48a5788c680cb78a"}},{"code_sha256_prefix":"770ce3af9e8428eb","entry":"scaled_dot_product","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"models/layersmt.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/models/layersmt.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"770ce3af9e8428eb"}},{"code_sha256_prefix":"c393b5de24f2b4b2","entry":"split_list","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"utils/dataset.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/utils/dataset.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"c393b5de24f2b4b2"}},{"code_sha256_prefix":"2a97a27262e3dd56","entry":"weightedL1","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"utils/loss.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/utils/loss.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"2a97a27262e3dd56"}},{"code_sha256_prefix":"36b2614c0d8fb14d","entry":"weightedL2","repo":"devzhk/dsno-pytorch","repo_kind":"official","path":"utils/loss.py","file_url":"https://github.com/devzhk/dsno-pytorch/blob/HEAD/utils/loss.py","link_basis":"harvester_set","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"36b2614c0d8fb14d"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}