arXiv:2608.01092cs.AI2026-08

用三维残基几何构建注意力位置编码,提升抗体表位预测精度

Inter-Residue Geometry Attention for Antibody-Specific Epitope Prediction

论文配图:Inter-Residue Geometry Attention for Antibody-Specific Epitope Prediction
图 1 · 摘自论文原文
  • 引入局部坐标系下的3D旋转位置编码,用空间几何替代序列顺序
  • 在AsEP数据集上达到最新最佳MCC,尤其在表位分组划分上表现优异
  • 对突变排序有效,适合关注结构互补性的抗体-抗原研究者

抗体特异性表位预测旨在识别抗原中被特定抗体识别的残基,依赖于抗体互补决定区(CDRs)与抗原表面之间的三维互补性。现有方法通常使用预训练语言模型嵌入,并通过图、表面或点云编码器注入结构信息,但注意力中的位置机制仍主要基于一维序列顺序。对于蛋白质而言,残基间的“位置偏移”不仅包括序列距离,更包含折叠后的三维空间位移。这引发一个关键问题:折叠后的残基几何能否作为注意力的位置机制?本文提出局部帧3D旋转位置编码(LF3DRoPE),将残基间位移表示为骨架定义的局部坐标系中的向量,并直接注入旋转注意力模块。该设计在保持连续方向几何的同时,确保对全局SE(3)变换的不变性。在AsEP基准测试中,LF3DRoPE在比率和表位分组两种划分下均取得当前最优的MCC值。消融实验与刚性变换测试表明,局部三维几何提供了超越序列顺序注意力的信息,同时保持对任意全局坐标系的不变性。突变排序结果进一步显示,LF3DRoPE能有效捕捉抗原-抗体间的结构兼容性。

原文摘要 · Abstract (English)

Antibody-specific epitope prediction aims to identify which antigen residues are recognized by a given antibody, a task that depends on the three-dimensional complementarity between antibody CDRs and the antigen surface. Existing methods usually leverage PLM embeddings and inject structure through additional graph, surface, or point-cloud encoders, where the positional mechanism inside attention remains largely tied to one-dimensional sequence order. For proteins, the analogue of a token offset is not only sequence separation, but also the three-dimensional displacement between residues after folding. This raises a question, can folded residue geometry serve as the positional mechanism of attention itself? We propose Local-Frame 3D Rotary Position Encoding (LF3DRoPE), which expresses inter-residue displacements in backbone-defined local frames and injects them directly into rotary attention. This design preserves continuous directional geometry while ensuring invariance to global $\mathrm{SE}(3)$ transformations. On the AsEP benchmark, LF3DRoPE achieves state-of-the-art $\mathrm{MCC}$ on both ratio and epitope-group splits. Ablations and rigid transformation tests show that local three-dimensional geometry provides information beyond sequence-order attention while preserving invariance to arbitrary global coordinate systems. Mutation ranking results further indicate that LF3DRoPE captures antigen-specific structural compatibility.

抗体预测三维注意力表位识别几何编码

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