{"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-2608-01298","title":"UDT: Reconciling U-Nets and Diffusion Transformers with Data-Adaptive Token Reduction","arxiv_id":"2608.01298","date":"2026-08-02","proceeding":null,"authors":["Junno Yun","Yaşar Utku Alçalar","Mehmet Akçakaya"],"abstract":"Diffusion Transformers (DiTs) have emerged as a core architecture in generative modeling due to their scalability and adaptability to multimodal tasks. DiTs comprise isotropic transformer blocks, and learn representations progressively across depth, where the denoising objective drives later layers to focus on fine-detail reconstruction. This results in degraded representation quality and an imbalanced encoder-decoder behavior. Prior approaches such as representation alignment (REPA) mitigate this by encouraging stronger early representations via training regularization. Alternatively, U-Net-style DiT architectures introduce explicit multi-scale encoder-decoder structures for improved convergence. But they build on standard U-Net wisdom via learnable operators for spatial downsampling, which are not well-suited to transformer architectures, introducing inefficiencies and compatibility issues with components such as cross-attention and representation regularization. In this work, we propose UDT, a U-Net diffusion transformer that combines the representation power of DiTs with the encoding-decoding benefits of U-Nets, through data-adaptive token merging for downsampling and upsampling, while preserving the DiT token dimension. Our baseline UDT architecture outperforms existing U-Net DiTs and achieves performance comparable to REPA across all model sizes. Furthermore, using architectural optimization and REPA, UDT outperforms SiT's 7.9 FID at 1400 epochs (w/o CFG) within 40 epochs (~ 40x faster convergence) for XL model size on 256x256 ImageNet. Finally, it achieves strong image generation performance with CFG, reaching FID of 1.38 (320 epochs) with SD-VAE and 1.35 (500 epochs) with VA-VAE, providing a new backbone for DiTs with strong empirical benefits.","url_abs":"https://arxiv.org/abs/2608.01298","url_pdf":"https://arxiv.org/pdf/2608.01298","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=2608.01298","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2608.01298"}},"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/JN-Yun/UDT","reach":null}],"summary":{"ran":7,"ran_fixture":4,"ran_draft_wrong":1,"ran_honours":1,"ran_violates":1,"unverified":6},"by_repo_kind":{"found_in_text":{"samples":19,"ran":13,"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":20,"samples":[{"code_sha256_prefix":"7a122209a9457bd0","entry":"FinalLayer","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"7a122209a9457bd0"}},{"code_sha256_prefix":"813b038557b60edd","entry":"LabelEmbedder","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"813b038557b60edd"}},{"code_sha256_prefix":"9a95d88a2519390e","entry":"RMSNorm","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"9a95d88a2519390e"}},{"code_sha256_prefix":"5997b89997ac7e41","entry":"SwiGLUFFN","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"5997b89997ac7e41"}},{"code_sha256_prefix":"132dc6ca8683503e","entry":"TimestepEmbedder","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"132dc6ca8683503e"}},{"code_sha256_prefix":"29cbfcbffd4bd38f","entry":"ToMeAttention","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"29cbfcbffd4bd38f"}},{"code_sha256_prefix":"64e4c00ba026c8e5","entry":"bipartite_soft_matching","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"64e4c00ba026c8e5"}},{"code_sha256_prefix":"d5580b3fb66f103c","entry":"bipartite_soft_matching_random2d","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"d5580b3fb66f103c"}},{"code_sha256_prefix":"58a82fed4cc8a4d8","entry":"build_token_schedule","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"58a82fed4cc8a4d8"}},{"code_sha256_prefix":"7b75bf9ed4e14ba7","entry":"get_1d_sincos_pos_embed_from_grid","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"7b75bf9ed4e14ba7"}},{"code_sha256_prefix":"0c991bde9f1d8f95","entry":"get_2d_sincos_pos_embed","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.py","link_basis":"first_harvest_node","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"0c991bde9f1d8f95"}},{"code_sha256_prefix":"d49c2d2942e53c88","entry":"log_udt_config","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"d49c2d2942e53c88"}},{"code_sha256_prefix":"f55137c53a3a4557","entry":"precompute_freqs_cis_2d","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.py","link_basis":"first_harvest_node","language":"python","status":"ran_violates","verification_level":1,"contract_check":"VIOLATES","metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"f55137c53a3a4557"}},{"code_sha256_prefix":"20ff6d11dd593018","entry":"reshape_for_broadcast","repo":null,"repo_kind":null,"path":null,"file_url":null,"link_basis":"identical_code_first_harvested_elsewhere","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":null,"inline_ok":false,"mcp_get_code":{"code_sha256":"20ff6d11dd593018"}},{"code_sha256_prefix":"f3e68b726a668eb8","entry":"UDT","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"f3e68b726a668eb8"}},{"code_sha256_prefix":"79eb22210c6b30d5","entry":"UDTBlock","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"79eb22210c6b30d5"}},{"code_sha256_prefix":"dfafa843246491e7","entry":"apply_rotary_emb","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"dfafa843246491e7"}},{"code_sha256_prefix":"6dc4ec7d0b39acf8","entry":"compiled_rms_norm_math","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"6dc4ec7d0b39acf8"}},{"code_sha256_prefix":"4509f0ab71bc54ca","entry":"get_2d_sincos_pos_embed_from_grid","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"4509f0ab71bc54ca"}},{"code_sha256_prefix":"f303c3aef5017fe1","entry":"modulate","repo":"JN-Yun/UDT","repo_kind":"found_in_text","path":"models/UDT.py","file_url":"https://github.com/JN-Yun/UDT/blob/HEAD/models/UDT.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":"f303c3aef5017fe1"}}]},"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}