{"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-2601-15500","title":"Low-Dimensional Adaptation of Rectified Flow: A Diffusion and Stochastic Localization Perspective","arxiv_id":"2601.15500","date":"2026-01-21","proceeding":null,"authors":["Saptarshi Roy","Alessandro Rinaldo","Purnamrita Sarkar"],"abstract":"In recent years, Rectified flow (RF) has gained considerable popularity largely due to its generation efficiency and state-of-the-art performance. In this paper, we investigate the degree to which RF automatically adapts to the intrinsic low dimensionality of the support of the target distribution to accelerate sampling. We show that, using a carefully designed choice of the time-discretization scheme and with sufficiently accurate drift estimates, the RF sampler enjoys an iteration complexity of order $O(k/\\varepsilon)$ (up to log factors), where $\\varepsilon$ is the precision in total variation distance and $k$ is the intrinsic dimension of the target distribution. In addition, we show that the denoising diffusion probabilistic model (DDPM) procedure is equivalent to a stochastic version of RF by establishing a novel connection between these processes and stochastic localization. Building on this connection, we further design a stochastic RF sampler that also adapts to the low-dimensionality of the target distribution under milder requirements on the accuracy of the drift estimates, and also with a specific time schedule. We illustrate with simulations on the synthetic data and text-to-image data experiments the improved performance of the proposed samplers implementing the newly designed time-discretization schedules.","url_abs":"https://arxiv.org/abs/2601.15500","url_pdf":"https://arxiv.org/pdf/2601.15500","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=2601.15500","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2601.15500"}},"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/roysaptaumich/Low-dim-RF","reach":null}],"summary":{"unverified":13},"by_repo_kind":{"found_in_text":{"samples":13,"ran":0,"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":13,"samples":[{"code_sha256_prefix":"9cfc794e19b4001c","entry":"MynewSDE_timegrid_generator","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"RFsampler_modules.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/RFsampler_modules.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":"9cfc794e19b4001c"}},{"code_sha256_prefix":"64816a79d3c177f3","entry":"calculate_discriminative_tv_qda","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"metric.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/metric.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":"64816a79d3c177f3"}},{"code_sha256_prefix":"2a98fc90af4ea45e","entry":"calculate_mmd","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"metric.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/metric.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":"2a98fc90af4ea45e"}},{"code_sha256_prefix":"acee3106924910c2","entry":"compute_norm_squared_full_derivative","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"real_data/straightness.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/real_data/straightness.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":"acee3106924910c2"}},{"code_sha256_prefix":"3379901e9c2a25f8","entry":"create_save_dir","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"real_data/utils.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/real_data/utils.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":"3379901e9c2a25f8"}},{"code_sha256_prefix":"e4b938638f0e2605","entry":"generate_data","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"data_generator.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/data_generator.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":"e4b938638f0e2605"}},{"code_sha256_prefix":"26df9086b5385607","entry":"geom_timegrid_generator","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"RFsampler_modules.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/RFsampler_modules.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":"26df9086b5385607"}},{"code_sha256_prefix":"c5957497bd84f1cb","entry":"get_dataloader","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"real_data/data.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/real_data/data.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":"c5957497bd84f1cb"}},{"code_sha256_prefix":"fc787a2f125d21ce","entry":"kl_divergence_gaussians","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"metric.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/metric.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":"fc787a2f125d21ce"}},{"code_sha256_prefix":"2590f16dba1b392a","entry":"log_p_hat_t","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"RFmodules.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/RFmodules.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":"2590f16dba1b392a"}},{"code_sha256_prefix":"643e229256ef78bc","entry":"postprocess","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"real_data/utils.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/real_data/utils.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":"643e229256ef78bc"}},{"code_sha256_prefix":"3e1b6d43e2f78b19","entry":"score_blurred","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"RFmodules.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/RFmodules.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":"3e1b6d43e2f78b19"}},{"code_sha256_prefix":"22aeded8818aef1c","entry":"score_kernel_exact","repo":"roysaptaumich/Low-dim-RF","repo_kind":"found_in_text","path":"RFmodules.py","file_url":"https://github.com/roysaptaumich/Low-dim-RF/blob/HEAD/RFmodules.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":"22aeded8818aef1c"}}]},"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}