arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2602.19385cs.CVcs.CLcs.LG

自适应数据增强与多臂老虎机:用于隐式模式识别的样本高效嵌入校准

Adaptive Data Augmentation with Multi-armed Bandit: Sample-Efficient Embedding Calibration for Implicit Pattern Recognition

Duke University(杜克大学);Center for Advanced AI, Accenture(Accenture高级人工智能中心)

Minxue Tang, Yangyang Yu, Aolin Ding, Maziyar Baran Pouyan, Taha Belkhouja, Yujia Bao

AI总结:

本文提出ADAMAB框架,通过多臂老虎机机制实现样本高效的隐式模式识别,实验表明其在少样本情况下准确率提升达40%。

摘要:

识别隐式视觉和文本模式在现代AI的许多实际应用中至关重要。然而,处理长尾模式识别任务仍对当前预训练基础模型如LLMs和VLMs构成挑战。尽管微调预训练模型可以提高识别隐式模式的准确性,但通常由于缺乏训练数据和高计算开销而不可行。在本文中,我们提出ADAMAB,一种用于少样本模式识别的高效嵌入校准框架。为了最大限度地减少计算成本,ADAMAB在固定嵌入模型上训练嵌入器无关的轻量校准器,而无需访问其参数。为了减少对大规模训练数据的依赖,我们引入了一种基于多臂老虎机(MAB)机制的自适应数据增强策略。通过修改的上置信界算法,ADAMAB减少了梯度偏移并保证了少样本训练中的理论收敛性。我们的多模态实验验证了ADAMAB的优越性能,当使用每个类别少于5个初始数据样本进行训练时,准确率可提高高达40%。

英文摘要:

Recognizing implicit visual and textual patterns is essential in many real-world applications of modern AI. However, tackling long-tail pattern recognition tasks remains challenging for current pre-trained foundation models such as LLMs and VLMs. While finetuning pre-trained models can improve accuracy in recognizing implicit patterns, it is usually infeasible due to a lack of training data and high computational overhead. In this paper, we propose ADAMAB, an efficient embedding calibration framework for few-shot pattern recognition. To maximally reduce the computational costs, ADAMAB trains embedder-agnostic light-weight calibrators on top of fixed embedding models without accessing their parameters. To mitigate the need for large-scale training data, we introduce an adaptive data augmentation strategy based on the Multi-Armed Bandit (MAB) mechanism. With a modified upper confidence bound algorithm, ADAMAB diminishes the gradient shifting and offers theoretically guaranteed convergence in few-shot training. Our multi-modal experiments justify the superior performance of ADAMAB, with up to 40% accuracy improvement when training with less than 5 initial data samples of each class.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑