{"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-2603-07276","title":"Variational Flow Maps: Make Some Noise for One-Step Conditional Generation","arxiv_id":"2603.07276","date":"2026-03-07","proceeding":"ICML","authors":["Abbas Mammadov","So Takao","Bohan Chen","Ricardo Baptista","Morteza Mardani","Yee Whye Teh","Julius Berner"],"abstract":"Flow maps enable high-quality image generation in a single forward pass. However, unlike iterative diffusion models, their lack of an explicit sampling trajectory impedes incorporating external constraints for conditional generation and solving inverse problems. We put forth Variational Flow Maps, a framework for conditional sampling that shifts the perspective of conditioning from \"guiding a sampling path\", to that of \"learning the proper initial noise\". Specifically, given an observation, we seek to learn a noise adapter model that outputs a noise distribution, so that after mapping to the data space via flow map, the samples respect the observation and data prior. To this end, we develop a principled variational objective that jointly trains the noise adapter and the flow map, improving noise-data alignment, such that sampling from complex data posterior is achieved with a simple adapter. Experiments on various inverse problems show that VFMs produce well-calibrated conditional samples in a single (or few) steps. For ImageNet, VFM attains competitive fidelity while accelerating the sampling by orders of magnitude compared to alternative iterative diffusion/flow models. Code is available at https://github.com/abbasmammadov/VFM","url_abs":"https://arxiv.org/abs/2603.07276","url_pdf":"https://arxiv.org/pdf/2603.07276","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":{"syntology_url":"https://syntology.ai/paper/2603.07276","atlas_url":"https://app.syntology.ai/?focus=2603.07276","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2603.07276"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-25T09:33:49+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/abbasmammadov/VFM","reach":null}],"summary":{"ran_draft_wrong":4,"ran_fixture":2,"unverified":1},"by_repo_kind":{"found_in_text":{"samples":7,"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":0,"samples":[{"code_sha256_prefix":"5d2f29a549efeacb","entry":"adaptive","repo":"abbasmammadov/VFM","repo_kind":"found_in_text","path":"loss.py","file_url":"https://github.com/abbasmammadov/VFM/blob/HEAD/loss.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"5d2f29a549efeacb"}},{"code_sha256_prefix":"f373f40c9eee6bd2","entry":"compute_guidance","repo":"abbasmammadov/VFM","repo_kind":"found_in_text","path":"loss.py","file_url":"https://github.com/abbasmammadov/VFM/blob/HEAD/loss.py","link_basis":"first_harvest_node","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"f373f40c9eee6bd2"}},{"code_sha256_prefix":"ce1652c9ff63a153","entry":"interpolant","repo":"abbasmammadov/VFM","repo_kind":"found_in_text","path":"loss.py","file_url":"https://github.com/abbasmammadov/VFM/blob/HEAD/loss.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"ce1652c9ff63a153"}},{"code_sha256_prefix":"76ad6b50bfae7ab3","entry":"ln_sampler","repo":"abbasmammadov/VFM","repo_kind":"found_in_text","path":"loss.py","file_url":"https://github.com/abbasmammadov/VFM/blob/HEAD/loss.py","link_basis":"first_harvest_node","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"76ad6b50bfae7ab3"}},{"code_sha256_prefix":"113e3a84f65fdd5c","entry":"mean_flat","repo":"abbasmammadov/VFM","repo_kind":"found_in_text","path":"loss.py","file_url":"https://github.com/abbasmammadov/VFM/blob/HEAD/loss.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":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"113e3a84f65fdd5c"}},{"code_sha256_prefix":"b766939802ffeabc","entry":"mse_loss","repo":"abbasmammadov/VFM","repo_kind":"found_in_text","path":"loss.py","file_url":"https://github.com/abbasmammadov/VFM/blob/HEAD/loss.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":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"b766939802ffeabc"}},{"code_sha256_prefix":"18b1a579c2d4c244","entry":"AdapterLoss","repo":"abbasmammadov/VFM","repo_kind":"found_in_text","path":"loss.py","file_url":"https://github.com/abbasmammadov/VFM/blob/HEAD/loss.py","link_basis":"first_harvest_node","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":"18b1a579c2d4c244"}}]},"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}