{"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/arxiv-2602-04139","title":"Generative Neural Operators through Diffusion Last Layer","arxiv_id":"2602.04139","date":"2026-02-04","proceeding":"ICML","authors":["Sungwon Park","Anthony Zhou","Hongjoong Kim","Amir Barati Farimani"],"abstract":"Neural operators provide a powerful framework for learning discretization invariant mappings between function spaces, but standard deterministic models do not capture predictive uncertainty. We introduce diffusion last layer (DLL), a modular probabilistic output head for neural operator backbones. DLL represents target fields through an input dependent low rank expansion inspired by the Karhunen-Loéve expansion and learns a conditional diffusion model over the corresponding coefficient space. This design enables efficient distributional modeling while preserving the structural advantages of operator learning. On stochastic PDE benchmarks with random forcing, DLL achieves strong distributional fidelity and performs competitively with pixel space and conventional latent diffusion baselines. In deterministic long horizon rollout tasks, DLL improves rollout stability over the underlying backbone and provides useful estimates of predictive uncertainty under compounding autoregressive errors. These results suggest that diffusion modeling in learned coefficient spaces offers a practical route to uncertainty aware neural operators.","url_abs":"https://arxiv.org/abs/2602.04139","url_pdf":"https://arxiv.org/pdf/2602.04139","source":{"archive":null,"snapshot":"2025-07-28","note":"not in the Papers with Code archive (frozen at the snapshot)","row_kind":"graph","title_abstract_authors_date":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)"},"code_links":[],"tasks":[],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"atlas_url":"https://app.syntology.ai/?focus=2602.04139","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2602.04139"}},"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":[{"mentioned_in_github":null,"is_official":null,"provenance":"deterministic:regex_extraction","mentioned_in_paper":null,"url":"https://github.com/sungwpark/dll-no","reach":null}],"summary":{"ran":1,"ran_draft_wrong":2,"ran_fixture":5,"ran_honours":1,"unverified":2},"by_repo_kind":{"found_in_text":{"samples":11,"ran":9,"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":11,"samples":[{"code_sha256_prefix":"81327c451840a293","entry":"StripMetadataModule","repo":"sungwpark/dll-no","repo_kind":"found_in_text","path":"src/train/diffusion/dll.py","file_url":"https://github.com/sungwpark/dll-no/blob/HEAD/src/train/diffusion/dll.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":"81327c451840a293"}},{"code_sha256_prefix":"f6ec137af9dd35eb","entry":"_format_horizon_tag","repo":"sungwpark/dll-no","repo_kind":"found_in_text","path":"src/train/diffusion/dll.py","file_url":"https://github.com/sungwpark/dll-no/blob/HEAD/src/train/diffusion/dll.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"f6ec137af9dd35eb"}},{"code_sha256_prefix":"1fae8bd072358580","entry":"_per_example_rmse_nrmse_vrmse","repo":"sungwpark/dll-no","repo_kind":"found_in_text","path":"src/train/diffusion/dll.py","file_url":"https://github.com/sungwpark/dll-no/blob/HEAD/src/train/diffusion/dll.py","link_basis":"first_harvest_node","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"1fae8bd072358580"}},{"code_sha256_prefix":"aeb8b7813745712f","entry":"_split_logspace_three_bands","repo":"sungwpark/dll-no","repo_kind":"found_in_text","path":"src/train/diffusion/dll.py","file_url":"https://github.com/sungwpark/dll-no/blob/HEAD/src/train/diffusion/dll.py","link_basis":"first_harvest_node","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"aeb8b7813745712f"}},{"code_sha256_prefix":"1abd73a3078f193a","entry":"_steps_from_percentiles","repo":"sungwpark/dll-no","repo_kind":"found_in_text","path":"src/train/diffusion/dll.py","file_url":"https://github.com/sungwpark/dll-no/blob/HEAD/src/train/diffusion/dll.py","link_basis":"first_harvest_node","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"1abd73a3078f193a"}},{"code_sha256_prefix":"33a044d58239b854","entry":"compute_ensemble_metrics","repo":"sungwpark/dll-no","repo_kind":"found_in_text","path":"src/train/diffusion/dll.py","file_url":"https://github.com/sungwpark/dll-no/blob/HEAD/src/train/diffusion/dll.py","link_basis":"first_harvest_node","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"33a044d58239b854"}},{"code_sha256_prefix":"57f671198cbeb2d2","entry":"compute_nrmse","repo":"sungwpark/dll-no","repo_kind":"found_in_text","path":"src/train/diffusion/dll.py","file_url":"https://github.com/sungwpark/dll-no/blob/HEAD/src/train/diffusion/dll.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"57f671198cbeb2d2"}},{"code_sha256_prefix":"f51a77ecb9efae34","entry":"compute_psrmse_three_bands","repo":"sungwpark/dll-no","repo_kind":"found_in_text","path":"src/train/diffusion/dll.py","file_url":"https://github.com/sungwpark/dll-no/blob/HEAD/src/train/diffusion/dll.py","link_basis":"first_harvest_node","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"f51a77ecb9efae34"}},{"code_sha256_prefix":"2463439748b30d84","entry":"compute_stochastic_mean_std_metrics","repo":"sungwpark/dll-no","repo_kind":"found_in_text","path":"src/train/diffusion/dll.py","file_url":"https://github.com/sungwpark/dll-no/blob/HEAD/src/train/diffusion/dll.py","link_basis":"first_harvest_node","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"2463439748b30d84"}},{"code_sha256_prefix":"db5250aa7149d8b1","entry":"DLL","repo":"sungwpark/dll-no","repo_kind":"found_in_text","path":"src/train/diffusion/dll.py","file_url":"https://github.com/sungwpark/dll-no/blob/HEAD/src/train/diffusion/dll.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":"db5250aa7149d8b1"}},{"code_sha256_prefix":"b4284290743f1894","entry":"DiffusionModel","repo":"sungwpark/dll-no","repo_kind":"found_in_text","path":"src/train/diffusion/dll.py","file_url":"https://github.com/sungwpark/dll-no/blob/HEAD/src/train/diffusion/dll.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":"b4284290743f1894"}}]},"arxiv_metadata":{"licence":"arXiv metadata, CC0 1.0 (https://info.arxiv.org/help/license)","fields":["title","abstract","authors","date"],"primary_category":"cs.LG","source":"arxiv_2026.jsonl"},"syntology_extracted_results":null}