X-Node: Self-Explanation is All We Need

## Контекст Графовые нейронные сети (GNN) показали выдающиеся результаты в задачах классификации в областях компьютерного зрения и медико-медицинских изображений, обнаруживая структурные зависимости в данных. Однако их решения часто остаются непонятными, что снижает доверие к их применению в критически важных клинических задачах. Недостаток интерпретируемости в существующих GNN-решениях ограничивает их распространение в сферах, где необходима понимаемая интерпретация решений. Мы предлагаем X-Node, новую само-объясняющуюся модель GNN, в которой каждый узел самостоятельно формирует свое объяснение в процессе классификации. ## Метод X-Node использует структурированный контекстный вектор для каждого узла, содержащий ключевые признаки: степень узла, центральность, кластеризация, вклад функциональных признаков и согласования между функциональными и тегами узла. Эти признаки вычисляются с помощью локальной топологии. Далее, этот вектор передается в Reasoner — легковесный модуль, который формирует компактный вектор объяснения. Этот вектор выполняет три функции: (1) восстановление локального представления узла для улучшения достоверности модели, (2) генерация естественно-языковых объяснений с помощью предварительно обученных глубоких моделей текстов (например, Grok или Gemini), и (3) улучшение обучения GNN с помощью "текстовой инъекции" — внедрения объяснений в процесс массажа сообщений. Мы тестировали X-Node на двух наборах данных, полученных из MedMNIST и MorphoMNIST, в сочетании с GCN, GAT и GIN. ## Результаты Наши эксперименты показали, что X-Node привносит дополнительную подробность в понимание решений GNN без существенного снижения точности классификации. Он успешно формирует локальные объяснения для каждого узла, включая качественные и естественно-языковые пояснения. Кроме того, X-Node сохраняет точность классификации, демонстрируя ее на уровне стандартных GNN-моделей, но при этом предоставляя понятные причины своих решений на уровне каждого узла. ## Значимость X-Node может применяться в клинических задачах, где необходима интерпретируемая классификация, таких как диагностика заболеваний на основе медицинских изображений или анализ структурных данных. Модель предоставляет понятные причины своих решений, что увеличивает доверие к ней в критически важных задачах. Благодаря своему само-объясняющемуся подходу, X-Node может стать основой для развития интерпретируемых GNN-моделей в медицинских и других приложениях. ## Выводы Мы представили X-Node — новую само-объясняющуюся модель GNN, которая формирует понятные объяс