arXiv:cs.LG(机器学习,全量分类)· Jinmo Lee, Dooho Lee, Minho Jeong, Jaemin Yoo·· 5 小时前AI 评分36
用表格基础模型做结构偏移下的分子性质预测:MolPAIR 框架
Molecular Property Prediction under Structural Shift with Tabular Foundation Models
AI 导读
研究者提出 MolPAIR(Molecular Pair-Augmented In-context Refinement),在无需任务特定参数更新的前提下,结合分子级与分子对上下文来提升结构偏移下的分子性质预测。
正文
Abstract:Predicting molecular properties for compounds that differ structurally from labeled training molecules is important for drug discovery and materials design. Tabular foundation models (TFMs) offer a promising approach through in-context learning, but their performance under structural shifts and the value of molecular comparisons in this setting remain underexplored. We study structural generalization in molecular property prediction and introduce MolPAIR (Molecular Pair-Augmented In-context Refinement), a framework that combines molecule-level and molecular-pair contexts without task-specific parameter updates. A global tabular foundation model (TFM) first predicts a query's property from labeled molecular examples. A second frozen TFM predicts differences in prediction errors between the query and labeled reference molecules, using these comparisons to refine the initial prediction. Across 58 MoleculeACE and Polaris tasks, CheMeleon representations combined with TabPFN-3 already outperform each evaluated baseline on a majority of tasks. MOLPAIR further improves this predictor on 46 of 58 tasks, with gains across four molecular representations and three TFM backbones. These results show that explicit molecular comparisons can strengthen tabular in-context learning for structural generalization while keeping the molecular encoder and pretrained model weights fixed. The code and datasets are available at this https URL.
| Subjects: | Machine Learning (cs.LG) |
| Cite as: | arXiv:2609.38744 [cs.LG] |
| (or arXiv:2609.38744v1 [cs.LG] for this version) | |
| https://doi.org/10.48550/arXiv.2609.38744 arXiv-issued DOI via DataCite (pending registration) |
Submission history
From: Jinmo Lee [view email]
[v1]
Wed, 30 Sep 2026 01:28:28 UTC (256 KB)
来源:arXiv:cs.LG(机器学习,全量分类) · arxiv.org