Optimal message passing for molecular prediction is simple, attentive and spatial

#### Контекст Молекулярные свойства, такие как активность и токсичность, широко используются в науке, промышленности и медицине. Однако предсказание этих свойств часто требует глубоких анализов, дорогостоящих экспериментов и многочисленных вычислений. Нейронные сети с пробросом сообщений (Message-Passing Neural Networks, MPNNs) позволили улучшить прогностическую точность за счет использования графов, характеризующих молекулы. Несмотря на эти успехи, существуют проблемы, такие как высокая сложность моделей, требующая больших ресурсов для обучения и использования. Было показано, что упрощение архитектуры модели и использование более широкого спектра дескрипторов графа могут повысить точность предсказаний и уменьшить накладные расходы на вычисления. #### Метод В этой работе предложены модели с простой, акцентуированной и пространственной обработкой сообщений для молекулярных предсказаний. Мы разработали архитектуры, которые добиваются самого высокого качества прогнозов, превосходящих даже сложные сети, обученные на внешних базах данных. Мы использовали минималистичные формулы сообщений, убрав самопонимание узлов и ограничив активность вывода аннотаций. Мы проанализировали разнообразие данных и определили, что структурная разнообразность может влиять на необходимость дополнительных компонент в MPNNs. Также мы исследовали влияние добавления пространственных признаков и работы с 3D-графами. Наши результаты показали, что 2D-графы, дополненные 3D-дескрипторами, дают точность, равную или превосходящей 3D-представления, при этом сокращая вычислительные затраты на 50%. #### Результаты Мы провели эксперименты на нескольких датасетах, сравнивая наши архитектуры с классическими MPNNs. Наши модели показали лучшие результаты в классификации и регрессии молекулярных свойств с помощью бидирекционального проброса сообщений с аннотациями. Мы подтвердили, что нормализация конвейера не всегда улучшает прогностическую точность во всех датасетах. Было показано, что упрощенные модели, убравшие самопонимание узлов, дают более высокую классовую разделяемость. Мы также проанализировали влияние 3D-признаков и пространственных функций, показав, что 2D-графы с 3D-дескрипторами обеспечивают высокую точность с минимальным трудоемкостным вкладом. #### Значимость Наши модели могут применяться в сферах, требующих быстрого и точного предсказания молекулярных свойств, таких как химическая синтезная программа, фармакология и химический разработка. Они предлагают существенное сокращение вычислительных затрат б