TriFNet: A triple-branch feature fusion network for pH determination by surface-enhanced Raman spectroscopy

Abstract

Due to the acidic tumor microenvironment caused by metabolic changes in tumor cells, the accurate pH detection of extracellular fluid is helpful for doctors in precise tumor resection. The combination of Raman spectroscopy and deep learning provides a solution for pH detection. However, most existing studies use one-dimensional convolutional neural networks (1D-CNNs) for spectral analysis, which limits the performance due to insufficient feature extraction. In this work, we propose a 2D triple-branch feature fusion network (TriFNet) for accurate pH determination using surface-enhanced Raman spectra (SERS). Specifically, we design a triple-branch network structure by converting Raman spectra into three types of images to extensively extract complex patterns in spectra. In addition, an attention fusion module, which leverages the complementarity among features in both space and channel, is designed to obtain the valuable information, achieving further accurate pH determination. On our Raman spectral dataset containing 14,137 samples, we achieved mean absolute error (MAE) of 0.059, standard deviation of the absolute error (SD) of 0.07, root mean squared error (RMSE) of 0.092, and coefficient of determination (R2) of 0.991 on the test set. Compared with other published methods, the four metrics showed an average improvement of 47%, 39%, 43%, and 6%, respectively. In addition, visualization validates the diagnostic capability of our model to correlate with biomolecular signatures. Meanwhile, our model has robustness to different SERS chips. These results prove the potential of our method to develop an effective technology based on Raman spectroscopy for accurate pH determination to guide surgery.