FACET: A Fragment-Aware Conformer Ensemble Transformer
University of Stuttgart and Max Planck Research School for Intelligent Systems · German Research Center for AI · National University of Singapore · Queensland University of Technology · University of Texas, Austin · VinUniversity · United States Naval Academy · University of Pennsylvania · Stanford University · University of Stuttgart & NEC Labs Europe · Lawrence Livermore National Labs
PDF 由论文原始站点提供，PaperCompass 不保存论文文件。
摘要
Accurately predicting molecular properties requires effective integration of structural information from both 2D molecular graphs and their corresponding equilibrium conformer ensembles. In this work, we propose FACET, a scalable Structure-Aware Graph Transformer that efficiently aggregates features from multiple 3D conformers while incorporating fragment-level information from 2D graphs. Unlike prior methods that rely on static geometric solvers or rigid fusion strategies, our approach utilizes a differentiable graph transformer to theoretically approximate the computationally expensive Fused Gromov–Wasserstein (FGW), enabling dynamic and scalable fusion of 2D and 3D structural information. We further enhance this mechanism by injecting fragment-specific structural priors into the attention layers, enabling the model to capture fine-grained molecular details. This unified design scales to large datasets, handling up to 75,000 molecules and hundreds of thousands of conformers, and provides over a 6× speedup compared to geometry-aware FGW-based baselines. Our method also achieves state-of-the-art results in molecular property prediction, Boltzmann-weighted ensemble modeling, and reaction-level tasks, and is particularly effective on chemically diverse compounds, including organocatalysts and transition-metal complexes.