FlowMol3: Flow Matching for 3D De Novo Small-Molecule Generation

2508.12629v1 cs.LG, q-bio.BM 2025-08-20
Авторы:

Ian Dunn, David R. Koes

Резюме на русском

#### Контекст Разработка моделей для генерации реалистичных молекул с желаемыми свойствами является ключевым направлением в синтетической химии. Это позволяет ускорить научные исследования в области новых лекарств, материалов и химических реакций. Однако достижение высокой точности и стабильности в генерации молекул остается сложной задачей. Многие нынешние модели сталкиваются с проблемами, такими как неточность в формировании геометрии молекул и распределения функциональных групп. В свете этого, FlowMol3 предлагает эффективное решение для генерации трехмерных молекул на основе метода "Flow Matching", который демонстрирует значительный прогресс в этой области. #### Метод FlowMol3 — это расширенная модель Flow Matching, основанная на методах генеративной моделирования с использованием трансформеров. Она обладает трехмерной архитектурой, которая одновременно учитывает топологию молекул и их трехмерное распределение. Основными техническими приемами являются: 1. **Self-Conditioning** — добавление информации о самой модели в процесс обучения, что позволяет улучшить стабильность и точность. 2. **Fake Atoms** — ввод фиктивных атомов для улучшения распределения геометрии. 3. **Train-Time Geometry Distortion** — динамическое изменение геометрии молекул во время обучения для более точного совпадения с данными. Эти приемы не требуют дополнительных ресурсов и могут быть легко реализованы в рамках существующих архитектур. #### Результаты FlowMol3 показала выдающиеся результаты в генерации трехмерных молекул. Она достигла близкого к 100% точности в генерации молекул с высокой лекарственной релевантностью. Также, модель показала значительные улучшения в учете функциональных групп и геометрии в сравнении с предыдущими моделями. Ключевой выгодой является ее эффективность — FlowMol3 работает с меньшим числом параметров, что делает ее более производительной и экономичной. #### Значимость Модель FlowMol3 может применяться в различных областях, включая синтез лекарств, новые материалы и экологически безопасные компоненты. Основные преимущества заключаются в высокой точности, небольшом количестве параметров и гибкости в применении. Ее развитие может повлиять на технологический прогресс в химии и науках о жизни, позволив создавать новые молекулы с желаемыми характеристиками быстрее и эффективнее. #### Выводы FlowMol3 продемонстрировала значительные достижения в области генерации трехмерных молекул. Ее удачное использование простых, но эффективных методик является примером того, как модели могут быть улучшены без глубоких изменений архитекту

Abstract

A generative model capable of sampling realistic molecules with desired properties could accelerate chemical discovery across a wide range of applications. Toward this goal, significant effort has focused on developing models that jointly sample molecular topology and 3D structure. We present FlowMol3, an open-source, multi-modal flow matching model that advances the state of the art for all-atom, small-molecule generation. Its substantial performance gains over previous FlowMol versions are achieved without changes to the graph neural network architecture or the underlying flow matching formulation. Instead, FlowMol3's improvements arise from three architecture-agnostic techniques that incur negligible computational cost: self-conditioning, fake atoms, and train-time geometry distortion. FlowMol3 achieves nearly 100% molecular validity for drug-like molecules with explicit hydrogens, more accurately reproduces the functional group composition and geometry of its training data, and does so with an order of magnitude fewer learnable parameters than comparable methods. We hypothesize that these techniques mitigate a general pathology affecting transport-based generative models, enabling detection and correction of distribution drift during inference. Our results highlight simple, transferable strategies for improving the stability and quality of diffusion- and flow-based molecular generative models.

Ссылки и действия