Large scale and diverse two-dimensional flake segmentation dataset by general-purpose and labor-efficient annotation framework
Source abstract
Abstract
Identifying 2D material flakes under a microscope is a crucial step in mechanical exfoliation. Although automated detection algorithms reduce manual search time, building generalizable foundational models still depends on large, diverse, high-quality, real-world datasets. However, manually creating such annotated datasets is laborious and time-consuming. We present the most comprehensive dataset to date for 2D flake segmentation, comprising 7454 microscopic images with approximately 30,000 annotated flake regions. It systematically captures variations in imaging conditions and flake dimensions, with size distributions spanning four orders of magnitude. To reduce labeling effort, we introduce a self-evolving annotation ecosystem that integrates active learning, semi-supervised learning, and foundation models within a batch-wise architecture. Additionally, we propose a region-aware evaluation framework that quantifies model performance via connected-region analysis, simultaneously assessing segmentation accuracy and flake count without bias. Our resources and methodology aim to improve 2D material detection and provide a practical, cost-effective framework for adaptable annotation systems.
Source: openalex_inverted_index · Quality: baseline
Research context
OpenAlex topics: Force Microscopy Techniques and Applications · Machine Learning in Materials Science · Advanced Sensor and Energy Harvesting Materials
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
References (62)
Show source-provided references
1. Das, M. et al. High-performance p-type field-effect transistors using substitutional doping and thickness control of two-dimensional materials. Nat. Electron. 8, 24–35 (2025).10.1038/s41928-024-01265-2 ↗
2. Pannone, A. et al. Robust chemical analysis with graphene chemosensors and machine learning. Nature 634, 572–578 (2024).10.1038/s41586-024-08003-w ↗
3. Wang, J. et al. Two dimensional MoS2 accelerates mechanically controlled polymerization and remodeling of hydrogel. Nat. Commun. 16, 1689 (2025).10.1038/s41467-025-57068-2 ↗
4. Yu, Y. et al. Hot-carrier engineering for two-dimensional integrated infrared optoelectronics. InfoMat 6, e12556 (2024).10.1002/inf2.12556 ↗
5. Chen, J. et al. Spin-enhanced self-powered light helicity detecting based on vertical WSe2–CrI3 p–n heterojunction. ACS Nano 18, 26261–26270 (2024).
6. Kang, J.-H. et al. Monolithic 3D integration of 2D materials-based electronics towards ultimate edge computing solutions. Nat. Mater. 22, 1470–1477 (2023).10.1038/s41563-023-01704-z ↗
7. Yan, S. et al. Dicarbon nitride and janus transition metal chalcogenides van der waals heterojunctions for photocatalytic water splitting. J. Phys.: Condens. Matter 35, 014003 (2022).
8. Khan, A. I. et al. Surface conduction and reduced electrical resistivity in ultrathin noncrystalline NbP semimetal. Science 387, 62–67 (2025).10.1126/science.adq7096 ↗
9. Xu, H. et al. A multifunctional optoelectronic device based on 2D material with wide bandgap. Light Sci. Appl. 12, 278 (2023).10.1038/s41377-023-01327-8 ↗
10. Das, S. et al. Transistors based on two-dimensional materials for future integrated circuits. Nat. Electron. 4, 786–799 (2021).10.1038/s41928-021-00670-1 ↗
11. Wang, F. et al. Lateral heterostructure formed by highly thermally conductive fluorinated graphene for efficient device thermal management. Adv. Sci. 11, 2401586 (2024).10.1002/advs.202401586 ↗
12. Yao, Z. et al. Transferrable, wet-chemistry-derived high-k amorphous metal oxide dielectrics for two-dimensional electronic devices. Nat. Commun. 16, 1482 (2025).10.1038/s41467-025-56815-9 ↗
13. Zhang, X. & Peng, B. The twisted two-dimensional ferroelectrics. J. Semiconductors 44, 11002 (2023).10.1088/1674-4926/44/1/011002 ↗
14. Jayachandran, D. et al. Three-dimensional integration of two-dimensional field-effect transistors. Nature 625, 276–281 (2024).10.1038/s41586-023-06860-5 ↗
15. Pendurthi, R. et al. Monolithic three-dimensional integration of complementary two-dimensional field-effect transistors. Nat. Nanotechnol. 19, 970–977 (2024).10.1038/s41565-024-01705-2 ↗
16. Ghosh, S. et al. Monolithic and heterogeneous three-dimensional integration of two-dimensional materials with high-density vias. Nat. Electron. 7, 892–903 (2024).10.1038/s41928-024-01251-8 ↗
17. Wang, H. et al. Recent advances in two-dimensional photovoltaic devices. J. Semiconductors 45, 51701 (2024).10.1088/1674-4926/45/5/051701 ↗
18. Wang, J. et al. Polymer-like tetramer acceptor enables stable and 19.75% efficiency binary organic solar cells. Nat. Commun. 16, 1784 (2025).10.1038/s41467-025-57118-9 ↗
19. Wang, G. et al. 27.09%-efficiency silicon heterojunction back contact solar cell and going beyond. Nat. Commun. 15, 8931 (2024).10.1038/s41467-024-53275-5 ↗
20. Chhowalla, M. et al. The chemistry of two-dimensional layered transition metal dichalcogenide nanosheets. Nat. Chem. 5, 263–275 (2013).10.1038/nchem.1589 ↗
21. Zhang, L., Tan, P.-H. & Wu, J. Reconfigurable devices based on two-dimensional materials for logic and analog applications. J. Semiconductors 46, 71701 (2025).10.1088/1674-4926/24100005 ↗
22. Li, H., Ling, J., Lin, J., Lu, X. & Xu, W. Interface engineering in two-dimensional heterostructures towards novel emitters. J. Semiconductors 44, 011001 (2023).10.1088/1674-4926/44/1/011001 ↗
23. Nakamoto, T. et al. Selective isolation of mono- to quadlayered 2D materials via sonication-assisted micromechanical exfoliation. ACS Nano 18, 2455–2463 (2024).10.1021/acsnano.3c11099 ↗
24. Huang, Y. et al. Universal mechanical exfoliation of large-area 2D crystals. Nat. Commun. 11, 2453 (2020).10.1038/s41467-020-16266-w ↗
25. Islam, M. A. et al. Exfoliation mechanisms of 2D materials and their applications. Appl. Phys. Rev. 9, 041301 (2022).10.1063/5.0090717 ↗
26. Ramezani, F. et al. Automatic detection of multilayer hexagonal boron nitride in optical images using deep learning-based computer vision. Sci. Rep. 13, 1595 (2023).10.1038/s41598-023-28664-3 ↗
27. Sanchez-Juarez, J., Granados-Baez, M., Aguilar-Lasserre, A. A. & Cardenas, J. Automated system for the detection of 2D materials using digital image processing and deep learning. Opt. Mater. Expr. 12, 1856–1868 (2022).10.1364/OME.454314 ↗
28. Masubuchi, S. et al. Deep-learning-based image segmentation integrated with optical microscopy for automatically searching for two-dimensional materials. npj 2D Mater. Appl. 4, 3 (2020).10.1038/s41699-020-0137-z ↗
29. Dong, X. et al. Deep-learning-based microscopic imagery classification, segmentation, and detection for the identification of 2D semiconductors. Adv. Theory Simul. 5, 2200140 (2022).10.1002/adts.202200140 ↗
30. Saito, Y. et al. Deep-learning-based quality filtering of mechanically exfoliated 2D crystals. npj Comput. Mater. 5, 124 (2019).10.1038/s41524-019-0262-4 ↗
31. Yang, H. et al. Identification and structural characterization of twisted atomically thin bilayer materials by deep learning. Nano Lett. 24, 2789–2797 (2024).10.1021/acs.nanolett.3c04815 ↗
32. Zhang, Y., Zhang, H., Zhou, S., Liu, G. & Zhu, J. Deep learning-based layer identification of 2D nanomaterials. Coatings 12, 1551 (2022).10.3390/coatings12101551 ↗
33. Han, B. et al. Deep-learning-enabled fast optical identification and characterization of 2D materials. Adv. Mater. 32, 2000953 (2020).10.1002/adma.202000953 ↗
34. Dong, X. et al. 3D deep learning enables accurate layer mapping of 2D materials. ACS Nano 15, 3139–3151 (2021).10.1021/acsnano.0c09685 ↗
35. Uslu, J.-L. et al. An open-source robust machine learning platform for real-time detection and classification of 2D material flakes. Mach. Learn.: Sci. Technol. 5, 015027 (2024).
36. Masubuchi, S. & Machida, T. Classifying optical microscope images of exfoliated graphene flakes by data-driven machine learning. npj 2D Mater. Appl. 3, 4 (2019).10.1038/s41699-018-0084-0 ↗
37. Sterbentz, R. M., Haley, K. L. & Island, J. O. Universal image segmentation for optical identification of 2D materials. Sci. Rep. 11, 5808 (2021).10.1038/s41598-021-85159-9 ↗
38. Lin, X. et al. Intelligent identification of two-dimensional nanostructures by machine-learning optical microscopy. Nano Res. 11, 6316–6324 (2018).10.1007/s12274-018-2155-0 ↗
39. Yang, J. & Yao, H. Automated identification and characterization of two-dimensional materials via machine learning-based processing of optical microscope images. Extrem. Mech. Lett. 39, 100771 (2020).10.1016/j.eml.2020.100771 ↗
40. Li, Y. et al. Rapid identification of two-dimensional materials via machine learning assisted optic microscopy. J. Materiomics 5, 413–421 (2019).10.1016/j.jmat.2019.03.003 ↗
41. Masubuchi, S. et al. Autonomous robotic searching and assembly of two-dimensional crystals to build van der waals superlattices. Nat. Commun. 9, 1413 (2018).10.1038/s41467-018-03723-w ↗
42. Kirillov, A. et al. Segment anything. In 2023 IEEE/CVF International Conference on Computer Vision (ICCV), 3992–4003 (2023).10.1109/ICCV51070.2023.00371 ↗
43. Li, X. et al. Deep active learning with noise stability. Proc. AAAI Conf. Artif. Intell. 38, 13655–13663 (2024).
44. Kim, H., Oh, M., Hwang, S., Kwak, S. & Ok, J. Adaptive superpixel for active learning in semantic segmentation. In 2023 IEEE/CVF International Conference on Computer Vision (ICCV), 943–953 (2023).10.1109/ICCV51070.2023.00093 ↗
45. Xie, B., Yuan, L., Li, S., Liu, C. H. & Cheng, X. Towards fewer annotations: Active learning via region impurity and prediction uncertainty for domain adaptive semantic segmentation. In 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 8058–8068 (2022).10.1109/CVPR52688.2022.00790 ↗
46. Geuenich, M. J., Gong, D.-w & Campbell, K. R. The impacts of active and self-supervised learning on efficient annotation of single-cell expression data. Nat. Commun. 15, 1014 (2024).10.1038/s41467-024-45198-y ↗
47. Zhang, W. et al. BoostMIS: Boosting medical image semi-supervised learning with adaptive pseudo labeling and informative active annotation. In 2022 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 20634–20644 (2022).10.1109/CVPR52688.2022.02001 ↗
48. Sun, Y. et al. Self-supervised learning with application for infant cerebellum segmentation and analysis. Nat. Commun. 14, 4717 (2023).10.1038/s41467-023-40446-z ↗
49. Wang, X., Wang, W., Cao, Y., Shen, C. & Huang, T. Images speak in images: A generalist painter for in-context visual learning. In 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 6830–6839 (2023).10.1109/CVPR52729.2023.00660 ↗
50. Liang, F. et al. Open-vocabulary semantic segmentation with mask-adapted CLIP. In 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 7061–7070 (2023).10.1109/CVPR52729.2023.00682 ↗
51. Wang, Z. et al. Revisiting evaluation metrics for semantic segmentation: Optimization and evaluation of fine-grained intersection over union. Adv. Neural Inf. Process. Syst. 36, 60144–60225 (2023).
52. Maier-Hein, L. et al. Metrics reloaded: Recommendations for image analysis validation. Nat. Methods 21, 195–212 (2024).10.1038/s41592-023-02151-z ↗
53. Sun, W. et al. Deep edge enhancement-based semantic segmentation network for farmland segmentation with satellite imagery. Comput. Electron. Agric. 202, 107273 (2022).10.1016/j.compag.2022.107273 ↗
54. Zhang, Z., Cai, H. & Han, S. EfficientViT-SAM: Accelerated segment anything model without performance loss. In 2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), 7859–7863 (2024).10.1109/CVPRW63382.2024.00782 ↗
55. Liu, P., Wang, L., Ranjan, R., He, G. & Zhao, L. A survey on active deep learning: From model driven to data driven. ACM Comput. Surv. 54, 221:1–221:34 (2022).10.1145/3510414 ↗
56. Gaillochet, M., Desrosiers, C. & Lombaert, H. Active learning for medical image segmentation with stochastic batches. Med. Image Anal. 90, 102958 (2023).10.1016/j.media.2023.102958 ↗
57. Wang, D. & Shang, Y. A new active labeling method for deep learning. In 2014 International Joint Conference on Neural Networks (IJCNN), 112–119 (2014).10.1109/IJCNN.2014.6889457 ↗
58. Xie, B. et al. Active learning for domain adaptation: An energy-based approach. Proc. AAAI Conf. Artif. Intell. 36, 8708–8716 (2022).
59. Xiong, Y. et al. Efficient deformable ConvNets: Rethinking dynamic and sparse operator for vision applications. In 2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 5652–5661 (2024).10.1109/CVPR52733.2024.00540 ↗
60. Ding, X. et al. UniRepLKNet: A universal perception large-kernel ConvNet for audio, video, point cloud, time-series and image recognition. In 2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 5513–5524 (2024).10.1109/CVPR52733.2024.00527 ↗
61. Zhu, L., Wang, X., Ke, Z., Zhang, W. & Lau, R. BiFormer: Vision transformer with bi-level routing attention. In 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 10323–10333 (2023).10.1109/CVPR52729.2023.00995 ↗
62. Xia, C., Wang, X., Lv, F., Hao, X. & Shi, Y. ViT-CoMer: Vision transformer with convolutional multi-scale feature interaction for dense predictions. In 2024 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 5493–5502 (2024).10.1109/CVPR52733.2024.00525 ↗