arXiv:2608.04257cs.LG2026-08

用几何信息提升分子模型微调效率,小参数量实现精准血脑屏障穿透预测。

Geometry-Informed Parameter-Efficient Fine-Tuning of Pre-trained Molecular GNNs for Blood-Brain Barrier Permeability Prediction

论文配图:Geometry-Informed Parameter-Efficient Fine-Tuning of Pre-trained Molecular GNNs for Blood-Brain Barrier Permeability Prediction
图 1 · 摘自论文原文
  • 构建多距离截断的几何图与线图,捕捉原子空间关系和边间交互
  • 仅更新10.1%参数,性能媲美全量微调,随机与骨架分割下均表现稳健
  • 适合药物研发中数据少、结构敏感场景,尤其关注血脑屏障穿透性研究

血脑屏障渗透性(BBBP)预测是中枢神经系统药物发现中的关键筛选任务,需评估候选分子能否穿过或避免穿过血脑屏障。然而,受限于数据量少、类别不平衡及对分子结构敏感等问题,该任务仍具挑战性。深度学习推动图神经网络(GNN)在分子表征学习中取得进展,预训练分子GNN为下游任务提供可迁移知识。但全量微调参数效率低且易过拟合,现有参数高效微调(PEFT)方法主要调整节点特征或二维共价图,难以捕捉三维几何结构与二阶交互。为此,我们提出BBBP-GeoPEFT,一种面向预训练分子GNN的几何感知参数高效微调框架。该方法从分子构象构建多截断距离图及其对应的线图,以捕捉原子空间关系与边间交互。轻量级辅助几何图编码器生成截断特异性表示,通过节点级截断注意力与门控残差连接融入每个预训练层。此设计在保留预训练知识的同时,以极小参数预算引入与渗透性相关的几何信息。在整理后的BBBP数据集上,实验表明BBBP-GeoPEFT在随机与骨架划分下均达到与全量微调相当或更优的性能,同时仅更新10.1%模型参数。

原文摘要 · Abstract (English)

Blood-brain barrier permeability (BBBP) prediction is a critical screening task in central nervous system drug discovery, where candidate molecules must be assessed for whether they can cross, or should be prevented from crossing, the blood-brain barrier. However, this task remains challenging because of limited, class-imbalanced datasets and sensitivity to molecular structure. Recent advances in deep learning have established graph neural networks (GNNs) as a powerful approach for molecular representation learning, while pre-trained molecular GNNs provide transferable knowledge for downstream tasks. However, full fine-tuning is often parameter-inefficient and prone to overfitting, whereas existing parameter-efficient fine-tuning (PEFT) methods mainly adapt node features or the two-dimensional covalent graph, limiting their ability to capture three-dimensional geometry and second-order interactions. To address these limitations, we propose BBBP-GeoPEFT, a geometry-informed PEFT framework for pre-trained molecular GNNs. BBBP-GeoPEFT constructs distance-based graphs at multiple cutoffs and their corresponding line graphs from molecular conformers to capture spatial atom and second-order edge interactions. Lightweight auxiliary geometric graph encoders generate cutoff-specific representations, which are incorporated into each pre-trained layer through node-wise cutoff attention and gated residual connections. This design preserves pre-trained knowledge while incorporating permeability-relevant geometric information with a small trainable-parameter budget. Experiments on a curated BBBP dataset show that BBBP-GeoPEFT achieves competitive performance compared with full fine-tuning and representative PEFT baselines. Under both random and scaffold splitting, BBBP-GeoPEFT achieves competitive or improved ROC-AUC and accuracy in most experiments while updating only 10.1% of the model parameters.

分子图神经网络参数高效微调血脑屏障几何建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。