Feature Augmentation of GNNs for ILPs: Local Uniqueness Suffices

## Контекст Линейные программы целых чисел (ILPs) являются ключевыми инструментами для решения оптимизационных задач в реальном мире, однако широко известны своей высокой сложностью. Недавно появилась методология Learning to Optimize (L2O), которая использует Graph Neural Networks (GNNs) в качестве основной архитектуры для решения ILPs. Однако, стандартные анонимные GNNs имеют ограниченную выразительность для ILPs, и часто применяемое улучшение — добавление глобальных уникальных идентификаторов (UIDs) — приводит к возникновению спурых корреляций, что сильно повлияло на общеупотребительность. Для решения этой проблемы мы предлагаем новый подход Local-UID, основанный на d-hop uniqueness coloring, который обеспечивает уникальность идентификаторов только в пределах d-хопного соседства каждого узла. На этой основе мы предлагаем ColorGNN, который включает цветовые информации через цвета-обученные представления, и ColorUID, легковесный вариант по уровню фич. Мы доказали, что Local-UIDs позволяют достичь выразительности Global-UIDs, при этом обеспечивая повышенную общеупотребительность. ## Метод Мы предлагаем Local-UID, схему, основанную на d-hop uniqueness coloring, которая гарантирует уникальность идентификаторов только в d-хопном соседстве каждого узла. Для улучшения выразительности GNNs мы представили ColorGNN с цветово-обученными представлениями и ColorUID с легковесной функцией уровня фич. Эти схемы были протестированы на трех ILP-benchmarks и показали их эффективность. Мы также доказали, что Local-UIDs могут позволить достичь тех же выразительных свойств, что и Global-UIDs, но с значительно большей общеупотребительностью. Эксперименты подтвердили, что наш подход не только улучшил результаты на ILP-benchmarks, но и вы mosted лучшую общеупотребительность в линейной оптимизации. ## Результаты Мы провели эксперименты на трех ILP-benchmarks и проанализировали результаты нашего подхода в сравнении с существующими методами. Наши эксперименты показали, что Local-UIDs позволяют достичь лучших результатов, особенно при работе с нестандартными данными. Также, мы протестировали нашу модель в задачах линейной оптимизации и показали, что она превосходит существующие методы в общей универсальности. Это подтвердило, что Local-UIDs могут предлагать существенные преимущества в решении ILPs и других задач оптимизации. ## Значимость Мы показали, что наш подход может быть применен в системах управления и оптимизации реальных процессов. Кроме того, технология цветовых представлений может быть использована в области графовых архитектур, улучшая общеупотребительность. Этот подход может быть использован в различных приложениях, включая логистику, произ