MaskTerial: A Foundation Model for Automated 2D Material Flake Detection
This work addresses the need for more accurate and efficient sample fabrication in materials science, though it is incremental as it builds on existing instance segmentation methods.
The researchers tackled the problem of automating the detection and classification of exfoliated 2D material flakes from optical microscope images, achieving significant improvements in detecting low-contrast materials like hexagonal boron nitride with a model that adapts quickly using only 5 to 10 images.
The detection and classification of exfoliated two-dimensional (2D) material flakes from optical microscope images can be automated using computer vision algorithms. This has the potential to increase the accuracy and objectivity of classification and the efficiency of sample fabrication, and it allows for large-scale data collection. Existing algorithms often exhibit challenges in identifying low-contrast materials and typically require large amounts of training data. Here, we present a deep learning model, called MaskTerial, that uses an instance segmentation network to reliably identify 2D material flakes. The model is extensively pre-trained using a synthetic data generator, that generates realistic microscopy images from unlabeled data. This results in a model that can to quickly adapt to new materials with as little as 5 to 10 images. Furthermore, an uncertainty estimation model is used to finally classify the predictions based on optical contrast. We evaluate our method on eight different datasets comprising five different 2D materials and demonstrate significant improvements over existing techniques in the detection of low-contrast materials such as hexagonal boron nitride.