IEEE Access (Jan 2024)

Two-Dimensional Quantum Material Identification via Self-Attention and Soft-Labeling in Deep Learning

  • Xuan Bac Nguyen,
  • Apoorva Bisht,
  • Benjamin Thompson,
  • Hugh Churchill,
  • Khoa Luu,
  • Samee U. Khan

DOI
https://doi.org/10.1109/ACCESS.2024.3465221
Journal volume & issue
Vol. 12
pp. 139683 – 139691

Abstract

Read online

Detecting two-dimensional (2D) materials in silicon chips presents a significant challenge in the field of quantum machines due to the difficulty of data collection. Specifically, among thousands of flakes, not all flakes are useful or well-annotated, resulting in noisy and hard samples within the dataset, which challenges the deep neural network (DNN) to learn. To address this problem, we propose a novel method for identifying quantum 2D flakes even when there is a high rate of missing annotations in the input images. In particular, we first propose a new mechanism for automatically detecting false negative flakes that are missing annotations. Second, we introduce an attention-based loss function to mitigate the negative impact of these unannotated flakes on the DNNs. The experimental results demonstrate that our method outperforms previous approaches.

Keywords