Search papers, labs, and topics across Lattice.
EquiformerV3 enhances the SE(3)-equivariant graph attention Transformer by optimizing its software implementation, incorporating equivariant merged layer normalization, and introducing SwiGLU-S^2 activations for improved expressivity and equivariance. The SwiGLU-S^2 activations and smooth-cutoff attention enable accurate modeling of potential energy surfaces, generalizing the model to energy-conserving simulations. EquiformerV3 achieves state-of-the-art results on OC20, OMat24, and Matbench Discovery by training with the auxiliary task of denoising non-equilibrium structures (DeNS).
EquiformerV3 achieves state-of-the-art performance in atomistic modeling by combining architectural improvements with optimized software, enabling accurate energy-conserving simulations.
As SE(3)-equivariant graph neural networks mature as a core tool for 3D atomistic modeling, improving their efficiency, expressivity, and physical consistency has become a central challenge for large-scale applications. In this work, we introduce EquiformerV3, the third generation of the SE(3)-equivariant graph attention Transformer, designed to advance all three dimensions: efficiency, expressivity, and generality. Building on EquiformerV2, we have the following three key advances. First, we optimize the software implementation, achieving 1.75times speedup. Second, we introduce simple and effective modifications to EquiformerV2, including equivariant merged layer normalization, improved feedforward network hyper-parameters, and attention with smooth radius cutoff. Third, we propose SwiGLU-S^2 activations to incorporate many-body interactions for better theoretical expressivity and to preserve strict equivariance while reducing the complexity of sampling S^2 grids. Together, SwiGLU-S^2 activations and smooth-cutoff attention enable accurate modeling of smoothly varying potential energy surfaces (PES), generalizing EquiformerV3 to tasks requiring energy-conserving simulations and higher-order derivatives of PES. With these improvements, EquiformerV3 trained with the auxiliary task of denoising non-equilibrium structures (DeNS) achieves state-of-the-art results on OC20, OMat24, and Matbench Discovery.
MIT CSAIL