Khái niệm cốt lõi
그래프 신경망(GNN)은 전통적인 조합 최적화 문제 해결 방법을 향상시킬 수 있는 잠재력이 있지만, 특히 무작위 그래프에서 좋은 초기화 및 지역 최소값 회피와 같은 과제를 해결해야 한다.
Tóm tắt
그래프 신경망을 이용한 조합 최적화 문제 해결 방안 연구
본 연구는 그래프 신경망(GNN)을 사용하여 조합 최적화(CO) 문제, 특히 최대 독립 집합(MIS) 문제를 해결하는 데 있어서의 효과와 한계를 조사하고, 기존의 탐욕 알고리즘보다 향상된 성능을 보이는 GNN 기반 솔버를 개발하는 것을 목표로 한다.
본 연구에서는 슈츠(Schuetz) 등이 제안한 비지도 학습 기반 QUBO(Quadratic Unconstrained Binary Optimization) 접근 방식을 기반으로, 노드 특징 초기화 개선, QUBO 함수 최적화, 지도 학습 방식과의 결합을 통해 GNN 솔버를 향상시켰다.
노드 임베딩 초기화 개선
연결이 적은 노드가 독립 집합에 포함될 가능성이 높다는 점을 고려하여, 각 노드의 연결 수(degree)를 기반으로 노드 특징을 초기화했다.
Min-max 정규화를 사용하여 각 노드의 degree를 정규화하고, 정규화된 degree가 낮은 노드가 독립 집합에 포함될 가능성을 높이도록 변환했다.
Hamiltonian 함수 수정
독립 집합에 선택된 노드에 대해 연결 수에 따라 보상을 다르게 하여, 연결이 적은 노드가 더 많이 선택되도록 Hamiltonian 함수를 수정했다.
노드 특징 초기화에서 계산된 초기 노드 특징을 사용하여 각 노드에 대한 보상을 생성했다.
그래프 신경망 아키텍처
슈츠의 논문과 유사하게 2개의 그래프 Convolution 레이어를 사용하고, 출력값을 sigmoid 함수를 통해 각 노드에 대한 확률로 변환했다.
QUBO 근사 손실이 0보다 크면 매개변수를 다시 초기화하여 모델이 지역 최소값에 갇히는 것을 방지했다.
또한, 최소화 결과가 좋지 않을 경우 프로세스를 다시 실행하도록 페널티 임계값을 설정했다.
탐욕적 디코딩 전략과의 통합
GNN의 예측 결과를 바탕으로 탐욕적 디코딩 전략을 적용하여 최종 솔루션의 유효성을 보장했다.
GNN 예측과 degree 기반 초기화를 결합한 최종 점수를 기반으로 탐욕적 디코딩을 수행하여 최종 솔루션을 찾았다.
지도 학습 + 비지도 학습 QUBO 접근 방식
GUROBI 솔버를 사용하여 생성한 ground truth 데이터를 활용하여 지도 학습 모델을 학습시켰다.
학습된 모델을 사용하여 보이지 않는 그래프에 대한 예측을 수행하고, 그 결과를 노드 특징으로 사용하여 비지도 학습 QUBO 접근 방식을 적용했다.