{"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-20581","title":"TriForces: Augmenting Atomistic GNNs for Transferable Representations","arxiv_id":"2605.20581","date":"2026-05-20","proceeding":"ICML","authors":["Ali Ramlaoui","Alexandre Duval","Hannah Bull","Victor Schmidt","Hugues Talbot","Fragkiskos D. Malliaros","Joseph Musielewicz"],"abstract":"Machine learning interatomic potentials (MLIPs) achieve excellent accuracy when trained on large Density Functional Theory (DFT) data. To be useful in practice, they must often be adapted to target chemistries using small and expensive task-specific datasets. However, MLIPs transfer inconsistently across domains, with representations that often loose accessible composition and structure information. To address this, we present TriForces, a model-agnostic three-stream framework that separates composition and structure information, combined with self-supervised learning to preserve transferable representations. TriForces improves performance on MatBench and QM9 over baselines without needing DFT labels and enables efficient similar structure retrieval through its learned latent space. On OMat24, in limited-data training regime, TriForces reduces energy MAE by 57% at 20K samples only and improves force MAE across sample sizes. We release pretrained TriForces variants across multiple MLIP architectures with code at https://github.com/Ramlaoui/triforces.","url_abs":"https://arxiv.org/abs/2605.20581","url_pdf":"https://arxiv.org/pdf/2605.20581","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=2605.20581","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2605.20581"}},"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/Ramlaoui/triforces","reach":null}],"summary":{"ran":8,"ran_draft_wrong":2,"unverified":5},"by_repo_kind":{"found_in_text":{"samples":15,"ran":10,"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":0,"samples":[{"code_sha256_prefix":"7c1a33604035ac54","entry":"BackboneOutputs","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"7c1a33604035ac54"}},{"code_sha256_prefix":"f88d7a1b8d523dff","entry":"BesselBasis","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"f88d7a1b8d523dff"}},{"code_sha256_prefix":"2bc29bee5fadb80b","entry":"CompositionStream","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"2bc29bee5fadb80b"}},{"code_sha256_prefix":"ef96af0d7a883dbd","entry":"CosineCutoff","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"ef96af0d7a883dbd"}},{"code_sha256_prefix":"24b091ed15e0e6b8","entry":"CountWeightedTransformerBlock","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"24b091ed15e0e6b8"}},{"code_sha256_prefix":"f39edfbf5fd194f1","entry":"GaussianRBF","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":"deterministic","behaviour_fingerprint":true,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"f39edfbf5fd194f1"}},{"code_sha256_prefix":"cb7733b37abe7544","entry":"LatticeEncoder","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"cb7733b37abe7544"}},{"code_sha256_prefix":"7b45ecc2fae462a1","entry":"MessagePassingLayer","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"7b45ecc2fae462a1"}},{"code_sha256_prefix":"fbb8aeb512070df2","entry":"_get_first_attr","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"OUTPUT_MISDECLARED","metamorphic_tier":"deterministic","behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"fbb8aeb512070df2"}},{"code_sha256_prefix":"18e5a8fc17261a63","entry":"_mean_pool","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":"invariant","behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"18e5a8fc17261a63"}},{"code_sha256_prefix":"67bafc602666472d","entry":"LearnablePowerSpectrumDescriptor","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"67bafc602666472d"}},{"code_sha256_prefix":"56446dd166d5d7e4","entry":"SphericalHarmonics","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"56446dd166d5d7e4"}},{"code_sha256_prefix":"8bfdb75986162b31","entry":"StructuralStreamPowerSpectrum","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"8bfdb75986162b31"}},{"code_sha256_prefix":"4e2d0996583b6740","entry":"TriForcesModel","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"4e2d0996583b6740"}},{"code_sha256_prefix":"3468b1cd5e917fed","entry":"_spherical_harmonics_l0_to_l4","repo":"Ramlaoui/triforces","repo_kind":"found_in_text","path":"src/triforces/models/triforces.py","file_url":"https://github.com/Ramlaoui/triforces/blob/HEAD/src/triforces/models/triforces.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"Apache-2.0","inline_ok":true,"mcp_get_code":{"code_sha256":"3468b1cd5e917fed"}}]},"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}