Sensor-Oriented Framework for Underwater Acoustic Signal Classification Using EMD–Wavelet Filtering and Bayesian-Optimized Random Forest
Abstract
1. Introduction
- A structured and adaptive pipeline is proposed for ship acoustic signal denoising and classification, ensuring seamless integration of signal decomposition, wavelet-based filtering, feature extraction, and machine learning classification. The pipeline automates IMF selection, wavelet parameter optimization, and classifier hyperparameter tuning, minimizing the need for manual intervention.
- The study employs Signal-to-Noise Ratio (SNR) and Stein’s Unbiased Risk Estimate (SURE) criteria to determine optimal wavelet filtering hyperparameters for each IMF, improving the efficacy of noise reduction while preserving relevant signal components. The impact of different wavelet families (Daubechies, Symlets, Coiflets, and Biorthogonal) is analyzed to determine the best-performing wavelet for ship signal processing.
- A comprehensive feature extraction strategy is implemented, integrating statistical, frequency-domain (FFT-based), and time-frequency (wavelet-based) features to enhance signal characterization. Feature importance analysis identifies the top 11 most relevant features, reducing computational complexity while maintaining high classification accuracy.
2. Methods
- I.
- Signal loading, transforming and decomposition:
- 1.1.
- Transform acoustic signals into numeric ones for the following processing.
- 1.2.
- Implement Empirical Mode Decomposition (EMD) of the raw acoustic signals to obtain Intrinsic Mode Functions (IMFs), which provide a time-adaptive representation of the signal components.
- 1.3.
- Select the first 50% of IMFs for further filtration, as these components typically contain the high-frequency noise components. This proportion was chosen because tests on multiple acoustic signals consistently showed that the early half of the IMFs is sufficient to cover the noise-dominated part, while later IMFs mainly reflect the useful low-frequency structure.
- II.
- Optimization of wavelet filter parameters for each IMFs:
- 2.1.
- Set up the range of wavelet filter hyperparameters variation, including the type of wavelets, the level of wavelet decomposition, and the thresholding coefficient value for processing the detail coefficients.
- 2.2.
- Define the functions used for optimizing the wavelet filter hyperparameters: Signal-to-Noise Ratio (SNR) and Stein’s Unbiased Risk Estimate (SURE).
- Signal-to-Noise Ratio (SNR) represents the ratio of the power of the true signal to the power of the noise
- Stein’s Unbiased Risk Estimate (SURE) provides an estimate of the mean squared error (MSE) of the threshold process and is useful when the noise follows a Gaussian distribution
- 2.3.
- Fix the wavelet decomposition level at 3 and set the threshold coefficient valueHere, the constant 1.4826 ensures consistency with the standard deviation if the noise follows a normal distribution.
- 2.4.
- Determine the optimal wavelet function for each IMF by maximizing the SNR criterion.
- 2.5.
- The second phase of IMF selection involves selecting IMFs with a maximum SNR value less than the median of the distribution of all calculated SNRs.
- 2.6.
- Determine the optimal wavelet decomposition level for each selected IMF by also maximizing the SNR criterion.
- 2.7.
- Determine the optimal threshold for detail coefficient processing by minimizing the SURE criterion, ensuring effective noise reduction.
- III.
- Filtration, reconstruction, normalization, and segmentation of signals.
- 3.1.
- Filtrate the IMFs selected in Step 2.5 using the optimal wavelet filter parameters determined for each respective IMF.
- 3.2.
- Reconstruct the signals using filtered and unfiltered IMFs.
- 3.1.
- Perform normalization and segmentation of the filtered signals to prepare a structured data set with appropriate labels to identify the examined objects.
- IV.
- Feature extraction. Three categories of features are extracted from the segmented signals to maximize classification performance:
- 4.1.
- Statistical characteristics (e.g., mean, standard deviation, kurtosis, skewness).
- 4.2.
- Frequency Domain Features using Fast Fourier Transform (FFT):
- Mean of FFT magnitudes ();
- Standard deviation of FFT magnitudes ();
- Energy of FFT magnitudes ();
- Max. FFT magnitude ();
- Dominant frequency ()
- 4.3.
- Time-Frequency Features Based on Wavelet Transform:
- Wavelet mean and std for each level (, , etc.)
- 4.4.
- Combine all features. Iterate through all segments and extract features. Combine all extracted features into a comprehensive dataset.
- V.
- Classification procedure implementation.
- 5.1.
- Split the dataset into training and testing subsets in a 0.7/0.3 ratio.
- 5.2.
- Optimize the hyperparameters of the Random Forest classifier using the Bayesian optimization method during model training with five-fold cross-validation.
- 5.3.
- Train the model using optimal hyperparameter values.
- 5.4.
- Evaluate the trained model using the test subset. Analyze the obtained results.
- VI.
- Make the final decision regarding the identification of the objects by applying a majority voting method to all segments of a corresponding signal.
2.1. Classification Quality Criteria
- Accuracy (ACC): The overall correctness of classification, defined as
- : The harmonic mean of precision and recall, computed as
2.2. Final Decision on Object Identification Using the Majority Voting Method
3. Results
3.1. Experimental Data
- Signal 1—Acoustic signal from a border patrol vessel.
- Signal 2—Acoustic signal from another border patrol vessel.
- Signal 3—Signal from a hydrographic survey boat.
- Signal 4—Acoustic recording from a military rocket boat.
- Signal 5—Signal from a reconnaissance ship.
3.2. Results of Empirical Mode Decomposition and Signal Filtering
- Versatility in Signal Representation: Orthogonal wavelets, such as Daubechies, symlets, and coiflets, offer excellent time-frequency localization, making them well-suited for analyzing non-stationary signals like ship acoustic signals.
- Compact Support and Vanishing Moments: Daubechies wavelets, particularly higher-order ones (db10–db30), provide compact support with an increasing number of vanishing moments, allowing for effective noise reduction and feature extraction.
- Symmetry and Phase Linearity: Symlets and coiflets are modified versions of Daubechies wavelets designed to enhance symmetry, reducing phase distortion in the reconstructed signals, which is crucial for accurate classification.
- Adaptability to Signal Structure: Biorthogonal wavelets (bior) allow for greater flexibility in balancing decomposition and reconstruction properties, making them advantageous for adaptive filtering and denoising.
- Proven Efficiency in Similar Studies: Prior research has demonstrated the efficiency of these wavelet families in applications involving acoustic signal processing, including empirical mode decomposition-based filtering and classification tasks.
Algorithm 1: Pipeline Filtering Procedure for Acoustic Signal Processing |
3.3. Results of Filtered Signals Classification
Algorithm 2: Classification of Filtered Acoustic Signals Using Optimized Random Forest |
3.4. Validation on Synthetic Signals
4. Discussion and Conclusions
Author Contributions
Funding
Institutional Review Board Statement
Informed Consent Statement
Data Availability Statement
Acknowledgments
Conflicts of Interest
Abbreviations
EMD | Empirical Mode Decomposition |
IMFs | Intrinsic Mode Functions |
SNR | Signal-to-Noise Ratio |
SURE | Stein’s Unbiased Risk Estimate |
SVM | Support Vector Machines |
RF | Random Forest |
CNN | Convolutional Neural Network |
LSTM | Long Short-Term Memory |
2D-ACVMD | two-dimensional Adaptive Chirp Mode Decomposition |
References
- Ahmad, F.; Ansari, M.; Anwar, R.; Shahzad, B.; Ikram, A. Deep Learning Based Classification of Underwater Acoustic Signals. Procedia Comput. Sci. 2024, 235, 1115–1124. [Google Scholar] [CrossRef]
- Lokhande, S.; Malarkodi, A.; Latha, G.; Srinivasan, S. Autonomous detection, localization and tracking of ships by underwater acoustic sensing using vector sensor array. Appl. Ocean. Res. 2025, 154, 104389. [Google Scholar] [CrossRef]
- Li, G.; Zhang, L.; Yang, H. Adaptive denoising model for ship-radiated noise based on dynamic weighted filtering. Appl. Ocean. Res. 2024, 236, 115042. [Google Scholar] [CrossRef]
- Yao, X.; Liu, S.; Chen, J.; Yan, S.; Ji, F.; Liu, H.; Chen, J. Underwater Acoustic Target Classification Using Scattering Transform with Small Sample Size. IEEE Sens. J. 2024, 24, 25998–26010. [Google Scholar] [CrossRef]
- Wu, F.; Yao, H.; Wang, H. Recognizing the State of Motion by Ship-Radiated Noise Using Time-Frequency Swin-Transformer. IEEE J. Ocean. Eng. 2024, 49, 667–678. [Google Scholar] [CrossRef]
- Zhang, Y.; Zeng, Q. MSLEFC: A low-frequency focused underwater acoustic signal classification and analysis system. Eng. Appl. Artif. Intell. 2023, 123, 106333. [Google Scholar] [CrossRef]
- Du, L.; Liu, M.; Lv, Z.; Wang, Z.; Wang, L.; Wang, G. Classification Method for Ship-Radiated Noise Based on Joint Feature Extraction. Commun. Comput. Inf. Sci. 2024, 2036, 75–90. [Google Scholar] [CrossRef]
- Nuruddin, H.; Azizi, M.H.; Mansor, M.N.; Ismail, S.B.; Kamal, I.Z. Forensic Investigation on the Propulsion Train System for Vibration Induced Phenomenon Onboard a Naval Vessel Using Fast Fourier Transform. In Advancement in Emerging Technologies and Engineering Applications; Lecture Notes in Mechanical Engineering; Springer: Singapore, 2020; pp. 25–32. [Google Scholar] [CrossRef]
- Chen, L.; Luo, X.; Zhou, H. A ship-radiated noise classification method based on domain knowledge embedding and attention mechanism. Eng. Appl. Artif. Intell. 2024, 127, 107320. [Google Scholar] [CrossRef]
- Li, D.; Liu, F.; Shen, T.; Chen, L.; Yang, X.; Zhao, D. Generalizable Underwater Acoustic Target Recognition Using Feature Extraction Module of Neural Network. Appl. Sci. 2022, 12, 10804. [Google Scholar] [CrossRef]
- Tang, J.; Gao, W.; Ma, E.; Sun, X.; Ma, J. Deep Learning Based Underwater Acoustic Target Recognition: Introduce a Recent Temporal 2D Modeling Method. Sensors 2024, 24, 1633. [Google Scholar] [CrossRef] [PubMed]
- Zhou, A.; Li, X.; Zhang, W.; Zhao, C.; Ren, K.; Ma, Y.; Song, J. An attention-based multi-scale convolution network for intelligent underwater acoustic signal recognition. Ocean Eng. 2023, 287, 115784. [Google Scholar] [CrossRef]
- Li, G.; Han, Y.; Yang, H. A new underwater acoustic signal denoising method based on modified uniform phase empirical mode decomposition, hierarchical amplitude-aware permutation entropy, and optimized improved wavelet threshold denoising. Ocean Eng. 2024, 293, 116629. [Google Scholar] [CrossRef]
- Jin, S.Y.; Su, Y.; Guo, C.J.; Ma, C.Y.; Fan, Y.X.; Tao, Z.Y. Intrinsic mode ensembled statistical cepstral coefficients for feature extraction of ship-radiated noise. Appl. Acoust. 2025, 227, 110255. [Google Scholar] [CrossRef]
- Li, Y.; Ning, F.; Jiang, X.; Yi, Y. Feature Extraction of Ship Radiation Signals Based on Wavelet Packet Decomposition and Energy Entropy. Math. Probl. Eng. 2022, 2022, 8092706. [Google Scholar] [CrossRef]
- Li, Y.X.; Wang, L. A novel noise reduction technique for underwater acoustic signals based on complete ensemble empirical mode decomposition with adaptive noise, minimum mean square variance criterion and least mean square adaptive filter. Def. Technol. 2020, 16, 543–554. [Google Scholar] [CrossRef]
- Yang, X.; Shao, J.; Yang, S.; Wang, X.; Chen, X. Feature Extraction and Classification Recognition of Ship Radiated Noise Based on 2D-ACVMD. In Proceedings of the 2023 3rd International Symposium on Computer Technology and Information Science, ISCTIS 2023, Chengdu, China, 7–9 July 2023; pp. 619–623. [Google Scholar] [CrossRef]
- Liu, D.; Yang, H.; Hou, W.; Wang, B. A Novel Underwater Acoustic Target Recognition Method Based on MFCC and RACNN. Sensors 2024, 24, 273. [Google Scholar] [CrossRef] [PubMed]
- Hong, F.; Liu, C.; Guo, L.; Chen, F.; Feng, H. Underwater acoustic target recognition with a residual network and the optimized feature extraction method. Appl. Sci. 2021, 11, 1442. [Google Scholar] [CrossRef]
- Wang, B.; Zhang, W.; Zhu, Y.; Wu, C.; Zhang, S. An Underwater Acoustic Target Recognition Method Based on AMNet. IEEE Geosci. Remote Sens. Lett. 2023, 20, 5501105. [Google Scholar] [CrossRef]
- Qi, P.; Yin, G.; Zhang, L. Underwater acoustic target recognition using RCRNN and wavelet-auditory feature. Multimed. Tools Appl. 2024, 83, 47295–47317. [Google Scholar] [CrossRef]
- Xu, W.; Han, X.; Zhao, Y.; Wang, L.; Jia, C.; Feng, S.; Han, J.; Zhang, L. Research on Underwater Acoustic Target Recognition Based on a 3D Fusion Feature Joint Neural Network. J. Mar. Sci. Eng. 2024, 12, 2063. [Google Scholar] [CrossRef]
- Cao, Y.; Yan, J.; Sun, K.; Luo, X. Hydroacoustic Target Detection Based on Improved GFCC and Lightweight Neural Network. In Proceedings of the 2023 42nd Chinese Control Conference (CCC), Tianjin, China, 24–26 July 2024; pp. 6239–6243. [Google Scholar] [CrossRef]
Classifier | Hyperparameter Range | Optimized Value (s) |
---|---|---|
SVM (RBF) | C: 0.1–10, : 0.0001–0.1 | , |
Random Forest | n_estimators: 50–200, max_depth: 1–20, min_samples_split: 2–20, min_samples_leaf: 1–10 | n_estimators = 161, max_depth = 18, min_samples_split = 4, min_samples_leaf = 1 |
Gradient Boosting | n_estimators: 50–200, learning_rate: 0.01–0.2, max_depth: 3–10, min_samples_split: 2–20, min_samples_leaf: 1–10 | n_estimators = 165, learning_rate ≈ 0.145, max_depth = 10, min_samples_split = 5, min_samples_leaf = 1 |
2-layer Neural Network | hidden1: 10–100, hidden2: 10–100 | hidden1 = 98, hidden2 = 17 |
KNN | n_neighbors: 1–30 | n_neighbors = 21 |
Signal | Precision | Recall | F1-Score | Correct Cases | Misclassified Cases | Accuracy, % |
---|---|---|---|---|---|---|
1 | 0.65 | 0.60 | 0.62 | 188 | 126 | 83 |
2 | 0.85 | 0.85 | 0.85 | 362 | 64 | |
3 | 0.86 | 0.82 | 0.84 | 364 | 79 | |
4 | 1.00 | 1.00 | 1.00 | 413 | 0 | |
5 | 0.71 | 0.80 | 0.75 | 280 | 71 |
Signal | Precision | Recall | F1-Score | Correct Cases | Misclassified Cases | Accuracy, % |
---|---|---|---|---|---|---|
1a | 0.83 | 0.79 | 0.81 | 395 | 105 | 87 |
1b | 0.87 | 0.83 | 0.85 | 414 | 86 | |
2a | 0.90 | 0.88 | 0.89 | 439 | 61 | |
2b | 0.91 | 0.90 | 0.91 | 450 | 50 | |
3a | 0.86 | 0.83 | 0.84 | 417 | 83 | |
3b | 0.84 | 0.80 | 0.82 | 400 | 100 | |
4a | 0.97 | 0.96 | 0.96 | 480 | 20 | |
4b | 0.99 | 0.98 | 0.98 | 488 | 12 | |
5a | 0.81 | 0.84 | 0.82 | 420 | 80 | |
5b | 0.86 | 0.85 | 0.85 | 430 | 70 |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2025 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
Share and Cite
Babichev, S.; Yarema, O.; Khomenko, Y.; Senchyshen, D.; Durnyak, B. Sensor-Oriented Framework for Underwater Acoustic Signal Classification Using EMD–Wavelet Filtering and Bayesian-Optimized Random Forest. Sensors 2025, 25, 5336. https://doi.org/10.3390/s25175336
Babichev S, Yarema O, Khomenko Y, Senchyshen D, Durnyak B. Sensor-Oriented Framework for Underwater Acoustic Signal Classification Using EMD–Wavelet Filtering and Bayesian-Optimized Random Forest. Sensors. 2025; 25(17):5336. https://doi.org/10.3390/s25175336
Chicago/Turabian StyleBabichev, Sergii, Oleg Yarema, Yevheniy Khomenko, Denys Senchyshen, and Bohdan Durnyak. 2025. "Sensor-Oriented Framework for Underwater Acoustic Signal Classification Using EMD–Wavelet Filtering and Bayesian-Optimized Random Forest" Sensors 25, no. 17: 5336. https://doi.org/10.3390/s25175336
APA StyleBabichev, S., Yarema, O., Khomenko, Y., Senchyshen, D., & Durnyak, B. (2025). Sensor-Oriented Framework for Underwater Acoustic Signal Classification Using EMD–Wavelet Filtering and Bayesian-Optimized Random Forest. Sensors, 25(17), 5336. https://doi.org/10.3390/s25175336