Enhancing Few-Shot Prediction of Ocean Sound Speed Profiles through Hierarchical Long Short-Term Memory Transfer Learning

Jiajun Lu,Hao Zhang,Sijia Li,Pengfei Wu,Wei Huang
DOI: https://doi.org/10.3390/jmse12071041
IF: 2.744
2024-06-22
Journal of Marine Science and Engineering
Abstract:The distribution of ocean sound speed profiles (SSPs) profoundly influences the design of underwater acoustic communication and positioning systems. Conventional methods for measuring sound speed by instruments entail high time costs, while sound speed inversion methods offer rapid estimation of SSPs. However, these methods heavily rely on sonar observational data and lack the capacity to swiftly estimate SSPs in arbitrary oceanic regions, particularly in scenarios with few-shot data. Precisely estimating non-cooperative maritime SSPs under such conditions poses a significant challenge. To explore temporal distribution patterns of sound speed and achieve precise SSP predictions with limited data, we propose a hierarchical long short-term memory transfer learning (H-LSTM-TL) framework. The core idea involves pre-training the base model on extensive public datasets, transferring the acquired knowledge to task models, and fine-tuning the task model on few-shot data to predict future SSPs. Through H-LSTM-TL, it accelerates model convergence, enhances sensitivity to few-shot input data, alleviates overfitting issues, and notably improves the accuracy of SSP predictions. Experimental results demonstrate that the H-LSTM-TL model exhibits strong generalization capabilities in few-shot data scenarios, effectively reducing overfitting problems and proving its applicability for rapid prediction of SSPs.
oceanography,engineering, marine, ocean
What problem does this paper attempt to address?
The paper primarily addresses the problem of predicting Sound Speed Profiles (SSPs) in the ocean, particularly how to improve prediction accuracy under limited data conditions (i.e., few-shot scenarios). The core contributions and technical solutions of the paper can be summarized as follows: 1. **Research Background and Challenges**: - Ocean sound speed profiles are crucial for designing underwater acoustic communication and positioning systems. - Traditional measurement methods are time-consuming and costly, while existing sound speed inversion methods rely on a large amount of sonar observation data and are difficult to quickly estimate the sound speed profile in any given sea area. - Accurately estimating the sound speed profile in non-cooperative sea areas under few-shot conditions is a significant challenge. 2. **Proposed Method**: - The paper proposes a framework based on Hierarchical Long Short-Term Memory Transfer Learning (H-LSTM-TL) to address the above issues. - The core ideas of this framework include: - Pre-training the base model using a large number of public datasets to capture the temporal patterns of sound speed distribution. - Transferring the pre-trained knowledge to the task model and fine-tuning the task model with a few samples to predict future sound speed profiles. - This approach can accelerate model convergence, enhance sensitivity to a small amount of input data, mitigate overfitting issues, and significantly improve the accuracy of sound speed profile predictions. 3. **Technical Details**: - **Hierarchical Processing**: Considering the inconsistency of sound speed changes at different depths, the paper introduces the concept of hierarchical processing to better adapt to the characteristics of data at different depths. - **Application of Transfer Learning**: Using transfer learning to address the overfitting problem under limited data conditions by transferring knowledge from other related tasks to the current task to enhance the model's generalization ability. - **Experimental Validation**: Obtaining a small number of sound speed profile samples through ocean experiments and evaluating the proposed model's accuracy in predicting sound speed profiles, demonstrating the effectiveness and feasibility of the H-LSTM-TL model. In summary, the main goal of this paper is to improve the accuracy of ocean sound speed profile predictions under few-shot conditions by combining hierarchical long short-term memory networks and transfer learning techniques. This approach not only helps address the data scarcity problem in practical applications but also enhances the generalization ability of the prediction model.