{"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-08267","title":"Inverting Data Transformations via Diffusion Sampling","arxiv_id":"2602.08267","date":"2026-02-09","proceeding":"ICML","authors":["Jinwoo Kim","Sékou-Oumar Kaba","Jiyun Park","Seunghoon Hong","Siamak Ravanbakhsh"],"abstract":"We study the problem of transformation inversion on general Lie groups: a datum is transformed by an unknown group element, and the goal is to recover an inverse transformation that maps it back to the original data distribution. Such unknown transformations arise widely in machine learning and scientific modeling, where they can significantly distort observations. We take a probabilistic view and model the posterior over transformations as a Boltzmann distribution defined by an energy function on the data space. To sample from this posterior, we introduce a diffusion process on Lie groups that keeps all updates on-manifold and only requires computations in the associated Lie algebra. Our method, Transformation-Inverting Energy Diffusion (TIED), relies on a new trivialized target-score identity that enables efficient score-based sampling of the transformation posterior. As a key application, we focus on test-time equivariance, where the objective is to improve the robustness of pretrained neural networks to input transformations. Experiments on image homographies and PDE symmetries demonstrate that TIED can restore transformed inputs to the training distribution at test time, showing improved performance over strong canonicalization and sampling baselines. Code is available at https://github.com/jw9730/tied.","url_abs":"https://arxiv.org/abs/2602.08267","url_pdf":"https://arxiv.org/pdf/2602.08267","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.08267","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2602.08267"}},"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/jw9730/tied","reach":null}],"summary":{"ran":1,"ran_fixture":1,"unverified":5},"by_repo_kind":{"found_in_text":{"samples":7,"ran":2,"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":7,"samples":[{"code_sha256_prefix":"da858e3f840d94fe","entry":"ConnectedMatrixLieGroup","repo":"jw9730/tied","repo_kind":"found_in_text","path":"src/group_sampler.py","file_url":"https://github.com/jw9730/tied/blob/HEAD/src/group_sampler.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":"da858e3f840d94fe"}},{"code_sha256_prefix":"726fd7fe1b607f1d","entry":"clip","repo":"jw9730/tied","repo_kind":"found_in_text","path":"src/group_sampler.py","file_url":"https://github.com/jw9730/tied/blob/HEAD/src/group_sampler.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":"726fd7fe1b607f1d"}},{"code_sha256_prefix":"f6d56fe6bf0c8746","entry":"Energy","repo":"jw9730/tied","repo_kind":"found_in_text","path":"src/group_sampler.py","file_url":"https://github.com/jw9730/tied/blob/HEAD/src/group_sampler.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":"f6d56fe6bf0c8746"}},{"code_sha256_prefix":"9b6f8f14d5a87899","entry":"EnergyDiffusionSampler","repo":"jw9730/tied","repo_kind":"found_in_text","path":"src/group_sampler.py","file_url":"https://github.com/jw9730/tied/blob/HEAD/src/group_sampler.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":"9b6f8f14d5a87899"}},{"code_sha256_prefix":"70114c5e741e3513","entry":"Group","repo":"jw9730/tied","repo_kind":"found_in_text","path":"src/group_sampler.py","file_url":"https://github.com/jw9730/tied/blob/HEAD/src/group_sampler.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":"70114c5e741e3513"}},{"code_sha256_prefix":"89ed2c7afdf35b21","entry":"GroupEnergySampler","repo":"jw9730/tied","repo_kind":"found_in_text","path":"src/group_sampler.py","file_url":"https://github.com/jw9730/tied/blob/HEAD/src/group_sampler.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":"89ed2c7afdf35b21"}},{"code_sha256_prefix":"81358894ad41c8c0","entry":"GroupSampler","repo":"jw9730/tied","repo_kind":"found_in_text","path":"src/group_sampler.py","file_url":"https://github.com/jw9730/tied/blob/HEAD/src/group_sampler.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":"81358894ad41c8c0"}}]},"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}