Progressive Layered Extraction Network Based on Correlation Sharing for Multi-target Prediction of Soil Nutrients
刊名 Agricultural Biotechnology
作者 Tielong SU1, Xuesong TIAN2, Zhengguang CHEN1*
作者单位 1.College of Information and Electrical Engineering, Heilongjiang Bayi Agricultural University, Daqing 163319, China; 2.Daqing Oilfield Shale Oil Exploration and Development Headquarters, Daqing 163319, China
DOI DOI:10.19759/j.cnki.2164-4993.2025.05.007
年份 2025
刊期 5
页码 34-37,41
关键词 Near-infrared spectroscopy; Progressive extraction network; Multi-task learning; Convolutional neural network; Long short-term memory network; Attention mechanism
摘要 With breakthroughs in data processing and pattern recognition through deep learning technologies, the use of advanced algorithmic models for analyzing and interpreting soil spectral information has provided an efficient and economical method for soil quality assessment. However, traditional single-output networks exhibit limitations in the prediction process, particularly in their inability to fully utilize the correlations among various elements. As a result, single-output networks tend to be optimized for a single task, neglecting the interrelationships among different soil elements, which limits prediction accuracy and model generalizability. To overcome this limitation, in this study, a multi-task learning architecture with a progressive extraction network was implemented for the simultaneous prediction of multiple indicators in soil, including nitrogen (N), organic carbon (OC), calcium carbonate (CaCO₃), cation exchange capacity (CEC), and pH. Furthermore, while incorporating the Pearson correlation coefficient, convolutional neural networks, long short-term memory networks and attention mechanisms were combined to extract local abstract features from the original spectra, thereby further improving the model. This architecture is referred to as the Relevance-sharing Progressive Layered Extraction Network. The model employs an adaptive joint loss optimization method to update the weights of individual task losses in the multi-task learning training process.