arXiv:2411.16567cs.LG2024-11被引 31

用GAN融合数据增强与模型微调,提升小样本学习性能

Enhancing Few-Shot Learning with Integrated Data and GAN Model Approaches

  • 结合GAN与MCMC采样生成更真实的小样本数据
  • 新算法在小样本图像和结构化数据上分类准确率显著提升
  • 适合数据稀缺场景如药物发现、恶意流量检测

本文提出一种创新框架,通过整合数据增强与模型微调,解决小样本学习中的数据不足问题。针对传统机器学习依赖大规模数据的局限性,尤其在药物发现、目标识别和恶意流量检测等场景,研究利用生成对抗网络(GAN)与先进优化技术,在有限数据下提升模型表现。为克服数据增强引入的噪声与偏差,同时结合基于模型的方法(如微调与度量学习),提出在GAN框架内融合马尔可夫链蒙特卡洛(MCMC)采样与判别模型集成策略,动态调整生成与判别分布,模拟更广泛的相关数据。进一步采用MHLoss与重参数化GAN集成,增强稳定性并加速收敛。实验表明,所提出的MhERGAN算法在小样本图像和结构化数据集上均实现优异分类性能,为数据稀缺环境下的高效模型适应与泛化提供了实用解决方案。

原文摘要 · Abstract (English)

This paper presents an innovative approach to enhancing few-shot learning by integrating data augmentation with model fine-tuning in a framework designed to tackle the challenges posed by small-sample data. Recognizing the critical limitations of traditional machine learning models that require large datasets-especially in fields such as drug discovery, target recognition, and malicious traffic detection-this study proposes a novel strategy that leverages Generative Adversarial Networks (GANs) and advanced optimization techniques to improve model performance with limited data. Specifically, the paper addresses the noise and bias issues introduced by data augmentation methods, contrasting them with model-based approaches, such as fine-tuning and metric learning, which rely heavily on related datasets. By combining Markov Chain Monte Carlo (MCMC) sampling and discriminative model ensemble strategies within a GAN framework, the proposed model adjusts generative and discriminative distributions to simulate a broader range of relevant data. Furthermore, it employs MHLoss and a reparameterized GAN ensemble to enhance stability and accelerate convergence, ultimately leading to improved classification performance on small-sample images and structured datasets. Results confirm that the MhERGAN algorithm developed in this research is highly effective for few-shot learning, offering a practical solution that bridges data scarcity with high-performing model adaptability and generalization.

小样本学习GAN数据增强模型微调

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