MoMa: A Modular Deep Learning Framework for Material Property Prediction
About
Deep learning methods for material property prediction have been widely explored to advance materials discovery. However, the prevailing pre-train then fine-tune paradigm often fails to address the inherent diversity and disparity of material tasks. To overcome these challenges, we introduce MoMa, a Modular framework for Materials that first trains specialized modules across a wide range of tasks and then adaptively composes synergistic modules tailored to each downstream scenario. Evaluation across 17 datasets demonstrates the superiority of MoMa, with a substantial 14% average improvement over the strongest baseline. Few-shot and continual learning experiments further highlight MoMa's potential for real-world applications. Pioneering a new paradigm of modular material learning, MoMa will be open-sourced to foster broader community collaboration.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Material Property Prediction | Matminer Experimental Band Gap (eV) 5-split average | MAE0.305 | 7 | |
| Material Property Prediction | Matminer Formation Enthalpy (eV/atom) (5-split average) | MAE (eV/atom)0.0839 | 7 | |
| Material Property Prediction | Matminer 2D Dielectric Constant 5-split average | MAE1.89 | 7 | |
| Material Property Prediction | Matminer 2D Formation Energy (eV/atom) (5-split average) | MAE (eV/atom)0.0495 | 7 | |
| Material Property Prediction | Matminer 2D Band Gap (eV) (5-split average) | MAE (eV)0.375 | 7 | |
| Material Property Prediction | Matminer 3D Poly Electronic 5-split average | MAE23 | 7 | |
| Material Property Prediction | Matminer 3D Band Gap (eV) 5-split average | MAE (eV)0.2 | 7 | |
| Material Property Prediction | Matminer Refractive Index 5-split average | MAE0.0523 | 7 | |
| Material Property Prediction | Matminer Electronic Dielectric Constant 5-split average | MAE0.0885 | 7 | |
| Material Property Prediction | Matminer Dielectric Constant 5-split average | MAE0.158 | 7 |