Skip to main content
CenXiv.org
此网站处于试运行阶段,支持我们!
我们衷心感谢所有贡献者的支持。
贡献
赞助
cenxiv logo > cs > arXiv:2507.02085

帮助 | 高级搜索

计算机科学 > 机器学习

arXiv:2507.02085 (cs)
[提交于 2025年7月2日 ]

标题: GeoAda:使用等变适配器高效微调几何扩散模型

标题: GeoAda: Efficiently Finetune Geometric Diffusion Models with Equivariant Adapters

Authors:Wanjia Zhao, Jiaqi Han, Siyi Gu, Mingjian Jiang, James Zou, Stefano Ermon
摘要: 几何扩散模型在分子动力学和结构生成中表现出显著的成功。然而,对于具有不同几何控制的下游任务,高效微调它们仍然研究不足。在本工作中,我们提出了一种SE(3)-等变适配器框架(GeoAda),该框架能够在不修改原始模型架构的情况下,实现可控生成任务的灵活且参数高效的微调。GeoAda引入了一种结构化的适配器设计:控制信号首先通过耦合算子进行编码,然后由可训练的选定预训练模型层副本进行处理,最后通过解耦算子投影回原空间,并随后进行等变零初始化卷积。通过仅微调这些轻量级适配器模块,GeoAda保持了模型的几何一致性,同时减轻了过拟合和灾难性遗忘。我们理论证明了所提出的适配器保持SE(3)-等变性,确保预训练扩散模型的几何归纳偏置在适应过程中保持完整。我们在多种几何控制类型上展示了GeoAda的广泛应用性,包括框架控制、全局控制、子图控制以及广泛的领域应用,如粒子动力学、分子动力学、人体运动预测和分子生成。实验结果表明,GeoAda在保持原始任务准确性的同时实现了最先进的微调性能,而其他基线方法由于过拟合和灾难性遗忘导致性能显著下降。
摘要: Geometric diffusion models have shown remarkable success in molecular dynamics and structure generation. However, efficiently fine-tuning them for downstream tasks with varying geometric controls remains underexplored. In this work, we propose an SE(3)-equivariant adapter framework ( GeoAda) that enables flexible and parameter-efficient fine-tuning for controlled generative tasks without modifying the original model architecture. GeoAda introduces a structured adapter design: control signals are first encoded through coupling operators, then processed by a trainable copy of selected pretrained model layers, and finally projected back via decoupling operators followed by an equivariant zero-initialized convolution. By fine-tuning only these lightweight adapter modules, GeoAda preserves the model's geometric consistency while mitigating overfitting and catastrophic forgetting. We theoretically prove that the proposed adapters maintain SE(3)-equivariance, ensuring that the geometric inductive biases of the pretrained diffusion model remain intact during adaptation. We demonstrate the wide applicability of GeoAda across diverse geometric control types, including frame control, global control, subgraph control, and a broad range of application domains such as particle dynamics, molecular dynamics, human motion prediction, and molecule generation. Empirical results show that GeoAda achieves state-of-the-art fine-tuning performance while preserving original task accuracy, whereas other baselines experience significant performance degradation due to overfitting and catastrophic forgetting.
主题: 机器学习 (cs.LG) ; 人工智能 (cs.AI)
引用方式: arXiv:2507.02085 [cs.LG]
  (或者 arXiv:2507.02085v1 [cs.LG] 对于此版本)
  https://doi.org/10.48550/arXiv.2507.02085
通过 DataCite 发表的 arXiv DOI

提交历史

来自: Wanjia Zhao [查看电子邮件]
[v1] 星期三, 2025 年 7 月 2 日 18:44:03 UTC (1,540 KB)
全文链接:

获取论文:

    查看标题为《》的 PDF
  • 查看中文 PDF
  • 查看 PDF
  • HTML(实验性)
  • TeX 源代码
  • 其他格式
许可图标 查看许可
当前浏览上下文:
cs.LG
< 上一篇   |   下一篇 >
新的 | 最近的 | 2025-07
切换浏览方式为:
cs
cs.AI

参考文献与引用

  • NASA ADS
  • 谷歌学术搜索
  • 语义学者
a 导出 BibTeX 引用 加载中...

BibTeX 格式的引用

×
数据由提供:

收藏

BibSonomy logo Reddit logo

文献和引用工具

文献资源探索 (什么是资源探索?)
连接的论文 (什么是连接的论文?)
Litmaps (什么是 Litmaps?)
scite 智能引用 (什么是智能引用?)

与本文相关的代码,数据和媒体

alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)

演示

复制 (什么是复制?)
Hugging Face Spaces (什么是 Spaces?)
TXYZ.AI (什么是 TXYZ.AI?)

推荐器和搜索工具

影响之花 (什么是影响之花?)
核心推荐器 (什么是核心?)
IArxiv 推荐器 (什么是 IArxiv?)
  • 作者
  • 地点
  • 机构
  • 主题

arXivLabs:与社区合作伙伴的实验项目

arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。

与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。

有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.

这篇论文的哪些作者是支持者? | 禁用 MathJax (什么是 MathJax?)
  • 关于
  • 帮助
  • contact arXivClick here to contact arXiv 联系
  • 订阅 arXiv 邮件列表点击这里订阅 订阅
  • 版权
  • 隐私政策
  • 网络无障碍帮助
  • arXiv 运营状态
    通过...获取状态通知 email 或者 slack

京ICP备2025123034号