HydraMix：小数据集图像分类的多图像特征混合
计算机视觉与模式识别
2025-01-17 v1
摘要
训练深度神经网络需要大量标注样本的数据集。数据集的收集和标注不仅成本极高，而且还面临法律和隐私问题。这些因素是许多实际应用的重要限制。为此，我们引入了 HydraMix，一种新型架构，通过混合来自相同类别的多个不同图像生成新的图像构图。HydraMix 在特征空间中学习由分段基于混合掩码引导的图像内容融合，并通过无监督训练和对抗训练进行优化。我们的数据增强方案允许在极小的数据集上从头训练模型。我们在 ciFAIR-10、STL-10 和 ciFAIR-100 上进行了大量实验。此外，我们引入了一种新型文本-图像指标来评估增强数据集的通用性。我们的结果表明，HydraMix 在小数据集上 outperform 了现有的领先方法。
引用
@article{arxiv.2501.09504,
title = {HydraMix: Multi-Image Feature Mixing for Small Data Image Classification},
author = {Christoph Reinders and Frederik Schubert and Bodo Rosenhahn},
journal= {arXiv preprint arXiv:2501.09504},
year = {2025}
}