Supervised contrastive loss helps uncover more robust features for photoacoustic prostate cancer identification

TL;DR

This paper shows that using supervised contrastive learning improves the accuracy and robustness of photoacoustic prostate cancer diagnosis.

Contribution

The novel SCL-adjust model enhances feature extraction and discrimination accuracy in photoacoustic prostate cancer detection.

Findings

The SCL-adjust model outperforms traditional methods by over 10% in accuracy.

Features from the SCL-adjust model are more resilient to noise and model transfer.

The proposed model improves transfer performance by approximately 5% compared to CNN.

Abstract

Photoacoustic spectral analysis has been demonstrated to be efficacious in the diagnosis of prostate cancer (PCa). With the incorporation of deep learning, its discrimination accuracy is progressively enhancing. Nevertheless, individual heterogeneity persists as a significant factor that impacts discrimination performance.

Extracting more reliable features from intricate biological tissue and augmenting discrimination accuracy of the prostate cancer.

Supervised contrastive learning is introduced to explore its performance in photoacoustic spectral feature extraction. Three distinct models, namely the CNN-based model, the supervised contrastive (SC) model, and the supervised contrastive loss adjust (SCL-adjust) model, have been compared, along with traditional feature extraction and machine learning-based methods.

The outcomes have indicated that the SCL-adjust model exhibits the optimal performance, its accuracy rate has increased by more than 10% compared with the traditional method. Besides, the features extracted from this model are more resilient, regardless of the presence of uniform or Gaussian noise and model transfer. Compared with CNN model, the transfer performance of the proposed model has improved by approximately 5%.

Supervised contrast learning is integrated into photoacoustic spectrum analysis and its effectiveness is verified. A comprehensive analysis is conducted on the performance improvement of the proposed SCL-adjust model in photoacoustic prostate cancer diagnosis, its resistance to noise, and its adaptability to the data heterogeneity of different systems.

Figures