PyKEEN 1.0: 지식 그래프 임베딩 훈련 및 평가를 위한 파이썬 라이브러리

최근 지식 그래프 임베딩(KGE)에 대한 관심이 크게 증가하였으며, KGE의 학습 및 평가를 위한 여러 소프트웨어 라이브러리가 개발되어 왔다. 각 라이브러리는 특정한 요구사항을 충족하기는 하지만, 우리는 공동 개발을 통해 초기 KGE 라이브러리 중 하나인 PyKEEN을 재설계하고 재구현하였다. PyKEEN 1.0은 다양한 상호작용 모델, 학습 방법, 손실 함수를 기반으로 지식 그래프 임베딩 모델(KGEM)을 조합할 수 있도록 하며, 역관계(역방향 관계)의 명시적 모델링을 지원한다. 또한, 제공된 하드웨어 자원을 최적화하여 활용하기 위한 자동 메모리 최적화 기능이 구현되었으며, Optuna와의 통합을 통해 광범위한 하이퍼파라미터 최적화(HPO) 기능을 제공한다.