Paper Reading AI Learner

Artificial Data Point Generation in Clustered Latent Space for Small Medical Datasets

2024-09-26 09:51:08
Yasaman Haghbin, Hadi Moradi, Reshad Hosseini

Abstract

One of the growing trends in machine learning is the use of data generation techniques, since the performance of machine learning models is dependent on the quantity of the training dataset. However, in many medical applications, collecting large datasets is challenging due to resource constraints, which leads to overfitting and poor generalization. This paper introduces a novel method, Artificial Data Point Generation in Clustered Latent Space (AGCL), designed to enhance classification performance on small medical datasets through synthetic data generation. The AGCL framework involves feature extraction, K-means clustering, cluster evaluation based on a class separation metric, and the generation of synthetic data points from clusters with distinct class representations. This method was applied to Parkinson's disease screening, utilizing facial expression data, and evaluated across multiple machine learning classifiers. Experimental results demonstrate that AGCL significantly improves classification accuracy compared to baseline, GN and kNNMTD. AGCL achieved the highest overall test accuracy of 83.33% and cross-validation accuracy of 90.90% in majority voting over different emotions, confirming its effectiveness in augmenting small datasets.

Abstract (translated)

机器学习的一个不断增长的趋势是使用数据生成技术,因为机器学习模型的性能取决于训练数据集的数量。然而,在许多医疗应用中,收集大量数据是非常具有挑战性的,由于资源限制,导致过拟合和欠拟合。本文介绍了一种名为人工数据点生成在聚类潜在空间(AGCL)的新方法,旨在通过合成数据生成来提高小型医疗数据的分类性能。AGCL框架包括特征提取、K-means聚类、基于类分离指标的聚类评估和从聚类中生成合成数据点。将该方法应用于帕金森病筛查,利用面部表情数据,并在多个机器学习分类器上进行评估。实验结果表明,AGCL相对于基线、GN和kNNMTD显著提高了分类准确率。AGCL在不同的情感上的整体测试准确率为83.33%,交叉验证准确率为90.90%,证实了其在增强小数据集的有效性。

URL

https://arxiv.org/abs/2409.17685

PDF

https://arxiv.org/pdf/2409.17685.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot