arXiv:2504.02842eess.SPcs.LG2025-04

用分布距离优化融合方法,提升多源心电图分类准确率

Enhanced ECG Arrhythmia Detection Accuracy by Optimizing Divergence-Based Data Fusion

  • 基于核密度估计与KL散度,动态优化特征融合参数
  • 在合并数据集上使异常心电图分类准确率显著高于归一化融合
  • 适合处理设备多样、数据异构的医疗AI场景

医疗AI在临床数据有限且异构的情况下面临重大挑战。整合来自不同来源和设备的数据对有效AI计算至关重要,但其多样性、复杂性及代表性不足使其难以实现。当前普遍采用归一化后融合的方法,但会引入冗余信息,降低信噪比,影响分类精度。为此,我们提出一种基于特征的融合算法,利用核密度估计(KDE)与Kullback-Leibler(KL)散度。方法包括对提取特征进行预处理与连续估计,再通过梯度下降法寻找最小化特征分布间KL散度的最优线性参数。使用自建数据集(2000名健康者与2000名患者,不同设备采集的ECG信号)及公开的PTB-XL数据集(含21,837条心电记录,来自18,885名患者)验证。采用轻量梯度提升机(LGBM)模型进行二分类。结果表明,所提融合方法在合并数据集上显著提升了异常心电图的分类准确率,优于传统归一化融合策略。该数据融合策略为异构数据的最优AI计算提供了新路径。

原文摘要 · Abstract (English)

AI computation in healthcare faces significant challenges when clinical datasets are limited and heterogeneous. Integrating datasets from multiple sources and different equipments is critical for effective AI computation but is complicated by their diversity, complexity, and lack of representativeness, so we often need to join multiple datasets for analysis. The currently used method is fusion after normalization. But when using this method, it can introduce redundant information, decreasing the signal-to-noise ratio and reducing classification accuracy. To tackle this issue, we propose a feature-based fusion algorithm utilizing Kernel Density Estimation (KDE) and Kullback-Leibler (KL) divergence. Our approach involves initially preprocessing and continuous estimation on the extracted features, followed by employing the gradient descent method to identify the optimal linear parameters that minimize the KL divergence between the feature distributions. Using our in-house datasets consisting of ECG signals collected from 2000 healthy and 2000 diseased individuals by different equipments and verifying our method by using the publicly available PTB-XL dataset which contains 21,837 ECG recordings from 18,885 patients. We employ a Light Gradient Boosting Machine (LGBM) model to do the binary classification. The results demonstrate that the proposed fusion method significantly enhances feature-based classification accuracy for abnormal ECG cases in the merged datasets, compared to the normalization method. This data fusion strategy provides a new approach to process heterogeneous datasets for the optimal AI computation results.

心电图数据融合KL散度医疗AI

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。