pyt-team/TopoModelX

해결하는 문제

TopoModelX (TMX)는 과학 및 공학 분야에서 위상 신경망(TNN)을 구현하기 위한 도구를 제공합니다. 단순한 그래프가 아닌 단체 복합체와 같은 위상 도메인을 활용하여 데이터 내의 고차원 상호작용을 포착할 수 있도록 합니다.

작동 방식

TMX는 PyTorch와 통합된 Python 모듈입니다. 단체 복합체와 그에 연결된 행렬(상향 및 하향 라플라시안, 인시던스 행렬 등)을 사용하여 신경망 레이어를 정의합니다. 예를 들어, Simplicial Attention Networks(SAN)의 구현을 포함하여 노드와 엣지의 특징을 동시에 처리하여 위상 데이터에 대한 순전파를 수행할 수 있습니다.

대상 사용자

단순한 그래프가 아닌 위상 복합체 구조로 데이터가 구성된 과학 및 공학 분야의 연구자 및 엔지니어를 위한 것입니다.

주요 특징

• 단체 주의 네트워크(SAN): 위상 딥러닝을 위한 전용 레이어를 구현합니다.
• PyTorch 통합: 효율적인 학습 및 학습 워크플로우를 위한 PyTorch 모듈로 구축되었습니다.
• TopoX의 일부: 위상 도메인에서의 머신러닝을 위한 광범위한 TopoX 패키지 세트와 통합됩니다.
• 희소 행렬 지원: 위상 라플라시안의 효율적인 처리를 위해 희소 행렬을 활용합니다.