Vol. 54, Issue 3, pp. 383-394 (2024)

Vol. 54 Issue 3 pp. 383-394

Reconstructing computational spectra using deep learning’s self-attention method

Hao Wu, Hui Wu, Xinyu Su, Jingjun Wu, Shuangli Liu

Keywords

spectral reconstruction, self-attention, encoding matrix, cross-correlation

Abstract

Miniaturized computational spectrometers have become a new research hotspot due to their portability and miniaturization. However, there are several issues, like low precision and poor stability. Because the problem of spectrum reconstruction accuracy is very evident, we suggested a novel approach to raise the reconstruction accuracy. A library of optical filtering functions was acquired using the time-domain finite-difference (FDTD) method. A cross-correlation algorithm was then used to choose 100 sparse filter functions, which were then built as an encoding matrix and then, based on the encoding matrix, a self-attention mechanism algorithm to improve the accuracy. The reconstructed spectrum’s mean square error (MSE) is 0.0019, and its similarity coefficient (R2) is 0.9780. This self-attention mechanism spectral reconstruction technique will open up new possibilities for high-accuracy reconstruction for various computational spectrometer types.

Vol. 54
Issue 3
pp. 383-394

Corresponding address

Optica Applicata
Wrocław University of Science and Technology
Faculty of Fundamental Problems of Technology
Wybrzeże Wyspiańskiego 27
50-370 Wrocław, Poland

Publisher

Wrocław University of Science and Technology
Faculty of Fundamental Problems of Technology
Wybrzeże Wyspiańskiego 27
50-370 Wrocław, Poland

Contact us

  • optica.applicata@pwr.edu.pl
  • +48 71 320 23 93
  • +48 71 328 36 96