Facet: highly efficient E(3)-equivariant networks for interatomic potentials

#### Контекст Модели машинного обучения (ML), предсказывающие энергии молекул или структур, являются важной частью современных исследований в области вычислительных материалов. Однако существуют значительные проблемы с их надёжностью и эффективностью. Одним из ключевых элементов является сохранение геометрических симметрий, таких как перестановки, повороты и сдвиги атомов. Эти симметрии требуют инвариантности или эквивариантности моделей к таким преобразованиям. Несмотря на развитие графных нейронных сетей (GNN), которые могут учитывать такие симметрии, они часто сталкиваются с проблемами эффективности и точности. Большая часть существующих моделей либо неэффективна, либо не может обеспечить необходимую точность в задачах, таких как прогнозирование характеристик материалов на основе геометрии ячейки кристалла. #### Метод Facet — это новая GNN-архитектура, разработанная с целью решения этих проблем. Она базируется на систематическом анализе существующих steerable GNNs и применении инновационных техник. Основным изобретением является замена дорогостоящих многослойных перцептронов (MLP) для обработки интераторных расстояний на более лёгкие сплайны, которые сохраняют точность но при этом уменьшают вычислительные и памятные нагрузки. Кроме того, разработана общеельно-целевая эквивариантная слоя, которая использует проекцию на сферическую решётку для перемешивания узловых данных и последующей обработки в стандартных MLP. Эта слой быстрее и более выразительна, чем существующие градиентные или гейт-слои. #### Результаты Facet протестирована на датасете MPTrj и сравнивалась с лидирующими методами. Она показала схожие результаты с меньшим числом параметров и значительно меньшим объёмом вычислительной нагрузки во время обучения. На задаче релаксации кристалла Facet работает в два раза быстрее, чем MACE-модели. Более того, проводились исследования на SevenNet-0, где удалось уменьшить число параметров на более чем 25% без потери точности. Эти результаты демонстрируют высокую эффективность и гибкость Facet в области ML-потенциалов. #### Значимость Facet открывает широкие возможности для эффективного машинного обучения в материаловедении. Она позволяет сократить сроки обучения больших моделей до 10 раз при сохранении высокой точности. Это может существенно повлиять на индустрию, снижая затраты на вычисления и увеличивая скорость открытия новых материалов. Будущие исследования могут сфокусироваться на расширении приложений Facet в прочие области, такие как квантовая химия и вычислительная б