GEqTrain은 3D 과학 데이터 모델링에 사용되는 Equivariant 그래프 신경망의 재활용성을 높이는 프레임워크입니다. 데이터 의미론, 모델 구성, 학습 목표를 분리하여 다양한 작업에 쉽게 적용할 수 있도록 설계됐습니다.
생분자 시스템 매핑, NMR 화학적 이동 예측, Equivariant 생성 모델링 등 세 가지 다른 문제를 하나의 소프트웨어 스택으로 처리하는 데 활용 가능하며, 기존 시스템에 비해 설정 변경만으로 경쟁력 있는 정확도를 달성합니다.
GEqDiff는 Equivariant 흐름 매칭을 기반으로 한 생성 모델로, 다양한 변환 속성을 가진 필드를 함께 재구성하고 높은 충실도로 복원하는 데 사용됩니다.