Big Data Research ›› 2022, Vol. 8 ›› Issue (5): 139-152.doi: 10.11959/j.issn.2096-0271.2022039

• STUDY • Previous Articles     Next Articles

Classification algorithm for imbalance data of ECG based on PSOFS and TSK fuzzy system

Xinhui LI1,2, Qing SHEN2,3, Xiongtao ZHANG1,2   

  1. 1 School of Information Engineering, Huzhou University, Huzhou 313000, China
    2 Zhejiang Province Key Laboratory of Smart Management &Application of Modern Agricultural Resources, Huzhou 313000, China
    3 School of Science and Engineering, Huzhou College, Huzhou 313000, China
  • Online:2022-09-15 Published:2022-09-01
  • Supported by:
    The National Natural Science Foundation of China(61771193);The National Natural Science Foundation of China(61802123);General Research Program of Zhejiang Educational Committee(Y202146028)

Abstract:

A new classification model of electrocardiogram (ECG) signal based on particle swarm optimization feature selection (PSOFS) and TSK (Takagi-Sugeno-Kang) fuzzy system was proposed, i.e., parallel ensemble fuzzy neural network based on PSOFS and TSK (PE-PT-FN), which was used for ECG prediction.Each class sample in the training set was randomly sampled, and the samples obtained by randomly sampled were added.Then, the feature selection method PSOFS was carried out independently and parallelly.In PSOFS, particles that were random initial positions represent different feature subsets and converge to the optimal positions after many iterations.Each subset had a corresponding feature subset.Several groups of TSK fuzzy neural network (TSK-FNN) were trained by each feature subset in parallel.Medical researchers could effectively find the correlation between ECG signal data and different types of disease through the interpretability of the fuzzy system and the feature subsets by the PSOFS algorithm.Experiments prove that PE-PT-FN greatly improves the macro-R to 92.35% while retaining interpretability.

Key words: TSK fuzzy neural network, particle swarm optimization feature selection, ensemble learning, classification of ECG signal, imbalance data

CLC Number: 

No Suggested Reading articles found!