{"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-01625","title":"PRIME: Protein Representation via Physics-Informed Multiscale Equivariant Hierarchies","arxiv_id":"2605.01625","date":"2026-05-02","proceeding":null,"authors":["Viet Thanh Duy Nguyen","John K. Johnstone","Truong-Son Hy"],"abstract":"Proteins are inherently multiscale physical systems whose functional properties emerge from coordinated structural organization across multiple spatial resolutions, ranging from atomic interactions to global fold topology. However, existing protein representation learning methods typically operate at a single structural level or treat different sources of structural information as parallel modalities, without explicitly modeling their hierarchical relationships. We introduce PRIME (Protein Representation via Physics-Informed Multiscale Equivariant Hierarchies), a unified framework that models proteins as a nested family of five physically grounded structural graphs spanning surface, atomic, residue, secondary-structure, and protein levels. Adjacent levels are connected through deterministic, physics-informed assignment operators, enabling bidirectional information exchange via bottom-up aggregation and top-down contextual refinement. Experiments on standard protein representation learning benchmarks demonstrate strong and competitive performance across diverse tasks, with particularly notable gains on the Fold Classification benchmark, where PRIME outperforms the strongest geometric GNN baseline by margins of 13.80 and 18.30 points on the harder Superfamily and Fold splits, and achieves a state-of-the-art accuracy of 84.10\\% on Reaction Class prediction, surpassing all baseline methods, including ESM. Ablation studies confirm that each structural level contributes complementary and non-redundant information, and adaptive cross-attention analysis reveals that PRIME autonomously identifies the most task-relevant structural resolutions at prediction time. Our source code is publicly available at https://github.com/HySonLab/PRIME","url_abs":"https://arxiv.org/abs/2605.01625","url_pdf":"https://arxiv.org/pdf/2605.01625","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.01625"}},"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/HySonLab/PRIME","reach":null}],"summary":{"ran":7,"ran_fixture":1,"ran_draft_wrong":1,"unverified":1},"by_repo_kind":{"found_in_text":{"samples":10,"ran":9,"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":"890ccf93ef063709","entry":"collate_fn","repo":"HySonLab/PRIME","repo_kind":"found_in_text","path":"plm_baseline.py","file_url":"https://github.com/HySonLab/PRIME/blob/HEAD/plm_baseline.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"890ccf93ef063709"}},{"code_sha256_prefix":"df3b20eb658d3f76","entry":"collect_attention_weights","repo":"HySonLab/PRIME","repo_kind":"found_in_text","path":"visualize_attention.py","file_url":"https://github.com/HySonLab/PRIME/blob/HEAD/visualize_attention.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"df3b20eb658d3f76"}},{"code_sha256_prefix":"fddb0ea79ee68552","entry":"evaluate","repo":"HySonLab/PRIME","repo_kind":"found_in_text","path":"plm_baseline.py","file_url":"https://github.com/HySonLab/PRIME/blob/HEAD/plm_baseline.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"fddb0ea79ee68552"}},{"code_sha256_prefix":"60c78926c622999d","entry":"get_activation","repo":"HySonLab/PRIME","repo_kind":"found_in_text","path":"models/classification_head.py","file_url":"https://github.com/HySonLab/PRIME/blob/HEAD/models/classification_head.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"60c78926c622999d"}},{"code_sha256_prefix":"c2f9c401f5379aac","entry":"get_edges","repo":"HySonLab/PRIME","repo_kind":"found_in_text","path":"models/egnn.py","file_url":"https://github.com/HySonLab/PRIME/blob/HEAD/models/egnn.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"c2f9c401f5379aac"}},{"code_sha256_prefix":"99ce6594cd4b4a1e","entry":"train_epoch","repo":"HySonLab/PRIME","repo_kind":"found_in_text","path":"plm_baseline.py","file_url":"https://github.com/HySonLab/PRIME/blob/HEAD/plm_baseline.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"99ce6594cd4b4a1e"}},{"code_sha256_prefix":"72d1a5a69354f146","entry":"train_one_epoch","repo":"HySonLab/PRIME","repo_kind":"found_in_text","path":"train_prime.py","file_url":"https://github.com/HySonLab/PRIME/blob/HEAD/train_prime.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"72d1a5a69354f146"}},{"code_sha256_prefix":"bc66f19be57720e7","entry":"unsorted_segment_mean","repo":"HySonLab/PRIME","repo_kind":"found_in_text","path":"models/egnn.py","file_url":"https://github.com/HySonLab/PRIME/blob/HEAD/models/egnn.py","link_basis":"plan_row","language":"python","status":"ran_fixture","verification_level":1,"contract_check":"RAISES","metamorphic_tier":"invariant","behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"bc66f19be57720e7"}},{"code_sha256_prefix":"5fb62a78eb65d3d7","entry":"unsorted_segment_sum","repo":"HySonLab/PRIME","repo_kind":"found_in_text","path":"models/egnn.py","file_url":"https://github.com/HySonLab/PRIME/blob/HEAD/models/egnn.py","link_basis":"plan_row","language":"python","status":"ran_draft_wrong","verification_level":1,"contract_check":"MISDECLARED","metamorphic_tier":"invariant","behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"5fb62a78eb65d3d7"}},{"code_sha256_prefix":"e74e991ea6639f67","entry":"evaluate","repo":"HySonLab/PRIME","repo_kind":"found_in_text","path":"train_prime.py","file_url":"https://github.com/HySonLab/PRIME/blob/HEAD/train_prime.py","link_basis":"first_harvest_node","language":"python","status":"unverified","verification_level":0,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"e74e991ea6639f67"}}]},"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}