Search papers, labs, and topics across Lattice.
This paper introduces a Tri-Branch Modular Fusion Neural Network that integrates three distinct data modalities—3D spatial geometry, discrete topological grammar, and physicochemical descriptors—to enhance molecular property prediction. By employing a shared late-fusion architecture, the model effectively addresses the limitations of traditional graph neural networks in capturing long-range dependencies and oversmoothing issues. Evaluated on the QM9 benchmark for atomization energy, the model achieves a validation Mean Absolute Error of 0.0207 eV, representing a 20.6% improvement over existing geometric baselines while maintaining a parameter-efficient design with fewer than one million parameters.
Integrating 3D geometry, topological grammar, and physicochemical descriptors leads to a 20.6% reduction in prediction error for molecular properties, all while using under one million parameters.
Molecular property prediction often relies on isolated data modalities, where continuous 3D graph neural networks (GNNs) struggle to efficiently capture long-range topological dependencies and exact macroscopic heuristics. In this work, we introduce a parameter-efficient Tri-Branch Modular Fusion Neural Network that synthesizes three orthogonal modalities: 3D spatial geometry (SchNet), discrete topological grammar (SMILES via ChemBERTa), and explicit macroscopic physicochemical descriptors (Deep & Cross Network). By bypassing standard scalar readouts and employing a shared late-fusion architecture, the framework establishes a mathematically rigorous multimodal latent space that effectively resolves the arithmetic and oversmoothing limitations of local message passing. We evaluate the proposed architecture on the QM9 benchmark, targeting the extensive thermodynamic property of atomization energy at 0 K ($U_0^{\mathrm{atom}}$). Through systematic combinatorial ablation and latent bottleneck optimization ($d_e=64$), the tri-modal framework achieves a validation Mean Absolute Error (MAE) of 0.0207 eV. Operating with fewer than one million parameters, this architecture decisively surpasses the sub-chemical accuracy threshold and yields a substantial 20.6% error reduction over a strictly controlled geometric baseline. Ultimately, our findings demonstrate that integrating orthogonal macroscopic and topological data streams provides a synergistic, $\mathcal{O}(1)$ physical shortcut. This multimodal alignment offers a highly efficient alternative to brute-force parameter scaling, establishing a robust surrogate model for high-throughput virtual screening (HTVS) pipelines.
