{"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/learning-to-group-auxiliary-datasets-for","title":"Learning to Group Auxiliary Datasets for Molecule","arxiv_id":"2307.04052","date":"2023-07-08","proceeding":"NeurIPS 2023 11","authors":[],"abstract":"The limited availability of annotations in small molecule datasets presents a challenge to machine learning models. To address this, one common strategy is to collaborate with additional auxiliary datasets. However, having more data does not always guarantee improvements. Negative transfer can occur when the knowledge in the target dataset differs or contradicts that of the auxiliary molecule datasets. In light of this, identifying the auxiliary molecule datasets that can benefit the target dataset when jointly trained remains a critical and unresolved problem. Through an empirical analysis, we observe that combining graph structure similarity and task similarity can serve as a more reliable indicator for identifying high-affinity auxiliary datasets. Motivated by this insight, we propose MolGroup, which separates the dataset affinity into task and structure affinity to predict the potential benefits of each auxiliary molecule dataset. MolGroup achieves this by utilizing a routing mechanism optimized through a bi-level optimization framework. Empowered by the meta gradient, the routing mechanism is optimized toward maximizing the target dataset's performance and quantifies the affinity as the gating score. As a result, MolGroup is capable of predicting the optimal combination of auxiliary datasets for each target dataset. Our extensive experiments demonstrate the efficiency and effectiveness of MolGroup, showing an average improvement of 4.41%/3.47% for GIN/Graphormer trained with the group of molecule datasets selected by MolGroup on 11 target molecule datasets.","url_abs":"https://arxiv.org/abs/2307.04052v2","url_pdf":"https://arxiv.org/pdf/2307.04052v2.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"learning-to-group-auxiliary-datasets-for","repo_url":"https://github.com/graph-and-geometric-learning/molgroup","is_official":0,"mentioned_in_paper":0,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok"}}],"tasks":[],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":"https://syntology.ai/paper/2307.04052","atlas_url":"https://app.syntology.ai/?focus=2307.04052","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2307.04052"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-25T09:33:49+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":[{"provenance":"deterministic:regex_extraction","url":"https://github.com/Graph-and-Geometric-Learning/MolGroup","reach":{"status":"ok"}},{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/graph-and-geometric-learning/molgroup","reach":{"status":"ok"}}],"summary":{"ran":13,"unverified":6},"by_repo_kind":{"listed":{"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":19,"samples":[{"code_sha256_prefix":"d92f0f2cb39a2fcf","entry":"AtomEncoder","repo":"graph-and-geometric-learning/molgroup","repo_kind":"listed","path":"molgroup/models/metagin.py","file_url":"https://github.com/graph-and-geometric-learning/molgroup/blob/HEAD/molgroup/models/metagin.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":"d92f0f2cb39a2fcf"}},{"code_sha256_prefix":"729abc3fd03bcd87","entry":"BNLayer","repo":"graph-and-geometric-learning/molgroup","repo_kind":"listed","path":"molgroup/models/metagin.py","file_url":"https://github.com/graph-and-geometric-learning/molgroup/blob/HEAD/molgroup/models/metagin.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":"729abc3fd03bcd87"}},{"code_sha256_prefix":"4d48ee0d85006b1f","entry":"BondEncoder","repo":"graph-and-geometric-learning/molgroup","repo_kind":"listed","path":"molgroup/models/metagin.py","file_url":"https://github.com/graph-and-geometric-learning/molgroup/blob/HEAD/molgroup/models/metagin.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":"4d48ee0d85006b1f"}},{"code_sha256_prefix":"5daa83c68e9f2d4c","entry":"EmbeddingLayer","repo":"graph-and-geometric-learning/molgroup","repo_kind":"listed","path":"molgroup/models/metagin.py","file_url":"https://github.com/graph-and-geometric-learning/molgroup/blob/HEAD/molgroup/models/metagin.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":"5daa83c68e9f2d4c"}},{"code_sha256_prefix":"afe12bde1e9bc1d4","entry":"GaterWrapper","repo":"graph-and-geometric-learning/molgroup","repo_kind":"listed","path":"molgroup/models/metagin.py","file_url":"https://github.com/graph-and-geometric-learning/molgroup/blob/HEAD/molgroup/models/metagin.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":"afe12bde1e9bc1d4"}},{"code_sha256_prefix":"ac36594669402b46","entry":"LinearLayer","repo":"graph-and-geometric-learning/molgroup","repo_kind":"listed","path":"molgroup/models/metagin.py","file_url":"https://github.com/graph-and-geometric-learning/molgroup/blob/HEAD/molgroup/models/metagin.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":"ac36594669402b46"}},{"code_sha256_prefix":"f3e7b76f061433e0","entry":"MixGate","repo":"graph-and-geometric-learning/molgroup","repo_kind":"listed","path":"molgroup/models/metagin.py","file_url":"https://github.com/graph-and-geometric-learning/molgroup/blob/HEAD/molgroup/models/metagin.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":"f3e7b76f061433e0"}},{"code_sha256_prefix":"68d1138ae99b280c","entry":"TwoLayerMLP","repo":"graph-and-geometric-learning/molgroup","repo_kind":"listed","path":"molgroup/models/metagin.py","file_url":"https://github.com/graph-and-geometric-learning/molgroup/blob/HEAD/molgroup/models/metagin.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":"68d1138ae99b280c"}},{"code_sha256_prefix":"250aa4c9f2b943bc","entry":"add_args","repo":"Graph-and-Geometric-Learning/MolGroup","repo_kind":"listed","path":"molgroup/utils/utils.py","file_url":"https://github.com/Graph-and-Geometric-Learning/MolGroup/blob/HEAD/molgroup/utils/utils.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":"250aa4c9f2b943bc"}},{"code_sha256_prefix":"4f4035bee18bf78c","entry":"convert_to_pyg","repo":"Graph-and-Geometric-Learning/MolGroup","repo_kind":"listed","path":"molgroup/utils/dataset.py","file_url":"https://github.com/Graph-and-Geometric-Learning/MolGroup/blob/HEAD/molgroup/utils/dataset.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":"4f4035bee18bf78c"}},{"code_sha256_prefix":"29264267bc4ef9cf","entry":"convert_to_single_emb","repo":"Graph-and-Geometric-Learning/MolGroup","repo_kind":"listed","path":"molgroup/utils/graphormer_collator.py","file_url":"https://github.com/Graph-and-Geometric-Learning/MolGroup/blob/HEAD/molgroup/utils/graphormer_collator.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":"29264267bc4ef9cf"}},{"code_sha256_prefix":"ed9c8a17fdb3d49c","entry":"pad_1d_unsqueeze","repo":"Graph-and-Geometric-Learning/MolGroup","repo_kind":"listed","path":"molgroup/utils/graphormer_collator.py","file_url":"https://github.com/Graph-and-Geometric-Learning/MolGroup/blob/HEAD/molgroup/utils/graphormer_collator.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":"ed9c8a17fdb3d49c"}},{"code_sha256_prefix":"276b930464659a52","entry":"train","repo":"Graph-and-Geometric-Learning/MolGroup","repo_kind":"listed","path":"molgroup/app/train_molgroup.py","file_url":"https://github.com/Graph-and-Geometric-Learning/MolGroup/blob/HEAD/molgroup/app/train_molgroup.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":"276b930464659a52"}},{"code_sha256_prefix":"8a66c57bdf38fe4a","entry":"MetaAtomEncoderGate","repo":"graph-and-geometric-learning/molgroup","repo_kind":"listed","path":"molgroup/models/metagin.py","file_url":"https://github.com/graph-and-geometric-learning/molgroup/blob/HEAD/molgroup/models/metagin.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":"8a66c57bdf38fe4a"}},{"code_sha256_prefix":"6e6443e381ca7f66","entry":"MetaGIN","repo":"graph-and-geometric-learning/molgroup","repo_kind":"listed","path":"molgroup/models/metagin.py","file_url":"https://github.com/graph-and-geometric-learning/molgroup/blob/HEAD/molgroup/models/metagin.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":"6e6443e381ca7f66"}},{"code_sha256_prefix":"c15046f7a29418e3","entry":"MetaGINConv","repo":"graph-and-geometric-learning/molgroup","repo_kind":"listed","path":"molgroup/models/metagin.py","file_url":"https://github.com/graph-and-geometric-learning/molgroup/blob/HEAD/molgroup/models/metagin.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":"c15046f7a29418e3"}},{"code_sha256_prefix":"b729251646a1d5bb","entry":"ParamWrapper","repo":"graph-and-geometric-learning/molgroup","repo_kind":"listed","path":"molgroup/models/metagin.py","file_url":"https://github.com/graph-and-geometric-learning/molgroup/blob/HEAD/molgroup/models/metagin.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":"b729251646a1d5bb"}},{"code_sha256_prefix":"91e80dbe2d80955c","entry":"meta_train","repo":"Graph-and-Geometric-Learning/MolGroup","repo_kind":"listed","path":"molgroup/app/train_molgroup.py","file_url":"https://github.com/Graph-and-Geometric-Learning/MolGroup/blob/HEAD/molgroup/app/train_molgroup.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":"91e80dbe2d80955c"}},{"code_sha256_prefix":"1393df1c173a4662","entry":"test","repo":"Graph-and-Geometric-Learning/MolGroup","repo_kind":"listed","path":"molgroup/app/train_molgroup.py","file_url":"https://github.com/Graph-and-Geometric-Learning/MolGroup/blob/HEAD/molgroup/app/train_molgroup.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":"1393df1c173a4662"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}