论文

用基于图的工具改进小语言模型的分子性质预测

Improving Molecular Property Prediction in Small Language Models Using Graph-based Tools

智能体系统Agent 工具调用

摘要

小语言模型(SLM)在从SMILES字符串零样本预测分子性质上展现前景,但常苦于结构盲视——序列表示欠指定关键的图拓扑线索。我们提出模块化的上下文增广提示框架:在推理时支持智能体工具使用——经训练的GNN专家模型提供带置信度的预测提示,另一个GNN抽取实例专属的解释子图(如子图SMILES与配套解释段落)。我们在MUTAG与Tox21上、以从仅SMILES到全工具的五种提示配置评估三个常用SLM。跨两个数据集:用图派生上下文增广提示带来可观的准确率增益——常超25%相对改进、在Tox21上最高74%。我们经基于必要性的边删除干预进一步验证抽取基序的功能相关性。尽管增益可观,与专用GNN模型的差距仍存——凸显文本条件化推理对分子结构的价值与限度。

用基于图的工具改进小语言模型的分子性质预测:论文配图
图 1:建议的代理循环。给定查询 SMILES,SLM 调用一个或多个可用工具与预先训练的 GNN 专家进行交互,生成相关上下文并增强原始提示,为最终推理提供帮助。