arXiv:2411.15809eess.IVcs.CV2024-11

用高阶动态模态分解增强心超图像分类,准确率最高提升22%。

A Novel Data Augmentation Tool for Enhancing Machine Learning Classification: A New Application of the Higher Order Dynamic Mode Decomposition for Improved Cardiac Disease Identification

  • 用高阶动态模态分解提取心超图像特征,生成新数据输入卷积网络
  • 结合原始图像与分解特征后,分类准确率最高提升22%
  • 适合做医学图像分类、小样本学习或数据增强的研究者参考

本文将高阶动态模态分解(HODMD)与卷积神经网络(CNN)结合,用于提升基于心超图像的多种心脏疾病分类准确率。研究使用来自健康小鼠及糖尿病心肌病、肥胖、主动脉缩窄性肥厚和心肌梗死小鼠的130个心超数据集。HODMD作为特征提取工具,识别每种疾病的关键动态模式,并将其作为输入提供给CNN。这一过程相当于对数据库进行维度扩展,首次在作者认知范围内将HODMD应用于机器学习中的数据增强。六个原始数据集被保留作为未见数据测试模型性能。实验对比两种训练方式:仅使用原始图像,以及结合原始图像与DMD模式。结果显示,融合特征后所有测试案例准确率均提升,最高达22%。这表明HODMD具有显著的数据增强潜力。

原文摘要 · Abstract (English)

In this work, a data-driven, modal decomposition method, the higher order dynamic mode decomposition (HODMD), is combined with a convolutional neural network (CNN) in order to improve the classification accuracy of several cardiac diseases using echocardiography images. The HODMD algorithm is used first as feature extraction technique for the echocardiography datasets, taken from both healthy mice and mice afflicted by different cardiac diseases (Diabetic Cardiomyopathy, Obesity, TAC Hypertrophy and Myocardial Infarction). A total number of 130 echocardiography datasets are used in this work. The dominant features related to each cardiac disease were identified and represented by the HODMD algorithm as a set of DMD modes, which then are used as the input to the CNN. In a way, the database dimension was augmented, hence HODMD has been used, for the first time to the authors knowledge, for data augmentation in the machine learning framework. Six sets of the original echocardiography databases were hold out to be used as unseen data to test the performance of the CNN. In order to demonstrate the efficiency of the HODMD technique, two testcases are studied: the CNN is first trained using the original echocardiography images only, and second training the CNN using a combination of the original images and the DMD modes. The classification performance of the designed trained CNN shows that combining the original images with the DMD modes improves the results in all the testcases, as it improves the accuracy by up to 22%. These results show the great potential of using the HODMD algorithm as a data augmentation technique.

数据增强医学图像动态模态分解分类

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