PyTorch Frame: 다중 모드 표 형 학습을 위한 모듈식 프레임워크

우리는 PyTorch Frame을 소개합니다. 이는 다중 모달 테이블 데이터에 대한 딥 러닝을 위한 PyTorch 기반 프레임워크입니다. PyTorch Frame은 복잡한 테이블 데이터를 처리하기 위한 PyTorch 기반 데이터 구조를 제공하여 테이블 딥 러닝을 쉽게 만들고, 모델 추상화를 도입하여 테이블 모델의 모듈화된 구현을 가능하게 하며, 복잡한 열(예: 텍스트 열에 대한 LLMs)을 처리하기 위해 외부 기초 모델의 통합을 허용합니다. 우리는 다양한 테이블 모델을 모듈화된 방식으로 구현하고, 이러한 모델들을 성공적으로 복잡한 다중 모달 테이블 데이터에 적용하며, 그래프 신경망(GNNs) 라이브러리인 PyTorch Geometric과 우리의 프레임워크를 통합하여 관계형 데이터베이스에서 엔드투엔드 학습을 수행함으로써 PyTorch Frame의 유용성을 입증합니다.