UniMate: A Unified Model for Mechanical Metamaterial Generation, Property Prediction, and Condition Confirmation
This work addresses a gap in machine learning for mechanical metamaterial design by enabling comprehensive handling of all three key modalities, which is incremental but important for real-world applications.
The paper tackles the problem of mechanical metamaterial design by proposing UniMate, a unified model that integrates 3D topology generation, property prediction, and condition confirmation, achieving performance improvements of up to 80.2%, 5.1%, and 50.2% over baselines in these tasks.
Metamaterials are artificial materials that are designed to meet unseen properties in nature, such as ultra-stiffness and negative materials indices. In mechanical metamaterial design, three key modalities are typically involved, i.e., 3D topology, density condition, and mechanical property. Real-world complex application scenarios place the demanding requirements on machine learning models to consider all three modalities together. However, a comprehensive literature review indicates that most existing works only consider two modalities, e.g., predicting mechanical properties given the 3D topology or generating 3D topology given the required properties. Therefore, there is still a significant gap for the state-of-the-art machine learning models capturing the whole. Hence, we propose a unified model named UNIMATE, which consists of a modality alignment module and a synergetic diffusion generation module. Experiments indicate that UNIMATE outperforms the other baseline models in topology generation task, property prediction task, and condition confirmation task by up to 80.2%, 5.1%, and 50.2%, respectively. We opensource our proposed UNIMATE model and corresponding results at https://github.com/wzhan24/UniMate.