| 专利号 | 2025106356450 | 申请日 | 2025-05-16 | 专利名称 | 基于几何感知注意模型的高效分子对接方法 |
| 授权日 | 2026-02-13 | 专利权人 | 中国石油大学(华东) | 发明人 | 王珣;宋弢;栗建新 |
| 主分类号 | G16B15/30 | 关键词 | 应用领域 | ||
| 摘要 | 本发明属于分子对接技术领域，特别涉及基于几何感知注意模型的高效分子对接方法。动态内存归约策略利用迭代求值和分块技术实现注意力矩阵的高效并行计算，并借助共享内存完成中间结果在不同线程束之间的通信；分层融合策略为三角形注意力设计两个高性能融合内核，使得每个线程块生成的数据全部保存在寄存器和共享内存中；张量平铺数据移动通过跨维度的数据搬运方式来重新组织线程与数据的映射关系，实现全局合并访存模式；多头并行注意力合并利用多头注意力合并策略优化QKV数据布局，减少中间结果额外的数据转置和拼接操作。本发明解决了现有的三角形注意力算子推理时的内存爆炸问题、三角形更新算子中的不规则访存问题以及冗余访存问题。 | ||||
| 创新点 | |||||
| 技术分类 | 标 签 | 战兴产业 | 生物产业 生物医学工程产业 | ||
| 运营方式 | 合作方式 | ||||
| 联系人 | 联系电话 | 电子邮箱 | |||
| 详细说明 | |||||
