Описание промпта

Процесс разворачивается в пять основных этапов: ввод данных, построение графа, обработка признаков, основное вычисление и прогнозирование результата, с акцентом на ключевую роль характеристик гетерогенного графа, построение индекса мета-путей и механизмы внимания на уровне узлов. Ниже приводится подробное описание процесса: Общий обзор процесса модели Модель THAN, основанная на гетерогенном графе (содержащем пользователей, каскадные узлы и различные взаимосвязи), захватывает семантические ассоциации между узлами посредством индексации мета-путей, объединяет механизм временного затухания и многоголовое внимание для вычисления вложений узлов и в конечном итоге используется для одношагового прогнозирования распространения каскада. Процесс можно разделить на 5 основных этапов, как показано ниже: 1. Ввод и инициализация данных Входные данные: Данные гетерогенного графа (граф): включает узлы (пользователь, каскад), ребра (социальные, взаимодействие, распространение) и атрибуты ребер (например, временная метка). Начальные признаки: признаки пользователя (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