프롬프트 설명

다섯 가지 핵심 단계로 전개되는 과정: 데이터 입력, 그래프 구성, 특징 처리, 핵심 연산, 출력 예측. 이 과정은 이종 그래프 특징, 메타 경로 인덱스 구성, 노드 수준 어텐션 메커니즘의 핵심 역할을 강조합니다. 다음은 상세한 과정 설명입니다. 모델 전체 과정 개요 THAN 모델은 이종 그래프(사용자, 캐스케이드 노드 및 다양한 관계 포함)를 기반으로 하며, 메타 경로 인덱싱을 통해 노드 간의 의미적 연관성을 포착하고, 시간 감쇠 메커니즘과 멀티 헤드 어텐션을 결합하여 노드 임베딩을 계산하며, 궁극적으로 캐스케이드 전파의 단일 단계 예측에 사용됩니다. 이 과정은 다음과 같이 5가지 핵심 단계로 나눌 수 있습니다. 1. 데이터 입력 및 초기화 입력 데이터: 이종 그래프 데이터 (graph): 노드 (사용자, 캐스케이드), 엣지 (소셜, 상호 작용, 확산) 및 엣지 속성 (예: 타임스탬프)을 포함합니다. 초기 특징: 사용자 특징 (user_initial_features) 및 캐스케이드 특징 (cascade_initial_features). 딕셔너리 형태로 저장 (id_to_idx는 노드 ID를 특징 인덱스에 매핑). 구성 파라미터: 메타 경로 유형 (예: U-U-social, U-U-interact, C-U-C), 어텐션 헤드 수, 시간 감쇠 계수 lambda_time 등. 초기화 작업: 장치 구성 (CPU/GPU) 및 메모리 최적화 (캐시 지우기, 비동기 데이터 전송). 모델 구성 요소 초기화: 특징 투영 레이어 (type_transform), 멀티 헤드 어텐션 파라미터 (att_params) 등. 2. 이종 그래프 메타 경로 인덱스 구성 build_metapath_index 메서드를 사용하여 메타 경로 인덱스를 미리 계산하여 다양한 유형의 노드 간의 연관성과 시간 감쇠 특징을 포착하여 후속 어텐션 계산 속도를 높입니다. 메타 경로 정의: U-U-social: 사용자 - 소셜 - 사용자 (타임스탬프 없음, 이웃 관계만 저장). U-U-interact: 사용자 - 상호 작용 - 사용자 (상호 작용 타임스탬프, 시간 감쇠 값 및 누적 합계 저장). C-U-C: 캐스케이드 - 사용자 - 캐스케이드 (중간 사용자를 통해 연결된 캐스케이드 관계, 확산 타임스탬프, 시간 감쇠 값 및 누적 합계 저장). 인덱스 내용: 시간 정보가 포함된 메타 경로 (U-U-interact, C-U-C)의 경우 노드 쌍별로 저장: 정렬된 타임스탬프 목록 (ts). 시간 감쇠 값 (decay, 1 - exp(-lambda*(t-T_earliest))를 기반으로 계산). 감쇠 값의 누적 합계 (prefix, 누적 가중치 계산 가속화). 인덱스 캐시: 반복적인 계산을 피하기 위해 계산 결과를 로컬에 저장 (save_metapath_full_cache). 3. 특징 투영 및 노드 임베딩 초기화 특징 투영: type_tr