{"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-2509-24868","title":"DRIFT-Net: A Spectral--Coupled Neural Operator for PDEs Learning","arxiv_id":"2509.24868","date":"2025-09-29","proceeding":"ICLR","authors":["Jiayi Li","Flora D. Salim"],"abstract":"Learning PDE dynamics with neural solvers can significantly improve wall-clock efficiency and accuracy compared with classical numerical solvers. In recent years, foundation models for PDEs have largely adopted multi-scale windowed self-attention, with the scOT backbone in Poseidon serving as a representative example. However, because of their locality, truly globally consistent spectral coupling can only be propagated gradually through deep stacking and window shifting. This weakens global coupling and leads to error accumulation and drift during closed-loop rollouts. To address this, we propose DRIFT-Net. It employs a dual-branch design comprising a spectral branch and an image branch. The spectral branch is responsible for capturing global, large-scale low-frequency information, whereas the image branch focuses on local details and nonstationary structures. Specifically, we first perform controlled, lightweight mixing within the low-frequency range. Then we fuse the spectral and image paths at each layer via bandwise weighting, which avoids the width inflation and training instability caused by naive concatenation. The fused result is transformed back into the spatial domain and added to the image branch, thereby preserving both global structure and high-frequency details across scales. Compared with strong attention-based baselines, DRIFT-Net achieves lower error and higher throughput with fewer parameters under identical training settings and budget. On Navier--Stokes benchmarks, the relative $L_{1}$ error is reduced by 7\\%--54\\%, the parameter count decreases by about 15\\%, and the throughput remains higher than scOT. Ablation studies and theoretical analyses further demonstrate the stability and effectiveness of this design. The code is available at https://github.com/cruiseresearchgroup/DRIFT-Net.","url_abs":"https://arxiv.org/abs/2509.24868","url_pdf":"https://arxiv.org/pdf/2509.24868","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=2509.24868","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2509.24868"}},"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/camlab-ethz/poseidon","reach":null},{"mentioned_in_github":null,"is_official":null,"provenance":"deterministic:regex_extraction","mentioned_in_paper":null,"url":"https://github.com/ashiq24/UNO","reach":{"status":"ok","spdx":"BSD-2-Clause"}}],"summary":{"ran":12,"unverified":5},"by_repo_kind":{"found_in_text":{"samples":17,"ran":12,"repositories":2}},"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":16,"samples":[{"code_sha256_prefix":"09044330f34201f0","entry":"ConditionalLayerNorm","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"09044330f34201f0"}},{"code_sha256_prefix":"e986a6be79277f9f","entry":"ConvNeXtBlock","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"e986a6be79277f9f"}},{"code_sha256_prefix":"4c877fd0479ab660","entry":"LayerNorm","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"4c877fd0479ab660"}},{"code_sha256_prefix":"8628adef554d5f6c","entry":"ResNetBlock","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"8628adef554d5f6c"}},{"code_sha256_prefix":"a2f781d6e7d8fab2","entry":"ScOTEmbeddings","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"a2f781d6e7d8fab2"}},{"code_sha256_prefix":"c1c2fd5b2b481b7e","entry":"ScOTLayer","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"c1c2fd5b2b481b7e"}},{"code_sha256_prefix":"2d2d908ebaad0342","entry":"ScOTOutput","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"2d2d908ebaad0342"}},{"code_sha256_prefix":"1899ef0b2271c9fa","entry":"ScOTPatchEmbeddings","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"NONE","inline_ok":false,"mcp_get_code":{"code_sha256":"1899ef0b2271c9fa"}},{"code_sha256_prefix":"4186cc8c008dfbb1","entry":"ScOTPatchMerging","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"4186cc8c008dfbb1"}},{"code_sha256_prefix":"c4493d951b873149","entry":"ScOTPatchRecovery","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"c4493d951b873149"}},{"code_sha256_prefix":"d808160c0cf58ace","entry":"ScOTPatchUnmerging","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"d808160c0cf58ace"}},{"code_sha256_prefix":"4c546cdfd4187fed","entry":"SpectralConv1d_Uno","repo":"ashiq24/UNO","repo_kind":"found_in_text","path":"integral_operators.py","file_url":"https://github.com/ashiq24/UNO/blob/HEAD/integral_operators.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"BSD-2-Clause","inline_ok":true,"mcp_get_code":{"code_sha256":"4c546cdfd4187fed"}},{"code_sha256_prefix":"31119ec3700bb45b","entry":"ScOT","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"31119ec3700bb45b"}},{"code_sha256_prefix":"aa4152c612d925a5","entry":"ScOTDecodeStage","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"aa4152c612d925a5"}},{"code_sha256_prefix":"9e771f2e43b2d451","entry":"ScOTDecoder","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"9e771f2e43b2d451"}},{"code_sha256_prefix":"668539a6b63eddb3","entry":"ScOTEncodeStage","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"668539a6b63eddb3"}},{"code_sha256_prefix":"ad710e7a5980e0bc","entry":"ScOTEncoder","repo":"camlab-ethz/poseidon","repo_kind":"found_in_text","path":"scOT/model.py","file_url":"https://github.com/camlab-ethz/poseidon/blob/HEAD/scOT/model.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":"ad710e7a5980e0bc"}}]},"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_api"},"syntology_extracted_results":{"kind":"leaderboard_placements","source":"Syntology's leaderboard-shaped extractor over the paper's own arXiv-HTML tables: a model pointed at a cell, the number was read from that cell and checked against the board's metric, dataset, split and scale, and an independent check accepted the entry; not reviewed by the paper's authors or the archive's editors","extractor_model":"global.anthropic.claude-sonnet-4-5-20250929-v1:0","verifier_model":null,"prompt_sha":"fa63d4bb9d755694","coverage":{"sentence":"Syntology has checked 6,264 of the 9,581 papers on this site that are newer than the archive; results from the others appear after they are checked.","papers_newer_than_archive":9581,"papers_checked":6264},"entries":[],"not_placed":{"boards":0,"rejected_by_independent_check":0,"refused_by_a_rule":0,"check_did_not_answer":0,"proposed_without_a_cell":0,"declined_by_site":0}}}