論文の概要: Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generation
- arxiv url: http://arxiv.org/abs/2602.15022v1
- Date: Mon, 16 Feb 2026 18:58:55 GMT
- ステータス: 翻訳完了
- システム内更新日: 2026-02-17 16:22:50.647915
- Title: Rethinking Diffusion Models with Symmetries through Canonicalization with Applications to Molecular Graph Generation
- Title(参考訳): 正準化による拡散モデルの再考と分子グラフ生成への応用
- Authors: Cai Zhou, Zijie Chen, Zian Li, Jike Wang, Kaiyi Jiang, Pan Li, Rose Yu, Muhan Zhang, Stephen Bates, Tommi Jaakkola,
- Abstract要約: CanonFlowは、挑戦的なGEOM-DRUGデータセット上で最先端のパフォーマンスを実現している。
- 参考スコア(独自算出の注目度): 56.361076943802594
- License: http://creativecommons.org/licenses/by/4.0/
- Abstract: Many generative tasks in chemistry and science involve distributions invariant to group symmetries (e.g., permutation and rotation). A common strategy enforces invariance and equivariance through architectural constraints such as equivariant denoisers and invariant priors. In this paper, we challenge this tradition through the alternative canonicalization perspective: first map each sample to an orbit representative with a canonical pose or order, train an unconstrained (non-equivariant) diffusion or flow model on the canonical slice, and finally recover the invariant distribution by sampling a random symmetry transform at generation time. Building on a formal quotient-space perspective, our work provides a comprehensive theory of canonical diffusion by proving: (i) the correctness, universality and superior expressivity of canonical generative models over invariant targets; (ii) canonicalization accelerates training by removing diffusion score complexity induced by group mixtures and reducing conditional variance in flow matching. We then show that aligned priors and optimal transport act complementarily with canonicalization and further improves training efficiency. We instantiate the framework for molecular graph generation under $S_n \times SE(3)$ symmetries. By leveraging geometric spectra-based canonicalization and mild positional encodings, canonical diffusion significantly outperforms equivariant baselines in 3D molecule generation tasks, with similar or even less computation. Moreover, with a novel architecture Canon, CanonFlow achieves state-of-the-art performance on the challenging GEOM-DRUG dataset, and the advantage remains large in few-step generation.
- Abstract(参考訳): 化学と科学における多くの生成的タスクは、群対称性(例えば、置換と回転)に不変な分布を含む。
共通戦略は、等変デノイザーや不変先例のようなアーキテクチャ上の制約を通じて不変性と等変を強制する。
本稿では,まず各サンプルを標準ポーズや順序で軌道代表にマップし,標準スライス上で非拘束的(非同変)拡散モデルやフローモデルを訓練し,ランダム対称性変換を生成時にサンプリングすることで不変分布を復元する。
形式的な商空間の観点に基づいて、我々の研究は証明による正準拡散の包括的理論を提供する。
一 不変目標に対する標準生成モデルの正当性、普遍性及び優れた表現性
(II)正準化は,群混合による拡散スコアの複雑性を除去し,フローマッチングにおける条件分散を低減することにより,トレーニングを加速させる。
次に、整列先と最適輸送が正準化と相補的に作用し、トレーニング効率をさらに向上することを示す。
分子グラフ生成のフレームワークを$S_n \times SE(3)$ symmetriesでインスタンス化する。
幾何スペクトルに基づく正準化と軽微な位置符号化を利用することで、正準拡散は3次元分子生成タスクにおける同変基底線を、類似あるいはより少ない計算で著しく上回る。
さらに、新しいアーキテクチャであるCanonでは、GEOM-DRUGデータセットに挑戦する上で、最先端のパフォーマンスを実現している。
関連論文リスト
- Approximate equivariance via projection-based regularisation [1.5755923640031846]
非同変モデルは、実行時のパフォーマンスと不完全な対称性のため、再び注目を集めている。
これは、対称性の尊重とデータ分布の適合の間に中間点を打つ、ほぼ同変モデルの開発を動機付けている。
本研究では,空間領域とスペクトル領域の両方において,非等分散ペナルティを正確に,効率的に計算するための数学的枠組みを提案する。
論文 参考訳(メタデータ) (2026-01-08T15:35:42Z) - Improving Equivariant Networks with Probabilistic Symmetry Breaking [9.164167226137664]
同変ネットワークは既知の対称性をニューラルネットワークにエンコードし、しばしば一般化を強化する。
これは(1)自己対称性が共通な領域での予測タスク、(2)高対称性の潜在空間から再構成するために対称性を破らなければならない生成モデルの両方に重要な問題を引き起こす。
このような分布を表すのに十分な条件を確立する新しい理論結果を示す。
論文 参考訳(メタデータ) (2025-03-27T21:04:49Z) - Rao-Blackwell Gradient Estimators for Equivariant Denoising Diffusion [55.95767828747407]
分子やタンパク質の生成のようなドメインでは、物理系はモデルにとって重要な固有の対称性を示す。
学習のばらつきを低減し、確率的に低い分散勾配推定器を提供するフレームワークを提案する。
また,軌道拡散法(Orbit Diffusion)と呼ばれる手法を用いて,損失とサンプリングの手順を取り入れた推定器の実用的実装を提案する。
論文 参考訳(メタデータ) (2025-02-14T03:26:57Z) - SymDiff: Equivariant Diffusion via Stochastic Symmetrisation [28.614292092399563]
対称性の枠組みを用いた同変拡散モデル構築手法であるSymDiffを提案する。
SymDiffは、サンプリング時にデプロイされる学習データ拡張に似ており、軽量で、計算効率が高く、任意のオフザシェルフモデル上に実装が容易である。
論文 参考訳(メタデータ) (2024-10-08T18:02:29Z) - Uniform Transformation: Refining Latent Representation in Variational Autoencoders [7.4316292428754105]
本稿では,不規則な潜伏分布に対応するために,新しい適応型3段階一様変換(UT)モジュールを提案する。
この手法は不規則分布を潜在空間の均一分布に再構成することにより、潜在表現の絡み合いと解釈可能性を大幅に向上させる。
実験により,提案するUTモジュールの有効性を実証し,ベンチマークデータセット間の絡み合いの指標を改良した。
論文 参考訳(メタデータ) (2024-07-02T21:46:23Z) - Gaussian Mixture Solvers for Diffusion Models [84.83349474361204]
本稿では,拡散モデルのためのGMSと呼ばれる,SDEに基づく新しい解法について紹介する。
画像生成およびストロークベース合成におけるサンプル品質の観点から,SDEに基づく多くの解法よりも優れる。
論文 参考訳(メタデータ) (2023-11-02T02:05:38Z) - Geometric Neural Diffusion Processes [55.891428654434634]
拡散モデルの枠組みを拡張して、無限次元モデリングに一連の幾何学的先行を組み込む。
これらの条件で、生成関数モデルが同じ対称性を持つことを示す。
論文 参考訳(メタデータ) (2023-07-11T16:51:38Z) - A Geometric Perspective on Diffusion Models [57.27857591493788]
本稿では,人気のある分散拡散型SDEのODEに基づくサンプリングについて検討する。
我々は、最適なODEベースのサンプリングと古典的な平均シフト(モード探索)アルゴリズムの理論的関係を確立する。
論文 参考訳(メタデータ) (2023-05-31T15:33:16Z) - GeoDiff: a Geometric Diffusion Model for Molecular Conformation
Generation [102.85440102147267]
分子配座予測のための新しい生成モデルGeoDiffを提案する。
GeoDiffは、既存の最先端のアプローチよりも優れているか、あるいは同等であることを示す。
論文 参考訳(メタデータ) (2022-03-06T09:47:01Z)
関連論文リストは本サイト内にある論文のタイトル・アブストラクトから自動的に作成しています。
指定された論文の情報です。
本サイトの運営者は本サイト(すべての情報・翻訳含む)の品質を保証せず、本サイト(すべての情報・翻訳含む)を使用して発生したあらゆる結果について一切の責任を負いません。