| 公式サイト | https://github.com/txie-93/cdvae |
|---|---|
| 公開度 | ★★★ |
| ライセンス |
MIT License |
| 開発者 |
Tian Xie、Xiang Fuほか |
| 関連論文 |
|
| その他 |
CDVAEは、拡散モデルとVAEを組み合わせ、結晶対称性や格子制約を考慮しながら周期構造を生成する。データセットは、炭素同素体(組成固定)、同一プロトタイプを共有するペロブスカイト、一般無機結晶(単位胞あたり20原子上限)といった代表的設定を想定して整備されている。設定管理にはHydraを用い、PyTorch Lightningによる学習ループや、再構成・生成の妥当性・物性最適化などに対応する評価スクリプトを備える。データ駆動科学の観点では、結晶生成モデル実装とデータセット整備の参照例となる。 |
CDVAE
CDVAEは結晶材料を対象とした拡散モデルベースの変分オートエンコーダである。既存の結晶構造から学習し、新規の周期構造の生成や、潜在空間での目標物性に向けた最適化を行う。ベンチマーク用データセット(Perov-5、Carbon-24、MP-20)および結晶生成・物性条件付けのための学習/評価スクリプトが提供されている。