보조인자를 활용한 그래프 신경망 기반의 결측치 복원 방법 연구

Abstract

결측치 처리는 심층학습 및 기계학습의 성능을 높이기 위한 중요한 과정이다. 결측치를 처리하는 기존의 방법들은 계산 비효율성 때문에 대규모 데이터를 다루는 데에는 적합하지 않다. 그래프 신경망이 대규모 데이터셋에서 우수한 성능을 보임에 따라, 결측치 처리를 그래프의 간선 예측 문제로 전환하여 해결하고자 하는 시도들이 있었고, 다양한 벤치마크 데이터셋에서 그래프 기반의 결측치 예측 알고리즘은 우수한 성능을 보였다. 그러나 밀도가 희박하고 잡음이 섞인 실제 데이터셋에 그래프 기반의 알고리즘을 적용하는 데에는 추가적인 노력이 필요하다. 본 논문은 보조인자를 도입하여, 대상인자의 간선을 예측하는 방법을 제안한다. 보조인자는 예측하고자 하는 대상인자를 제외한 데이터셋의 인자들로, 결측치 처리 모델이 그래프 정점 표현 사이의 차이를 반영할 수 있도록 돕는 역할을 한다. 실험결과들을 통해 그래프 기반의 결측치 처리 방법의 성능을 검증하고, 보조인자의 도입을 통해 실제 데이터셋의 상하한 값 내의 다양한 값을 생성하는 것을 확인할 수 있었다.