Skip to content

【Hackathon 10th Spring No.17】MOFDiff模型复现#271

Open
r-cloudforge wants to merge 5 commits intoPaddlePaddle:developfrom
CloudForge-Solutions:task/017-mofdiff-reproduction
Open

【Hackathon 10th Spring No.17】MOFDiff模型复现#271
r-cloudforge wants to merge 5 commits intoPaddlePaddle:developfrom
CloudForge-Solutions:task/017-mofdiff-reproduction

Conversation

@r-cloudforge
Copy link
Copy Markdown

通过迁移 https://github.com/microsoft/MOFDiff 项目实现 MOFDiff 的PaddlePaddle版本复现。

主要改动

  1. 新增 ppmat/models/mofdiff/ 模型目录 — 完整的MOFDiff实现,包括VAE编码/解码器和VP扩散过程
  2. 新增 structure_generation/configs/mofdiff/ 训练配置
  3. 新增 test/mofdiff/ 测试代码 — 18个单元测试

模型特性

  • VAE + VP (Variance Preserving) 扩散模型,用于金属有机框架(MOF)结构生成
  • SimpleGNN编码器/解码器用于图结构处理
  • 高斯傅里叶投影时间嵌入
  • VP噪声调度器,支持前向SDE和反向ODE
  • 拓扑感知的连续和离散特征处理
  • 基于PaddleMaterials标准API:forward(data){"loss_dict", "pred_dict"}
  • sample() 方法支持完整采样流程
  • 已注册到 ppmat/models/__init__.py

测试

cd test && python -m pytest mofdiff/ -v
# 18 passed

参考链接

cloudforge1 and others added 4 commits March 28, 2026 10:00
Implement coarse-grained diffusion model for MOF generation.
Includes VP diffusion, GaussianFourierProjection, VAE encoder,
simplified GNN encoder/decoder, and 18 unit tests.

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
@CLAassistant
Copy link
Copy Markdown

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.


cloudforge1 seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account.
You have signed the CLA already but the status is still pending? Let us recheck it.

@paddle-bot
Copy link
Copy Markdown

paddle-bot bot commented Apr 11, 2026

Thanks for your contribution!

@paddle-bot paddle-bot bot added the contributor External developers label Apr 11, 2026
…weight

- Replace linear coord noise with VP diffusion (consistent with BB types)
- Make KL loss weight configurable (default 0.1)

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

contributor External developers

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants