1
|
Liu LR, Huang MY, Huang ST, Kung LC, Lee CH, Yao WT, Tsai MF, Hsu CH, Chu YC, Hung FH, Chiu HW. An Arrhythmia classification approach via deep learning using single-lead ECG without QRS wave detection. Heliyon 2024; 10:e27200. [PMID: 38486759 PMCID: PMC10937691 DOI: 10.1016/j.heliyon.2024.e27200] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Download PDF] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 01/07/2024] [Revised: 02/18/2024] [Accepted: 02/26/2024] [Indexed: 03/17/2024] Open
Abstract
Arrhythmia, a frequently encountered and life-threatening cardiac disorder, can manifest as a transient or isolated event. Traditional automatic arrhythmia detection methods have predominantly relied on QRS-wave signal detection. Contemporary research has focused on the utilization of wearable devices for continuous monitoring of heart rates and rhythms through single-lead electrocardiogram (ECG), which holds the potential to promptly detect arrhythmias. However, in this study, we employed a convolutional neural network (CNN) to classify distinct arrhythmias without QRS wave detection step. The ECG data utilized in this study were sourced from the publicly accessible PhysioNet databases. Taking into account the impact of the duration of ECG signal on accuracy, this study trained one-dimensional CNN models with 5-s and 10-s segments, respectively, and compared their results. In the results, the CNN model exhibited the capability to differentiate between Normal Sinus Rhythm (NSR) and various arrhythmias, including Atrial Fibrillation (AFIB), Atrial Flutter (AFL), Wolff-Parkinson-White syndrome (WPW), Ventricular Fibrillation (VF), Ventricular Tachycardia (VT), Ventricular Flutter (VFL), Mobitz II AV Block (MII), and Sinus Bradycardia (SB). Both 10-s and 5-s ECG segments exhibited comparable results, with an average classification accuracy of 97.31%. It reveals the feasibility of utilizing even shorter 5-s recordings for detecting arrhythmias in everyday scenarios. Detecting arrhythmias with a single lead aligns well with the practicality of wearable devices for daily use, and shorter detection times also align with their clinical utility in emergency situations.
Collapse
Affiliation(s)
- Liong-Rung Liu
- Graduate Institute of Biomedical Informatics, College of Medical Science and Technology, Taipei Medical University, Taipei, Taiwan
- Department of Emergency Medicine, Mackay Memorial Hospital, Taipei, Taiwan
- Department of Medicine, Mackay Medical College, New Taipei City, Taiwan
| | - Ming-Yuan Huang
- Department of Emergency Medicine, Mackay Memorial Hospital, Taipei, Taiwan
- Department of Medicine, Mackay Medical College, New Taipei City, Taiwan
| | - Shu-Tien Huang
- Graduate Institute of Biomedical Informatics, College of Medical Science and Technology, Taipei Medical University, Taipei, Taiwan
- Department of Emergency Medicine, Mackay Memorial Hospital, Taipei, Taiwan
- Department of Medicine, Mackay Medical College, New Taipei City, Taiwan
| | - Lu-Chih Kung
- Department of Emergency Medicine, Mackay Memorial Hospital, Taipei, Taiwan
- Department of Medicine, Mackay Medical College, New Taipei City, Taiwan
| | - Chao-hsiung Lee
- Department of Emergency Medicine, Mackay Memorial Hospital, Taipei, Taiwan
- Department of Medicine, Mackay Medical College, New Taipei City, Taiwan
| | - Wen-Teng Yao
- Division of Plastic Surgery, Department of Surgery, Mackay Memorial Hospital, Taipei, Taiwan
- Department of Medicine, Mackay Medical College, New Taipei City, Taiwan
| | - Ming-Feng Tsai
- Graduate Institute of Biomedical Informatics, College of Medical Science and Technology, Taipei Medical University, Taipei, Taiwan
- Division of Plastic Surgery, Department of Surgery, Mackay Memorial Hospital, Taipei, Taiwan
- Department of Medicine, Mackay Medical College, New Taipei City, Taiwan
| | - Cheng-Hung Hsu
- Graduate Institute of Biomedical Informatics, College of Medical Science and Technology, Taipei Medical University, Taipei, Taiwan
| | - Yu-Chang Chu
- Graduate Institute of Biomedical Informatics, College of Medical Science and Technology, Taipei Medical University, Taipei, Taiwan
| | - Fei-Hung Hung
- Health Data Analytics and Statistics Center, Office of Data Science, Taipei Medical University, Taipei, Taiwan
| | - Hung-Wen Chiu
- Graduate Institute of Biomedical Informatics, College of Medical Science and Technology, Taipei Medical University, Taipei, Taiwan
- Clinical Big Data Research Center, Taipei Medical University Hospital, Taipei, Taiwan
- Bioinformatics Data Science Center, Wan Fang Hospital, Taipei Medical University, Taipei, Taiwan
| |
Collapse
|
2
|
Vozzi F, Pedrelli L, Dimitri GM, Micheli A, Persiani E, Piacenti M, Rossi A, Solarino G, Pieragnoli P, Checchi L, Zucchelli G, Mazzocchetti L, De Lucia R, Nesti M, Notarstefano P, Morales MA. Echo state networks for the recognition of type 1 Brugada syndrome from conventional 12-LEAD ECG. Heliyon 2024; 10:e25404. [PMID: 38333823 PMCID: PMC10850578 DOI: 10.1016/j.heliyon.2024.e25404] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Download PDF] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 07/04/2023] [Revised: 10/16/2023] [Accepted: 01/25/2024] [Indexed: 02/10/2024] Open
Abstract
Artificial Intelligence (AI) applications and Machine Learning (ML) methods have gained much attention in recent years for their ability to automatically detect patterns in data without being explicitly taught rules. Specific features characterise the ECGs of patients with Brugada Syndrome (BrS); however, there is still ambiguity regarding the correct diagnosis of BrS and its differentiation from other pathologies. This work presents an application of Echo State Networks (ESN) in the Recurrent Neural Networks (RNN) class for diagnosing BrS from the ECG time series. 12-lead ECGs were obtained from patients with a definite clinical diagnosis of spontaneous BrS Type 1 pattern (Group A), patients who underwent provocative pharmacological testing to induce BrS type 1 pattern, which resulted in positive (Group B) or negative (Group C), and control subjects (Group D). One extracted beat in the V2 lead was used as input, and the dataset was used to train and evaluate the ESN model using a double cross-validation approach. ESN performance was compared with that of 4 cardiologists trained in electrophysiology. The model performance was assessed in the dataset, with a correct global diagnosis observed in 91.5 % of cases compared to clinicians (88.0 %). High specificity (94.5 %), sensitivity (87.0 %) and AUC (94.7 %) for BrS recognition by ESN were observed in Groups A + B vs. C + D. Our results show that this ML model can discriminate Type 1 BrS ECGs with high accuracy comparable to expert clinicians. Future availability of larger datasets may improve the model performance and increase the potential of the ESN as a clinical support system tool for daily clinical practice.
Collapse
Affiliation(s)
| | - Luca Pedrelli
- Department of Computer Science, University of Pisa, Pisa, Italy
| | - Giovanna Maria Dimitri
- Department of Computer Science, University of Pisa, Pisa, Italy
- Department of Information Engineering and Mathematics, University of Siena, Siena, Italy
| | - Alessio Micheli
- Department of Computer Science, University of Pisa, Pisa, Italy
| | | | | | - Andrea Rossi
- Fondazione Toscana Gabriele Monasterio, Pisa, Italy
| | | | | | - Luca Checchi
- Ospedale Careggi, University of Florence, Firenze, Italy
| | - Giulio Zucchelli
- Second Division of Cardiology, Azienda Ospedaliero Universitaria Pisana, Pisa, Italy
| | - Lorenzo Mazzocchetti
- Second Division of Cardiology, Azienda Ospedaliero Universitaria Pisana, Pisa, Italy
| | - Raffaele De Lucia
- Second Division of Cardiology, Azienda Ospedaliero Universitaria Pisana, Pisa, Italy
| | - Martina Nesti
- Cardiovascular and Neurological Department, San Donato Hospital, Arezzo, Italy
| | | | | |
Collapse
|
3
|
Li X, Chen S, Wu J, Li J, Wang T, Tang J, Hu T, Wu W. Satellite cloud image segmentation based on lightweight convolutional neural network. PLoS One 2023; 18:e0280408. [PMID: 36745635 PMCID: PMC9901801 DOI: 10.1371/journal.pone.0280408] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [MESH Headings] [Grants] [Track Full Text] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 10/16/2022] [Accepted: 12/28/2022] [Indexed: 02/07/2023] Open
Abstract
More than 50% of the images captured by optical satellites are covered by clouds, which reduces the available information in the images and seriously affects the subsequent applications of satellite images. Therefore, the identification and segmentation of cloud regions come to be one of the most important problems in current satellite image processing. Due to the complexity and variability of satellite images, especially when the ground is covered with snow, the boundary information of cloud regions is difficult to be accurately identified. The fast and accurate segmentation of cloud regions is a difficult point in the current research. We propose a lightweight convolutional neural network. Firstly, channel attention is used to optimize the effective information in the feature maps as a way to improve the network's ability to extract semantic information at each scale. Then, we fuse high and low-dimensional feature maps to enhance the network's ability to obtain small-scale semantic information. In addition, the feature aggregation module automatically adjusts the input multi-level feature weights to highlight the details of different features. Finally, we design the fully connected conditional random field to solve the problem that some noise in the input image and local minima during training is passed to the output layer resulting in the loss of edge features. Experimental results show that the proposed method achieves 0.9695 and 0.8218 for overall accuracy and recall, respectively, which has higher segmentation accuracy with the shortest time consumption compared with other state-of-the-art methods.
Collapse
Affiliation(s)
- Xi Li
- Foundation Department, Chongqing Medical and Pharmaceutical College, Chongqing, China
- School of Communication and Information Engineering, Chongqing University of Posts and Telecommunications, Chongqing, Chongqing, China
| | - Shilan Chen
- Chongqing Botong Water Conservancy Information Network Co.,Ltd., Chongqing, China
| | - Jin Wu
- Chongqing Botong Water Conservancy Information Network Co.,Ltd., Chongqing, China
| | - Jun Li
- Foundation Department, Chongqing Medical and Pharmaceutical College, Chongqing, China
| | - Ting Wang
- Foundation Department, Chongqing Medical and Pharmaceutical College, Chongqing, China
| | - Junquan Tang
- Foundation Department, Chongqing Medical and Pharmaceutical College, Chongqing, China
| | - Tongyi Hu
- Foundation Department, Chongqing Medical and Pharmaceutical College, Chongqing, China
| | - Wenzhu Wu
- Foundation Department, Chongqing Medical and Pharmaceutical College, Chongqing, China
- * E-mail:
| |
Collapse
|
4
|
Baressi Šegota S, Lorencin I, Kovač Z, Car Z. On Approximating the pIC50 Value of COVID-19 Medicines In Silico with Artificial Neural Networks. Biomedicines 2023; 11:biomedicines11020284. [PMID: 36830823 PMCID: PMC9952997 DOI: 10.3390/biomedicines11020284] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 12/19/2022] [Revised: 01/16/2023] [Accepted: 01/17/2023] [Indexed: 01/21/2023] Open
Abstract
In the case of pandemics such as COVID-19, the rapid development of medicines addressing the symptoms is necessary to alleviate the pressure on the medical system. One of the key steps in medicine evaluation is the determination of pIC50 factor, which is a negative logarithmic expression of the half maximal inhibitory concentration (IC50). Determining this value can be a lengthy and complicated process. A tool allowing for a quick approximation of pIC50 based on the molecular makeup of medicine could be valuable. In this paper, the creation of the artificial intelligence (AI)-based model is performed using a publicly available dataset of molecules and their pIC50 values. The modeling algorithms used are artificial and convolutional neural networks (ANN and CNN). Three approaches are tested-modeling using just molecular properties (MP), encoded SMILES representation of the molecule, and the combination of both input types. Models are evaluated using the coefficient of determination (R2) and mean absolute percentage error (MAPE) in a five-fold cross-validation scheme to assure the validity of the results. The obtained models show that the highest quality regression (R2¯=0.99, σR2¯=0.001; MAPE¯=0.009%, σMAPE¯=0.009), by a large margin, is obtained when using a hybrid neural network trained with both MP and SMILES.
Collapse
Affiliation(s)
- Sandi Baressi Šegota
- Department of Automation and Electronics, Faculty of Engineering, University of Rijeka, Vukovarska 58, 51000 Rijeka, Croatia
- Correspondence: ; Tel.: +385-51-505-715
| | - Ivan Lorencin
- Department of Automation and Electronics, Faculty of Engineering, University of Rijeka, Vukovarska 58, 51000 Rijeka, Croatia
| | - Zoran Kovač
- Faculty of Dental Medicine, University of Rijeka, Krešimirova 40/42, 51000 Rijeka, Croatia
| | - Zlatan Car
- Department of Automation and Electronics, Faculty of Engineering, University of Rijeka, Vukovarska 58, 51000 Rijeka, Croatia
| |
Collapse
|
5
|
Tutuko B, Darmawahyuni A, Nurmaini S, Tondas AE, Naufal Rachmatullah M, Teguh SBP, Firdaus F, Sapitri AI, Passarella R. DAE-ConvBiLSTM: End-to-end learning single-lead electrocardiogram signal for heart abnormalities detection. PLoS One 2022; 17:e0277932. [PMID: 36584187 PMCID: PMC9803308 DOI: 10.1371/journal.pone.0277932] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [MESH Headings] [Grants] [Track Full Text] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 08/01/2022] [Accepted: 11/08/2022] [Indexed: 01/01/2023] Open
Abstract
BACKGROUND The electrocardiogram (ECG) is a widely used diagnostic that observes the heart activities of patients to ascertain a heart abnormality diagnosis. The artifacts or noises are primarily associated with the problem of ECG signal processing. Conventional denoising techniques have been proposed in previous literature; however, some lacks, such as the determination of suitable wavelet basis function and threshold, can be a time-consuming process. This paper presents end-to-end learning using a denoising auto-encoder (DAE) for denoising algorithms and convolutional-bidirectional long short-term memory (ConvBiLSTM) for ECG delineation to classify ECG waveforms in terms of the PQRST-wave and isoelectric lines. The denoising reconstruction using unsupervised learning based on the encoder-decoder process can be proposed to improve the drawbacks. First, The ECG signals are reduced to a low-dimensional vector in the encoder. Second, the decoder reconstructed the signals. The last, the reconstructed signals of ECG can be processed to ConvBiLSTM. The proposed architecture of DAE-ConvBiLSTM is the end-to-end diagnosis of heart abnormality detection. RESULTS As a result, the performance of DAE-ConvBiLSTM has obtained an average of above 98.59% accuracy, sensitivity, specificity, precision, and F1 score from the existing studies. The DAE-ConvBiLSTM has also experimented with detecting T-wave (due to ventricular repolarisation) morphology abnormalities. CONCLUSION The development architecture for detecting heart abnormalities using an unsupervised learning DAE and supervised learning ConvBiLSTM can be proposed for an end-to-end learning algorithm. In the future, the precise accuracy of the ECG main waveform will affect heart abnormalities detection in clinical practice.
Collapse
Affiliation(s)
- Bambang Tutuko
- Intelligent System Research Group, Faculty of Computer Science, Universitas Sriwijaya, Palembang, Indonesia
| | - Annisa Darmawahyuni
- Intelligent System Research Group, Faculty of Computer Science, Universitas Sriwijaya, Palembang, Indonesia
- * E-mail: , .id (SN); , .id (AD)
| | - Siti Nurmaini
- Intelligent System Research Group, Faculty of Computer Science, Universitas Sriwijaya, Palembang, Indonesia
- * E-mail: , .id (SN); , .id (AD)
| | - Alexander Edo Tondas
- Department of Cardiology & Vascular Medicine, Dr. Mohammad Hoesin Hospital, Palembang, Indonesia
| | | | | | - Firdaus Firdaus
- Intelligent System Research Group, Faculty of Computer Science, Universitas Sriwijaya, Palembang, Indonesia
| | - Ade Iriani Sapitri
- Intelligent System Research Group, Faculty of Computer Science, Universitas Sriwijaya, Palembang, Indonesia
| | - Rossi Passarella
- Intelligent System Research Group, Faculty of Computer Science, Universitas Sriwijaya, Palembang, Indonesia
| |
Collapse
|
6
|
Nawaz M, Ahmed J. Cloud-based healthcare framework for real-time anomaly detection and classification of 1-D ECG signals. PLoS One 2022; 17:e0279305. [PMID: 36574391 PMCID: PMC9794080 DOI: 10.1371/journal.pone.0279305] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [MESH Headings] [Track Full Text] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 09/30/2022] [Accepted: 12/05/2022] [Indexed: 12/28/2022] Open
Abstract
Real-time data collection and pre-processing have enabled the recognition, realization, and prediction of diseases by extracting and analysing the important features of physiological data. In this research, an intelligent end-to-end system for anomaly detection and classification of raw, one-dimensional (1D) electrocardiogram (ECG) signals is given to assess cardiovascular activity automatically. The acquired raw ECG data is pre-processed carefully before storing it in the cloud, and then deeply analyzed for anomaly detection. A deep learning-based auto-encoder(AE) algorithm is applied for the anomaly detection of 1D ECG time-series signals. As a next step, the implemented system identifies it by a multi-label classification algorithm. To improve the classification accuracy and model robustness the improved feature-engineered parameters of the large and diverse datasets have been incorporated. The training has been done using the amazon web service (AWS) machine learning services and cloud-based storage for a unified solution. Multi-class classification of raw ECG signals is challenging due to a large number of possible label combinations and noise susceptibility. To overcome this problem, a performance comparison of a large set of machine algorithms in terms of classification accuracy is presented on an improved feature-engineered dataset. The proposed system reduces the raw signal size up to 95% using wavelet time scattering features to make it less compute-intensive. The results show that among several state-of-the-art techniques, the long short-term memory (LSTM) method has shown 100% classification accuracy, and an F1 score on the three-class test dataset. The ECG signal anomaly detection algorithm shows 98% accuracy using deep LSTM auto-encoders with a reconstructed error threshold of 0.02 in terms of absolute error loss. Our approach provides performance and predictive improvement with an average mean absolute error loss of 0.0072 for normal signals and 0.078 for anomalous signals.
Collapse
Affiliation(s)
- Menaa Nawaz
- Department of electrical engineering, Riphah International University, Islamabad, Pakistan
- * E-mail:
| | - Jameel Ahmed
- Department of electrical engineering, Riphah International University, Islamabad, Pakistan
| |
Collapse
|
7
|
Hammad M, Meshoul S, Dziwiński P, Pławiak P, Elgendy IA. Efficient Lightweight Multimodel Deep Fusion Based on ECG for Arrhythmia Classification. SENSORS (BASEL, SWITZERLAND) 2022; 22:9347. [PMID: 36502049 PMCID: PMC9736761 DOI: 10.3390/s22239347] [Citation(s) in RCA: 1] [Impact Index Per Article: 0.5] [Reference Citation Analysis] [Abstract] [Key Words] [MESH Headings] [Grants] [Track Full Text] [Figures] [Subscribe] [Scholar Register] [Received: 10/19/2022] [Revised: 11/29/2022] [Accepted: 11/29/2022] [Indexed: 06/17/2023]
Abstract
An arrhythmia happens when the electrical signals that organize the heartbeat do not work accurately. Most cases of arrhythmias may increase the risk of stroke or cardiac arrest. As a result, early detection of arrhythmia reduces fatality rates. This research aims to provide a lightweight multimodel based on convolutional neural networks (CNNs) that can transfer knowledge from many lightweight deep learning models and decant it into one model to aid in the diagnosis of arrhythmia by using electrocardiogram (ECG) signals. Thus, we gained a multimodel able to classify arrhythmia from ECG signals. Our system's effectiveness is examined by using a publicly accessible database and a comparison to the current methodologies for arrhythmia classification. The results we achieved by using our multimodel are better than those obtained by using a single model and better than most of the previous detection methods. It is worth mentioning that this model produced accurate classification results on small collection of data. Experts in this field can use this model as a guide to help them make decisions and save time.
Collapse
Affiliation(s)
- Mohamed Hammad
- Department of Information Technology, Faculty of Computers and Information, Menoufia University, Shibin El Kom 32511, Egypt
| | - Souham Meshoul
- Department of Information Technology, College of Computer and Information Sciences, Princess Nourah Bint Abdulrahman University, P.O. Box 84428, Riyadh 11671, Saudi Arabia
| | - Piotr Dziwiński
- Department of Intelligent Computer Systems, Czestochowa University of Technology, Armii Krajowej 36, 42-218 Czestochowa, Poland
| | - Paweł Pławiak
- Department of Computer Science, Faculty of Computer Science and Telecommunications, Cracow University of Technology, Warszawska 24, 31-155 Krakow, Poland
- Institute of Theoretical and Applied Informatics, Polish Academy of Sciences, Baltycka 5, 44-100 Gliwice, Poland
| | - Ibrahim A. Elgendy
- Department of Computer Science, Faculty of Computers and Information, Menoufia University, Shibin El Kom 32511, Egypt
| |
Collapse
|
8
|
Sakr AS, Pławiak P, Tadeusiewicz R, Pławiak J, Sakr M, Hammad M. ECG-COVID: An End-to-End Deep Model Based on Electrocardiogram for COVID-19 Detection. Inf Sci (N Y) 2022; 619:324-339. [PMCID: PMC9673093 DOI: 10.1016/j.ins.2022.11.069] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [Key Words] [Track Full Text] [Journal Information] [Subscribe] [Scholar Register] [Received: 02/23/2022] [Revised: 10/05/2022] [Accepted: 11/14/2022] [Indexed: 11/19/2022]
Abstract
The early and accurate detection of COVID-19 is vital nowadays to avoid the vast and rapid spread of this virus and ease lockdown restrictions. As a result, researchers developed methods to diagnose COVID-19. However, these methods have several limitations. Therefore, presenting new methods is essential to improve the diagnosis of COVID-19. Recently, investigation of the electrocardiogram (ECG) signals becoming an easy way to detect COVID-19 since the ECG process is non-invasive and easy to use. Therefore, we proposed in this paper a novel end-to-end deep learning model (ECG-COVID) based on ECG for COVID-19 detection. We employed several deep Convolutional Neural Networks (CNNs) on a dataset of 1109 ECG images, which is built for screening the perception of COVID-19 and cardiac patients. After that, we selected the most efficient model as our model for evaluation. The proposed model is end-to-end where the input ECG images are fed directly to the model for the final decision without using any additional stages. The proposed method achieved an average accuracy of 98.81%, Precision of 98.8%, Sensitivity of 98.8% and, F1-score of 98.81% for COVID-19 detection. As cases of corona continue to rise and hospitalizations continue again, hospitals may find our study helpful when dealing with these patients who did not get significantly worse.
Collapse
Affiliation(s)
- Ahmed S. Sakr
- Department of Information System, Faculty of Computers and Information, Menoufia University, Egypt
| | - Paweł Pławiak
- Department of Computer Science, Faculty of Computer Science and Telecommunications, Cracow University of Technology, Warszawska 24, 31-155 Krakow, Poland,Institute of Theoretical and Applied Informatics, Polish Academy of Sciences, Bałtycka 5, 44-100 Gliwice, Poland,Corresponding authors
| | - Ryszard Tadeusiewicz
- AGH University of Science and Technology, Department of Biocybernetics and Biomedical Engineering, Krakow, Poland
| | - Joanna Pławiak
- Faculty of Electrical and Computer Engineering, Cracow University of Technology, Warsaw 24, 31-155 Krakow, Poland
| | - Mohamed Sakr
- Computer Science Department, Faculty of Computers and Information, Menoufia University, Egypt
| | - Mohamed Hammad
- Department of Information Technology, Faculty of Computers and Information, Menoufia University, Egypt,Corresponding authors
| |
Collapse
|
9
|
Huang Y, Wang H, Wang R. Deep learning recommendation algorithm based on semantic mining. PLoS One 2022; 17:e0274940. [PMID: 36155978 PMCID: PMC9512199 DOI: 10.1371/journal.pone.0274940] [Citation(s) in RCA: 0] [Impact Index Per Article: 0] [Reference Citation Analysis] [Abstract] [MESH Headings] [Track Full Text] [Download PDF] [Figures] [Journal Information] [Subscribe] [Scholar Register] [Received: 06/07/2022] [Accepted: 09/08/2022] [Indexed: 11/19/2022] Open
Abstract
This paper proposes Deep Semantic Mining based Recommendation (DSMR), which can extract user features and item attribute features more accurately by deeply mining the semantic information of review text and item description documents recommend. First, the proposed model uses the BERT pre-training model to process review texts and item description documents, and deeply mine user characteristics and item attributes, which effectively alleviates the problems of data sparseness and item cold start; Then, the forward LSTM is used to pay attention to the changes of user preferences over time, and a more accurate recommendation is obtained; finally, in the model training stage, the experimental data are randomly divided into 1 to 5 points, 1:1:1:1:1. Extraction ensures that the amount of data for each score is equal, so that the results are more accurate and the model is more robust. Experiments are carried out on four commonly used Amazon public data sets, and the results show that with the root mean square error as the evaluation index, the error of DSMR recommendation results is at least 11.95% lower on average than the two classic recommendation models based only on rating data. At the same time, it is better than the three latest recommendation models based on review text, and it is 5.1% lower than the best model on average.
Collapse
Affiliation(s)
- Yongxin Huang
- The China Academy of Information and Communications Technology (CAICT), Beijing, China
| | - Hezheng Wang
- China Information and Communication Technology, Beijing, China
- * E-mail:
| | - Rui Wang
- University of Chinese Academy of Sciences, Beijing, China
| |
Collapse
|