LeMat-Traj: A Scalable and Unified Dataset of Materials Trajectories for Atomistic Modeling

## Контекст Для создания точных моделей атомных взаимодействий, основанных на машинном обучении (MLIPs), необходимы большие и качественные данные о атомных конфигурациях. Однако существующие данные, полученные с помощью теории квантовой механики (Density Functional Theory, DFT), часто разбросаны в разных открытых и платных репозиториях, имеют разные форматы и метаданные, что ограничивает их эффективность. Добыча таких данных требует серьезных вычислительных ресурсов, и их объединение становится сложной задачей. Наша мотивация заключается в создании универсального набора данных, который станет базой для эффективного обучения MLIPs и повышения их точности. ## Метод Мы предлагаем **LeMat-Traj**, собранный из более чем 120 миллионов атомных конфигураций, полученных из трех крупных репозиториев: Materials Project, Alexandria и OQMD. Данные тщательно проверены и стандартизированы по формату, позволяя легко использовать их в обучении MLIPs. Добавляются оба типа конфигураций: с относительно низким энергетическим содержанием и с высоким уровнем силы. Для удобства исследователей разработана **LeMaterial-Fetcher** — модульная библиотека для получения и обработки данных. Эта библиотека позволяет добавлять новые источники данных и гарантировать повторяемость работ. ## Результаты Мы провели эксперименты, обучая MLIPs на LeMat-Traj и сравнивая их с другими подходами. Наш набор данных позволяет значительно сократить ошибки в предсказании сил, особенно при задачах релаксации. Библиотека LeMaterial-Fetcher позволила создать прозрачную и открытую инфраструктуру для работы с данными, что привлекло участие сообщества в развитии датасета. ## Значимость LeMat-Traj значительно упрощает процесс обучения MLIPs, обеспечивая доступ к большому, стандартизированному набору данных. Это вносит значительный вклад в развитие точных моделей, что может применяться в материаловедении, энергетике и химии. Библиотека LeMaterial-Fetcher, в свою очередь, позволяет улучшать и расширять датасет, создавая прозрачные инструменты для сообщества. ## Выводы Представленный набор данных LeMat-Traj и библиотека LeMaterial-Fetcher являются ключевыми инструментами для развития MLIPs. Мы наблюдаем существенную повышенную точность моделей и снижение ошибок в предсказании сил. Будущие исследования будут направлены на расширение датасета, внедрение новых функционалов DFT и исследование новых методов обучения MLIPs.