[논문 리뷰] SkexGen: Autoregressive Generation of CAD Construction Sequences with Disentangled Codebooks

Xiang Xu, Karl D. D. Willis외 4명|arXiv (Cornell University)|2022. 07. 11.

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 구축 시퀀스를 생성하는 생성 모델을 개발하는 것.
• 위상 변화에 취약하고 생성에 전문 지식이 필요한 파arametric CAD 모델의 한계를 해결하는 것.
• 위상, 기하학 및 연출 변형을 분리된 잠재 표현을 통해 분리함으로써 설계 공간 탐색을 효율적으로 가능하게 하는 것.
• 기존 자동회귀 모델을 향상시키기 위해 다양성과 제어 가능성을 향상시키는 코드북 기반 제어 메커니즘을 도입하는 것.
• 표준 생성을 초월하여 코드북 벡터의 보간 및 혼합을 통해 새로운 설계 합성을 가능하게 함을 입증하는 것.

제안 방법

• SkexGen는 스케치-익스트루전 시퀀스에서 위상, 기하학 또는 연출 변형을 각각 별도의 코드북에 인코딩하기 위해 세 개의 별도 Transformer 인코더를 사용한다.
• 모델은 각 코드북이 별도의 설계 속성을 저장하도록 이산 벡터 양자화 VAE(VQ-VAE) 프레임워크를 활용하여 시퀀스를 이산 토큰으로 매핑한다.
• 자동회귀 Transformer 디코더는 샘플링된 코드북 벡터를 조건으로 하여 2D 스케치와 3D 익스트루션을 단계별로 제어 가능한 방식으로 생성한다.
• 코드북 벡터는 양자화되어 자동회귀 디코딩 중 조건 신호로 사용되며, 설계 속성의 정밀한 조작을 가능하게 한다.
• 모델은 대규모 스케치-익스트루전 CAD 데이터셋에서 훈련되며, 재구성 및 적대적 목적을 통해 생성 품질을 향상시킨다.
• 다른 소스 모델의 코드북 벡터를 선형으로 보간하거나 조합함으로써 보간 및 코드 혼합을 가능하게 한다.

실험 결과

연구 질문

• RQ1 분리된 코드북은 CAD 구축 시퀀스에서 위상, 기하학 및 연출 변형을 효과적으로 분리하고 제어할 수 있는가?
• RQ2 이산 코드북 벡터를 조건으로 하는 자동회귀 생성은 다채롭고 고품질의 CAD 모델을 제어 가능한 방식으로 생성할 수 있는가?
• RQ3 설계 속성의 분리가 통합 또는 비분리된 표현에 비해 생성된 CAD 모델의 품질과 다양성에 어떤 영향을 미치는가?
• RQ4 분리된 코드북의 보간 및 혼합이 훈련 데이터에 없던 새로운 타당한 CAD 설계를 생성할 수 있는가?
• RQ5 모델은 정성적 및 정량적 벤치마크에서 기존 최신 기술 수준의 방법들을 얼마나 뛰어나게 성능을 내는가?

주요 결과

• SkexGen는 분리도 분류 정확도 99.8±0.1%를 달성하여 잠재 코드북 내에서 위상, 기하학 및 연출 요소를 효과적으로 분리함을 입증한다.
• DeepCAD 및 CurveGen 등의 베이스라인에 비해 복잡한 다단계 스케치-익스트루션 시퀀스를 포함한 훨씬 더 다양한 실재성 있는 CAD 모델을 생성한다.
• 여러 코드북을 제거하거나 단일 코드북을 사용할 경우 생성 품질이 떨어지며, 이는 분리의 유용성을 확인한다.
• 코드북 벡터 간의 보간은 선형적이지 않은 형태 변화를 포함한 의미 있는 비연속적 형태 변화(예: 선에서 원으로의 전이, 실체에서 빈도로의 전이)를 생성한다.
• 코드 혼합은 하이브리드 설계를 가능하게 하며, 예를 들어 한 모델의 위상과 다른 모델의 기하학을 조합함으로써 훈련 데이터에 존재하지 않는 새로운 구성 요소를 생성한다.
• 비록 추론 속도가 느리며(10,000개 샘플에 90초), 하지만 단일 자동회귀 디코더 덕분에 CurveGen보다 두 배 빠르며, 품질과 제어 가능성에서 슈퍼리어한 성능을 보인다.