mlsanityData to 14 Oct 2026

MolFM-Lite: Multi-Modal Molecular Property Prediction with Conformer Ensemble Attention and Cross-Modal Fusion

arXiv 2602.22405 ↗First posted 25 Feb 2026cs.LG cs.CVcited by 0

Not on the map. Why?

Abstract (arXiv)

Most machine learning models for molecular property prediction rely on a single molecular representation (either a sequence, a graph, or a 3D structure) and treat molecular geometry as static. We present MolFM-Lite, a multi-modal model that jointly encodes SELFIES sequences (1D), molecular graphs (2D), and conformer ensembles (3D) through cross-attention fusion, while conditioning predictions on experimental context via Feature-wise Linear Modulation (FiLM). Our main methodological contributions are: (1) a conformer ensemble attention mechanism that combines learnable attention with Boltzmann-weighted priors over multiple RDKit-generated conformers, capturing the thermodynamic distribution of molecular shapes; and (2) a cross-modal fusion layer where each modality can attend to others, enabling complementary information sharing. We evaluate on four MoleculeNet scaffold-split benchmarks using our model's own splits, and report all baselines re-evaluated under the same protocol. Comprehensive ablation studies across all four datasets confirm that each architectural component contributes independently, with tri-modal fusion providing 7-11% AUC improvement over single-modality baselines and conformer ensembles adding approximately 2% over single-conformer variants. Pre-training on ZINC250K (~250K molecules) using cross-modal contrastive and masked-atom objectives enables effective weight initialization at modest compute cost. We release all code, trained models, and data splits to support reproducibility.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 18 of its 35 references that are in the map. First posted on arXiv 25 Feb 2026.

Builds on

18 of 35 references.

References · 18

  1. 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,361
  2. 2017Graph Attention Networkscited by 8,923
  3. 2018Representation Learning with Contrastive Predictive Codingcited by 7,777
  4. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,448
  5. 2018How Powerful are Graph Neural Networks?cited by 4,393
  6. 2017FiLM: Visual Reasoning with a General Conditioning Layercited by 2,333
  7. 2017MoleculeNet: A Benchmark for Molecular Machine Learningcited by 1,051
  8. 2018MINE: Mutual Information Neural Estimationcited by 683
  9. 2017SchNet: A continuous-filter convolutional neural network for modeling quantum interactionscited by 571
  10. 2020Directional Message Passing for Molecular Graphscited by 561
  11. 2020ChemBERTa: Large-Scale Self-Supervised Pretraining for Molecular Property Predictioncited by 337
  12. 2021Do Transformers Really Perform Bad for Graph Representation?cited by 320
6 more
  1. 2020GEOM: Energy-annotated molecular conformations for property prediction and molecular generationcited by 243
  2. 2021ChemRL-GEM: Geometry Enhanced Molecular Representation Learning for Property Predictioncited by 204
  3. 2022Multi-modal Molecule Structure-text Model for Text-based Retrieval and Editingcited by 166
  4. 20213D Infomax improves GNNs for Molecular Property Predictioncited by 146
  5. 2020Molecular representation learning with language models and domain-relevant auxiliary taskscited by 94
  6. 2022GPS++: An Optimised Hybrid MPNN/Transformer for Molecular Property Predictioncited by 28

Built on by

Not cited yet.