Image segmentation of exfoliated two-dimensional materials by generative adversarial network-based data augmentation

Xiaoyu Cheng,Chenxue Xie,Yulun Liu,Ruixue Bai,Nanhai Xiao,Yanbo Ren,Xilin Zhang,Hui Ma,Chongyun Jiang
DOI: https://doi.org/10.1088/1674-1056/ad23d8
2024-01-30
Chinese Physics B
Abstract:Mechanically cleaved two-dimensional materials are random in size and thickness. Recognizing atomically thin flakes by human experts is inefficient and unsuitable for scalable production. Deep learning algorithms have been adopted as an alternative, nevertheless a major challenge is a lack of sufficient actual training images. Here we report the generation of synthetic two-dimensional materials images using StyleGAN3 to complement the dataset. DeepLabv3Plus network is trained with the synthetic images which reduced overfitting and improved recognition accuracy to over 90%. A semi-supervisory technique for labeling images is introduced to reduce manual efforts. The sharper edges recognized by this method facilitates material stacking with precise edge alignment, which benefits exploring novel properties of layered-material devices that crucially depend on the interlayer twist-angle. This feasible and efficient method allows for the rapid and high-quality manufacturing of atomically thin materials and devices.
physics, multidisciplinary
What problem does this paper attempt to address?