[論文レビュー] SkexGen: Autoregressive Generation of CAD Construction Sequences with Disentangled Codebooks

Xiang Xu, Karl D. D. Willis+4 more|arXiv (Cornell University)|Jul 11, 2022

Manufacturing Process and Optimization被引用数 11

ひとこと要約

SkexGenは、分離されたコードブックを用いてトポロジー、ジオメトリ、エクストルージョンの変異を別々に符号化することで、CAD構築シーケンスの自己回帰的生成モデルを導入する。別々のTransformerエンコーダーとデコーダーを活用することで、高品質で多様なCAD生成が可能となり、設計要因に対する明確なユーザー制御が実現され、制御性とデザインスペース探索の分野で最先端の性能を達成する。

ABSTRACT

We present SkexGen, a novel autoregressive generative model for computer-aided design (CAD) construction sequences containing sketch-and-extrude modeling operations. Our model utilizes distinct Transformer architectures to encode topological, geometric, and extrusion variations of construction sequences into disentangled codebooks. Autoregressive Transformer decoders generate CAD construction sequences sharing certain properties specified by the codebook vectors. Extensive experiments demonstrate that our disentangled codebook representation generates diverse and high-quality CAD models, enhances user control, and enables efficient exploration of the design space. The code is available at https://samxuxiang.github.io/skexgen.

研究の動機と目的

• 設計の変異に対する明確なユーザー制御を可能にする、多様で高品質なCAD構築シーケンスを生成する生成モデルの開発。
• トポロジーの変更に対して脆弱であり、作成に専門的知識を要するパrametric CADモデルの限界を克服すること。
• トポロジー、ジオメトリ、エクストルージョンの変異を分離された潜在表現によって分離することで、デザインスペースの効率的探索を可能にすること。
• 生成の多様性と制御性を向上させるコードブックベースの制御メカニズムを導入することで、既存の自己回帰的モデルを改善すること。
• 標準的な生成を超えた、新たなデザイン合成のためのコードブックベクトルの補間および混合の可能性を実証すること。

提案手法

• SkexGenは、スケッチ・アンド・エクストルードシーケンスから、それぞれトポロジー、ジオメトリ、エクストルージョンの変異を別々のコードブックに符号化するための3つの独立したTransformerエンコーダーを用いる。
• モデルは、シーケンスを離散トークンにマッピングするための離散的ベクトル量子化VAE（VQ-VAE）フレームワークを採用し、各コードブックが異なる設計要因を格納する。
• 自己回帰的Transformerデコーダーは、サンプリングされたコードブックベクトルを条件として用い、2次元スケッチと3次元エクストルージョンを段階的に制御的に生成する。
• コードブックベクトルは量子化され、自己回帰的デコーディング中に条件信号として使用され、設計要因の正確な操作が可能になる。
• モデルは大規模なスケッチ・アンド・エクストルードCADデータセットで学習され、再構成と敵対的目的を用いて生成品質を向上させる。
• 補間とコードミキシングは、異なるソースモデルからのコードブックベクトルを線形補間または組み合わせた後にデコーディングすることで実現される。

実験結果

リサーチクエスチョン

• RQ1 分離されたコードブックは、CAD構築シーケンスにおけるトポロジー、ジオメトリ、エクストルージョンの変異を効果的に分離・制御できるか？
• RQ2 離散的コードブックベクトルを条件とした自己回帰的生成は、多様で高品質なCADモデルを、制御性を向上させながら生成できるか？
• RQ3 設計要因の分離が、統合的または非分離的表現と比較して、生成されたCADモデルの品質と多様性に与える影響は何か？
• RQ4 分離されたコードブックの補間およびミキシングにより、学習データに存在しない新しい実現可能なCADデザインを生成できるか？
• RQ5 質的および量的ベンチマークにおいて、本モデルは既存の最先端手法をどの程度上回っているか？

主な発見

• SkexGenは、分離分類精度99.8±0.1%を達成し、潜在コードブックにおけるトポロジー、ジオメトリ、エクストルージョン要因の有効な分離を示している。
• DeepCAD や CurveGen などのベースラインと比較して、複雑な複数ステップのスケッチ・アンド・エクストルードシーケンスを含む、はるかに多様で現実的であるCADモデルを生成している。
• 複数のコードブックを削除するか、単一のコードブックを使用すると生成品質が低下するため、分離の利点が確認された。
• コードブックベクトル間の補間は、線から円への変換や、実心から中空への変化など、離散的なトポロジー変更を伴う意味のある非滑らかなモルフォロジーを生成する。
• コードミキシングによりハイブリッドデザインが可能となり、例えば1つのモデルのトポロジーと別のモデルのジオメトリを組み合わせることで、学習データに存在しない新しい構成が生成される。
• 推論が遅い（10,000サンプルで90秒）ものの、単一の自己回帰的デコーダーのおかげでCurveGenの2倍の速度で動作し、品質と制御性においても上回っている。