{"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-2605-07100","title":"TRACE: Transport Alignment Conformal Prediction via Diffusion and Flow Matching Models","arxiv_id":"2605.07100","date":"2026-05-08","proceeding":null,"authors":["Zhenhan Fang","Aixin Tan","Jian Huang"],"abstract":"Constructing valid and informative conformal prediction regions for multi-dimensional outputs remains a fundamental challenge. While conformal prediction provides finite-sample, distribution-free coverage guarantees, its practical performance critically depends on the choice of nonconformity score. Existing approaches often rely on restrictive geometric assumptions or require explicit likelihood evaluation and invertible transformations, limiting their applicability in complex generative settings. In this work, we introduce TRACE (TRansport Alignment Conformal Estimation), a conformal prediction framework that defines nonconformity through transport alignment in diffusion and flow matching models. Rather than evaluating likelihoods, we measure how well a candidate output aligns with the learned generative dynamics by averaging denoising or velocity-matching errors along stochastic transport trajectories. The resulting transport-based scores are scalar-valued and can be calibrated using split conformal prediction, yielding valid marginal coverage under exchangeability. We further analyze the statistical properties of the proposed scores and their sensitivity to computational budget. Experiments on synthetic and real datasets demonstrate valid coverage and show that the resulting regions adapt naturally to multimodal and non-convex conditional distributions.","url_abs":"https://arxiv.org/abs/2605.07100","url_pdf":"https://arxiv.org/pdf/2605.07100","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":null,"mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2605.07100"}},"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/ZFangUI/TRACE","reach":null}],"summary":{"ran":5,"ran_honours":1},"by_repo_kind":{"found_in_text":{"samples":6,"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":6,"samples":[{"code_sha256_prefix":"644219f338ab45b1","entry":"ConditionalDDPM","repo":"ZFangUI/TRACE","repo_kind":"found_in_text","path":"models/diffusion.py","file_url":"https://github.com/ZFangUI/TRACE/blob/HEAD/models/diffusion.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"644219f338ab45b1"}},{"code_sha256_prefix":"4bb970172e27a171","entry":"FiLMLayer","repo":"ZFangUI/TRACE","repo_kind":"found_in_text","path":"models/diffusion.py","file_url":"https://github.com/ZFangUI/TRACE/blob/HEAD/models/diffusion.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"4bb970172e27a171"}},{"code_sha256_prefix":"75790904b7ca33b4","entry":"FiLMResBlock","repo":"ZFangUI/TRACE","repo_kind":"found_in_text","path":"models/diffusion.py","file_url":"https://github.com/ZFangUI/TRACE/blob/HEAD/models/diffusion.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"75790904b7ca33b4"}},{"code_sha256_prefix":"59581d77f2cbaa71","entry":"NoisePredictor","repo":"ZFangUI/TRACE","repo_kind":"found_in_text","path":"models/diffusion.py","file_url":"https://github.com/ZFangUI/TRACE/blob/HEAD/models/diffusion.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"59581d77f2cbaa71"}},{"code_sha256_prefix":"5f28cf0da256ce9d","entry":"SinusoidalEmbedding","repo":"ZFangUI/TRACE","repo_kind":"found_in_text","path":"models/diffusion.py","file_url":"https://github.com/ZFangUI/TRACE/blob/HEAD/models/diffusion.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"5f28cf0da256ce9d"}},{"code_sha256_prefix":"aacf60494b8fedb7","entry":"cosine_schedule","repo":"ZFangUI/TRACE","repo_kind":"found_in_text","path":"models/diffusion.py","file_url":"https://github.com/ZFangUI/TRACE/blob/HEAD/models/diffusion.py","link_basis":"first_harvest_node","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":"well_formed","behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"aacf60494b8fedb7"}}]},"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}