Universal NIR-II fluorescence image enhancement via covariance weighted attention network

Tian, Jie
DOI: https://doi.org/10.1007/s00530-024-01533-3
IF: 3.9
2024-11-07
Multimedia Systems
Abstract:The second near-infrared (NIR-II) fluorescence imaging has become a new imaging mode due to its characteristics of real-time intraoperative imaging. The NIR-IIb window (1500–1700 nm) has stronger light penetration and has a clearer imaging effect than the NIR-IIa window (1000–1300 nm). The molecular probe currently used for human imaging (indocyanine green) can only be used for NIR-IIa window imaging, and there is a lack of effective molecular probes for NIR-IIb imaging. In addition, there are various types of NIR-II fluorescence images, and it is difficult for a single neural network model to enhance multiple types of images. To solve these problems, we design a universal NIR-II fluorescence image enhancement model to transfer the style of NIR-IIb images to NIR-IIa images and improve the quality of fluorescence images. The model is based on an encoder-decoder framework and can process multiple types of NIR-II fluorescence images simultaneously. Specifically, it includes a module for fusion of low-quality and high-quality image feature maps, the covariance weighted attention network (CWANet), which improves the attention mechanism through covariance weight, allowing the model to filter the style features that are irrelevant or detrimental to image structure in the attention mechanism and pay attention to the key content details of the image. Furthermore, we propose a multiple high-frequency matching loss, which matches the high-frequency information of the enhanced image with the NIR-IIa image to further maintain the structure. Extensive experiments demonstrate that our model generates results that exceed state-of-the-art models, with impressive image enhancement effect.
computer science, information systems, theory & methods
What problem does this paper attempt to address?