A Novel Feature Extraction Method Based on Legendre Multi-Wavelet Transform and Auto-Encoder for Steel Surface Defect Classification
Effective steel surface defect classification with low computational cost is essential for online quality inspection. The challenge of this task is large intra-class differences and unclear inter-class distances shown in various surface defects. This paper proposes a novel feature extraction method...
| Main Authors: | , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
IEEE
2024-01-01
|
| Series: | IEEE Access |
| Subjects: | |
| Online Access: | https://ieeexplore.ieee.org/document/10380552/ |
| _version_ | 1797357069286768640 |
|---|---|
| author | Xiaoyang Zheng Weishuo Liu Yan Huang |
| author_facet | Xiaoyang Zheng Weishuo Liu Yan Huang |
| author_sort | Xiaoyang Zheng |
| collection | DOAJ |
| description | Effective steel surface defect classification with low computational cost is essential for online quality inspection. The challenge of this task is large intra-class differences and unclear inter-class distances shown in various surface defects. This paper proposes a novel feature extraction method based on Legendre multi-wavelet transform and Auto-Encoder network (LWT-AE) for effectively recognizing the steel surface defects categories. More precisely, the finite element approximation theory is implemented to address the strong capacity of LW bases with various regularities and vanish moments for thoroughly matching the complex geometry characteristics of the steel surface defects, which provides strict theory foundation for the feature extraction in LWT frequency domain without losing any defect information due to its orthogonality. Then, the statistical and texture parameters are utilized to sparsely extract the defect features from LWT frequency domain, resulting in removing the redundancy components corresponding to the defects. AE network is utilized to further reduce the dimension of the extracted features and automatically select the most valuable defect features. Furthermore, two classifiers (SVM and BPNN) are used to rectify the generalization ability of the proposed method. Finally, extensive experiments are conducted on two datasets to verify the effectiveness and robustness of the proposed method. The highest classification accuracies 99.44% and 95.37% compared with other methods are attained on the NEU-CLS dataset and X-SDD dataset, respectively. To summary, the proposed method not only has simple structure, but also can reliably identify different types of the steel surface defects, which provides a suitable online application technique for the actual steel surface defect classification. |
| first_indexed | 2024-03-08T14:37:57Z |
| format | Article |
| id | doaj.art-230abec7294741c08a598453e2bc1b7a |
| institution | Directory Open Access Journal |
| issn | 2169-3536 |
| language | English |
| last_indexed | 2024-03-08T14:37:57Z |
| publishDate | 2024-01-01 |
| publisher | IEEE |
| record_format | Article |
| series | IEEE Access |
| spelling | doaj.art-230abec7294741c08a598453e2bc1b7a2024-01-12T00:02:23ZengIEEEIEEE Access2169-35362024-01-01125092510210.1109/ACCESS.2024.334962810380552A Novel Feature Extraction Method Based on Legendre Multi-Wavelet Transform and Auto-Encoder for Steel Surface Defect ClassificationXiaoyang Zheng0https://orcid.org/0000-0002-1719-4169Weishuo Liu1https://orcid.org/0000-0001-9257-0817Yan Huang2https://orcid.org/0009-0006-7578-5019School of Artificial Intelligence, Chongqing University of Technology, Chongqing, ChinaSchool of Artificial Intelligence, Chongqing University of Technology, Chongqing, ChinaSchool of Artificial Intelligence, Chongqing University of Technology, Chongqing, ChinaEffective steel surface defect classification with low computational cost is essential for online quality inspection. The challenge of this task is large intra-class differences and unclear inter-class distances shown in various surface defects. This paper proposes a novel feature extraction method based on Legendre multi-wavelet transform and Auto-Encoder network (LWT-AE) for effectively recognizing the steel surface defects categories. More precisely, the finite element approximation theory is implemented to address the strong capacity of LW bases with various regularities and vanish moments for thoroughly matching the complex geometry characteristics of the steel surface defects, which provides strict theory foundation for the feature extraction in LWT frequency domain without losing any defect information due to its orthogonality. Then, the statistical and texture parameters are utilized to sparsely extract the defect features from LWT frequency domain, resulting in removing the redundancy components corresponding to the defects. AE network is utilized to further reduce the dimension of the extracted features and automatically select the most valuable defect features. Furthermore, two classifiers (SVM and BPNN) are used to rectify the generalization ability of the proposed method. Finally, extensive experiments are conducted on two datasets to verify the effectiveness and robustness of the proposed method. The highest classification accuracies 99.44% and 95.37% compared with other methods are attained on the NEU-CLS dataset and X-SDD dataset, respectively. To summary, the proposed method not only has simple structure, but also can reliably identify different types of the steel surface defects, which provides a suitable online application technique for the actual steel surface defect classification.https://ieeexplore.ieee.org/document/10380552/Classificationsteel surface defectsLegendre multi-wavelet transformauto-encoder |
| spellingShingle | Xiaoyang Zheng Weishuo Liu Yan Huang A Novel Feature Extraction Method Based on Legendre Multi-Wavelet Transform and Auto-Encoder for Steel Surface Defect Classification IEEE Access Classification steel surface defects Legendre multi-wavelet transform auto-encoder |
| title | A Novel Feature Extraction Method Based on Legendre Multi-Wavelet Transform and Auto-Encoder for Steel Surface Defect Classification |
| title_full | A Novel Feature Extraction Method Based on Legendre Multi-Wavelet Transform and Auto-Encoder for Steel Surface Defect Classification |
| title_fullStr | A Novel Feature Extraction Method Based on Legendre Multi-Wavelet Transform and Auto-Encoder for Steel Surface Defect Classification |
| title_full_unstemmed | A Novel Feature Extraction Method Based on Legendre Multi-Wavelet Transform and Auto-Encoder for Steel Surface Defect Classification |
| title_short | A Novel Feature Extraction Method Based on Legendre Multi-Wavelet Transform and Auto-Encoder for Steel Surface Defect Classification |
| title_sort | novel feature extraction method based on legendre multi wavelet transform and auto encoder for steel surface defect classification |
| topic | Classification steel surface defects Legendre multi-wavelet transform auto-encoder |
| url | https://ieeexplore.ieee.org/document/10380552/ |
| work_keys_str_mv | AT xiaoyangzheng anovelfeatureextractionmethodbasedonlegendremultiwavelettransformandautoencoderforsteelsurfacedefectclassification AT weishuoliu anovelfeatureextractionmethodbasedonlegendremultiwavelettransformandautoencoderforsteelsurfacedefectclassification AT yanhuang anovelfeatureextractionmethodbasedonlegendremultiwavelettransformandautoencoderforsteelsurfacedefectclassification AT xiaoyangzheng novelfeatureextractionmethodbasedonlegendremultiwavelettransformandautoencoderforsteelsurfacedefectclassification AT weishuoliu novelfeatureextractionmethodbasedonlegendremultiwavelettransformandautoencoderforsteelsurfacedefectclassification AT yanhuang novelfeatureextractionmethodbasedonlegendremultiwavelettransformandautoencoderforsteelsurfacedefectclassification |