{"status":"ok","message-type":"work","message-version":"1.0.0","message":{"indexed":{"date-parts":[[2026,7,7]],"date-time":"2026-07-07T13:13:45Z","timestamp":1783430025698,"version":"3.54.6"},"reference-count":41,"publisher":"MDPI AG","issue":"5","license":[{"start":{"date-parts":[[2025,5,17]],"date-time":"2025-05-17T00:00:00Z","timestamp":1747440000000},"content-version":"vor","delay-in-days":0,"URL":"https:\/\/creativecommons.org\/licenses\/by\/4.0\/"}],"content-domain":{"domain":[],"crossmark-restriction":false},"short-container-title":["J. Imaging"],"abstract":"<jats:p>Chemical oxygen demand (COD) serves as a key indicator of organic pollution in water bodies, and its rapid and accurate detection is crucial for environmental protection. Recently, ultraviolet\u2013visible (UV\u2013Vis) spectroscopy has gained popularity for COD detection due to its convenience and the absence of chemical reagents. Meanwhile, deep learning has emerged as an effective approach for automatically extracting spectral features and predicting COD. This paper proposes transforming one-dimensional spectra into two-dimensional spectrum images and employing convolutional neural networks (CNNs) to extract features and model automatically. However, training such deep learning models requires a vast dataset of water samples, alongside the complex task of labeling this data. To address these challenges, we introduce a transfer learning model based on VGG-16 for spectrum images. In this approach, parameters in the initial layers of the model are frozen, while those in the later layers are fine-tuned with the spectrum images. The effectiveness of this method is demonstrated through experiments conducted on our dataset, where the results indicate that it significantly enhances the accuracy of COD prediction compared to traditional methods and other deep learning methods such as partial least squares regression (PLSR), support vector machine (SVM), artificial neural network (ANN), and CNN-based methods.<\/jats:p>","DOI":"10.3390\/jimaging11050159","type":"journal-article","created":{"date-parts":[[2025,5,19]],"date-time":"2025-05-19T03:51:03Z","timestamp":1747626663000},"page":"159","update-policy":"https:\/\/doi.org\/10.3390\/mdpi_crossmark_policy","source":"Crossref","is-referenced-by-count":4,"title":["A Transfer Learning-Based VGG-16 Model for COD Detection in UV\u2013Vis Spectroscopy"],"prefix":"10.3390","volume":"11","author":[{"ORCID":"https:\/\/orcid.org\/0000-0003-4155-2875","authenticated-orcid":false,"given":"Jingwei","family":"Li","sequence":"first","affiliation":[{"name":"College of Electrical, Energy and Power Engineering, Yangzhou University, Yangzhou 225100, China"}],"role":[{"vocabulary":"crossref","role":"author"}]},{"given":"Iqbal Muhammad","family":"Tauqeer","sequence":"additional","affiliation":[{"name":"College of Electrical, Energy and Power Engineering, Yangzhou University, Yangzhou 225100, China"}],"role":[{"vocabulary":"crossref","role":"author"}]},{"ORCID":"https:\/\/orcid.org\/0000-0001-9337-4340","authenticated-orcid":false,"given":"Zhiyu","family":"Shao","sequence":"additional","affiliation":[{"name":"College of Electrical, Energy and Power Engineering, Yangzhou University, Yangzhou 225100, China"}],"role":[{"vocabulary":"crossref","role":"author"}]},{"given":"Haidong","family":"Yu","sequence":"additional","affiliation":[{"name":"College of Electrical, Energy and Power Engineering, Yangzhou University, Yangzhou 225100, China"}],"role":[{"vocabulary":"crossref","role":"author"}]}],"member":"1968","published-online":{"date-parts":[[2025,5,17]]},"reference":[{"key":"ref_1","doi-asserted-by":"crossref","first-page":"117154","DOI":"10.1016\/j.ecoenv.2024.117154","article-title":"A comprehensive risk assessment of microplastics in soil, water, and atmosphere: Implications for human health and environmental safety","volume":"285","author":"Wang","year":"2024","journal-title":"Ecotox. Environ. Safe."},{"key":"ref_2","doi-asserted-by":"crossref","first-page":"1","DOI":"10.1007\/s10653-024-02226-6","article-title":"Assessment of biochemical biomarkers and environmental stress indicators in some freshwater fish","volume":"46","author":"Abdallah","year":"2024","journal-title":"Environ. Geochem. Health"},{"key":"ref_3","doi-asserted-by":"crossref","first-page":"321","DOI":"10.3103\/S1063455X20050045","article-title":"A Machine Learning Approach towards Automatic Water Quality Monitoring","volume":"42","author":"Bansal","year":"2020","journal-title":"J. Water Chem. Technol."},{"key":"ref_4","doi-asserted-by":"crossref","unstructured":"Pansa-Ngat, P., Jedsukontorn, T., and Hunsom, M. (2018). Optimal Hydrogen Production Coupled with Pollutant Removal from Biodiesel Wastewater Using a Thermally Treated TiO2 Photocatalyst (P25): Influence of the Operating Conditions. Nanomaterials, 8.","DOI":"10.3390\/nano8020096"},{"key":"ref_5","doi-asserted-by":"crossref","first-page":"e11095","DOI":"10.1002\/wer.11095","article-title":"Comparative analysis of water quality index and river classification in Kereh River, Penang, Malaysia: Impact of untreated swine wastewater from Kampung Selamat pig farms","volume":"96","author":"Waras","year":"2024","journal-title":"Water Environ. Res."},{"key":"ref_6","doi-asserted-by":"crossref","unstructured":"Nica, A.V., Olaru, E.A., Bradu, C., Dumitru, A., and Avramescu, S.M. (2022). Catalytic Ozonation of Ibuprofen in Aqueous Media over Polyaniline\u2013Derived Nitrogen Containing Carbon Nanostructures. Nanomaterials, 12.","DOI":"10.3390\/nano12193468"},{"key":"ref_7","doi-asserted-by":"crossref","unstructured":"Guo, Y., Liu, C., Ye, R., and Duan, Q. (2020). Advances on Water Quality Detection by UV-Vis Spectroscopy. Appl. Sci., 10.","DOI":"10.3390\/app10196874"},{"key":"ref_8","doi-asserted-by":"crossref","first-page":"19530","DOI":"10.1007\/s11356-021-17056-7","article-title":"Stormwater monitoring using on-line UV-Vis spectroscopy","volume":"29","author":"Huang","year":"2022","journal-title":"Environ. Sci. Pollut. Res."},{"key":"ref_9","doi-asserted-by":"crossref","first-page":"1","DOI":"10.3390\/s20226671","article-title":"Spectrophotometric Online Detection of Drinking Water Disinfectant: A Machine Learning Approach","volume":"20","author":"Hossain","year":"2020","journal-title":"Sensors"},{"key":"ref_10","doi-asserted-by":"crossref","first-page":"1133","DOI":"10.1016\/j.watres.2011.12.005","article-title":"Wastewater quality monitoring system using sensor fusion and machine learning techniques","volume":"46","author":"Qin","year":"2012","journal-title":"Water Res."},{"key":"ref_11","doi-asserted-by":"crossref","first-page":"3799","DOI":"10.1039\/C4AY00198B","article-title":"A rapid determination method for chemical oxygen demand in aquaculture wastewater using the ultraviolet absorbance spectrum and chemometrics","volume":"6","author":"Cao","year":"2014","journal-title":"Anal. Methods"},{"key":"ref_12","doi-asserted-by":"crossref","first-page":"20691","DOI":"10.1039\/C9RA10732K","article-title":"Global calibration model of UV-Vis spectroscopy for COD estimation in the effluent of rural sewage treatment facilities","volume":"10","author":"Li","year":"2020","journal-title":"RSC Adv."},{"key":"ref_13","doi-asserted-by":"crossref","first-page":"1","DOI":"10.1007\/s10661-023-11738-0","article-title":"Development of statistical regression and artificial neural network models for estimating nitrogen, phosphorus, COD, and suspended solid concentrations in eutrophic rivers using UV\u2013Vis spectroscopy","volume":"195","author":"Lyu","year":"2023","journal-title":"Environ. Monit. Assess."},{"key":"ref_14","doi-asserted-by":"crossref","first-page":"1","DOI":"10.3389\/fenvs.2023.1175363","article-title":"Research on COD measurement method based on UV-Vis absorption spectra of transmissive and reflective detection systems","volume":"11","author":"Xu","year":"2023","journal-title":"Front. Environ. Sci."},{"key":"ref_15","doi-asserted-by":"crossref","first-page":"104830","DOI":"10.1016\/j.chemolab.2023.104830","article-title":"Simultaneous measurement of chemical oxygen demand and turbidity in water based on broad optical spectra using backpropagation neural network","volume":"237","author":"Zhou","year":"2023","journal-title":"Chemometr. Intell. Lab. Syst."},{"key":"ref_16","doi-asserted-by":"crossref","first-page":"1","DOI":"10.1080\/15481603.2024.2393489","article-title":"Deep learning-based mapping of total suspended solids in rivers across South Korea using high resolution satellite imagery","volume":"61","author":"Moon","year":"2024","journal-title":"GIsci. Remote Sens."},{"key":"ref_17","doi-asserted-by":"crossref","first-page":"1","DOI":"10.1016\/j.heliyon.2024.e38141","article-title":"Application of mask R-CNN for building detection in UAV remote sensing images","volume":"10","author":"Hou","year":"2024","journal-title":"Heliyon"},{"key":"ref_18","doi-asserted-by":"crossref","first-page":"165","DOI":"10.69793\/ijmcs\/01.2025\/abdulsalam","article-title":"Artificial Intelligence Techniques to Identify Individuals through Palm Image Recognition","volume":"20","author":"Abdulsalam","year":"2025","journal-title":"Int. J. Math. Comput. Sci."},{"key":"ref_19","first-page":"2981","article-title":"Study on the Predication Modeling of COD for Water Based on UV-VIS Spectroscopy and CNN Algorithm of Deep Learning","volume":"40","author":"Jia","year":"2020","journal-title":"Spectrosc. Spect. Anal."},{"key":"ref_20","doi-asserted-by":"crossref","first-page":"1","DOI":"10.3389\/fenvs.2022.1027693","article-title":"Water chemical oxygen demand prediction model based on the CNN and ultraviolet-visible spectroscopy","volume":"10","author":"Ye","year":"2022","journal-title":"Front. Environ. Sci."},{"key":"ref_21","doi-asserted-by":"crossref","unstructured":"Xia, M., Yang, R., Zhao, N., Chen, X., Dong, M., and Chen, J. (2023). A Method of Water COD Retrieval Based on 1D CNN and 2D Gabor Transform for Absorption\u2013Fluorescence Spectra. Micromachines, 14.","DOI":"10.3390\/mi14061128"},{"key":"ref_22","doi-asserted-by":"crossref","first-page":"193","DOI":"10.1039\/D3RA05472A","article-title":"An improved prediction model for COD measurements using UV-Vis spectroscopy","volume":"14","author":"Guan","year":"2024","journal-title":"RSC Adv."},{"key":"ref_23","doi-asserted-by":"crossref","first-page":"238","DOI":"10.21629\/JSEE.2019.02.02","article-title":"Effective distributed convolutional neural network architecture for remote sensing images target classification with a pre-training approach","volume":"30","author":"Li","year":"2019","journal-title":"J. Syst. Eng. Electron."},{"key":"ref_24","doi-asserted-by":"crossref","first-page":"043001","DOI":"10.3788\/COL202321.043001","article-title":"Rapid classification of copper concentrate by portable laser-induced breakdown spectroscopy combined with transfer learning and deep convolutional neural network","volume":"21","author":"Li","year":"2023","journal-title":"Chin. Opt. Lett."},{"key":"ref_25","doi-asserted-by":"crossref","first-page":"1","DOI":"10.1007\/s00530-024-01271-6","article-title":"Target aware network architecture search and compression for efficient knowledge transfer","volume":"30","author":"Basha","year":"2024","journal-title":"Multimed. Syst."},{"key":"ref_26","doi-asserted-by":"crossref","first-page":"2242","DOI":"10.1109\/TDEI.2024.3355031","article-title":"Classification Method of Composite Insulator Surface Image Based on GAN and CNN","volume":"31","author":"Zhou","year":"2024","journal-title":"IEEE Trans. Dielectr. Electr. Insul."},{"key":"ref_27","first-page":"1","article-title":"Evaluation of transfer ensemble learning-based convolutional neural network models for the identification of chronic gingivitis from oral photographs","volume":"24","author":"Li","year":"2024","journal-title":"BMC Oral Health"},{"key":"ref_28","first-page":"799","article-title":"Development of VGG-16 Transfer Learning Framework for Geographical Landmark Recognition","volume":"17","author":"Bansal","year":"2023","journal-title":"Intell. Decis. Technol."},{"key":"ref_29","doi-asserted-by":"crossref","first-page":"1","DOI":"10.1016\/j.engappai.2024.109022","article-title":"Fault diagnosis of monoblock centrifugal pumps using pre-trained deep learning models and scalogram images","volume":"136","author":"Prasshanth","year":"2024","journal-title":"Eng. Appl. Artif. Intel."},{"key":"ref_30","unstructured":"Simonyan, K., and Zisserman, A. (2014). Very deep convolutional networks for large-scale image recognition. arXiv."},{"key":"ref_31","doi-asserted-by":"crossref","first-page":"1","DOI":"10.3390\/math12193032","article-title":"Optimizing Convolutional Neural Network Architectures","volume":"12","author":"Balderas","year":"2024","journal-title":"Mathematics"},{"key":"ref_32","doi-asserted-by":"crossref","first-page":"1","DOI":"10.1038\/s41598-024-56609-x","article-title":"Towards a universal mechanism for successful deep learning","volume":"14","author":"Meir","year":"2024","journal-title":"Sci. Rep."},{"key":"ref_33","doi-asserted-by":"crossref","first-page":"125235","DOI":"10.1016\/j.saa.2024.125235","article-title":"Quantitative detection of hepatocyte mixture based on terahertz time-domain spectroscopy using spectral image analysis methods","volume":"326","author":"Cao","year":"2025","journal-title":"Spectrochim. Acta A"},{"key":"ref_34","doi-asserted-by":"crossref","first-page":"5279","DOI":"10.1007\/s13042-024-02236-x","article-title":"Exploiting graph neural network with one-shot learning for fault diagnosis of rotating machinery","volume":"15","author":"Yang","year":"2024","journal-title":"Int. J. Mach. Learn. Cyb."},{"key":"ref_35","first-page":"1","article-title":"Parkinson\u2019s disease detection and stage classification: Quantitative gait evaluation through variational mode decomposition and DCNN architecture","volume":"36","author":"Balaji","year":"2024","journal-title":"Connect. Sci."},{"key":"ref_36","first-page":"1","article-title":"Detection of heart arrhythmia with electrocardiography. Netw. Model. Anal. Health Inform","volume":"13","author":"Jat","year":"2024","journal-title":"Bioinformatics"},{"key":"ref_37","doi-asserted-by":"crossref","unstructured":"Li, X., Zhang, J., Xiao, B., Zeng, Y., Lv, S., Qian, J., and Du, Z. (2024). Fault Diagnosis of Hydropower Units Based on Gramian Angular Summation Field and Parallel CNN. Energies, 17.","DOI":"10.3390\/en17133084"},{"key":"ref_38","doi-asserted-by":"crossref","first-page":"3797","DOI":"10.1007\/s00431-024-05646-9","article-title":"Optimization and evaluation of facial recognition models for Williams-Beuren syndrome","volume":"183","author":"Huang","year":"2024","journal-title":"Eur. J. Pediatr."},{"key":"ref_39","first-page":"133","article-title":"Leveraging Transfer Learning for Efficient Diagnosis of COPD Using CXR Images and Explainable AI Techniques","volume":"24","author":"Ikechukwu","year":"2024","journal-title":"Intel. Artific."},{"key":"ref_40","doi-asserted-by":"crossref","first-page":"1","DOI":"10.1186\/s00015-024-00458-3","article-title":"Direct mineral content prediction from drill core images via transfer learning","volume":"117","author":"Boiger","year":"2024","journal-title":"Swiss. J. Geosci."},{"key":"ref_41","doi-asserted-by":"crossref","first-page":"319","DOI":"10.1007\/s10661-018-6702-7","article-title":"Use of ultraviolet\u2013visible spectrophotometry associated with artificial neural networks as an alternative for determining the water quality index. Environ","volume":"190","author":"Alves","year":"2018","journal-title":"Monit. Assess."}],"container-title":["Journal of Imaging"],"original-title":[],"language":"en","link":[{"URL":"https:\/\/www.mdpi.com\/2313-433X\/11\/5\/159\/pdf","content-type":"unspecified","content-version":"vor","intended-application":"similarity-checking"}],"deposited":{"date-parts":[[2025,10,9]],"date-time":"2025-10-09T17:34:20Z","timestamp":1760031260000},"score":1,"resource":{"primary":{"URL":"https:\/\/www.mdpi.com\/2313-433X\/11\/5\/159"}},"subtitle":[],"short-title":[],"issued":{"date-parts":[[2025,5,17]]},"references-count":41,"journal-issue":{"issue":"5","published-online":{"date-parts":[[2025,5]]}},"alternative-id":["jimaging11050159"],"URL":"https:\/\/doi.org\/10.3390\/jimaging11050159","relation":{},"ISSN":["2313-433X"],"issn-type":[{"value":"2313-433X","type":"electronic"}],"subject":[],"published":{"date-parts":[[2025,5,17]]}}}