提出可验证的少样本迁移学习方法,提升模型可靠性。
Model Diffusion for Certifiable Few-shot Transfer Learning
- 用扩散模型生成参数高效微调的候选解集
- 在少样本下获得非平凡的泛化误差上界
- 适合对安全性和可信度要求高的场景
当前深度学习中,通过参数高效微调(PEFT)适配预训练基础模型是解决低数据问题的常用有效方法。然而,尽管实证效果良好,现有方法缺乏可证明的泛化保证,难以满足高重要性应用中的伦理或法律要求。本文提出一种新迁移学习框架,旨在为下游任务提供非平凡的学习理论泛化保证,即使在少样本情形下亦可实现。具体地,先利用上游任务训练一个PEFT参数的概率分布;再通过采样-评估过程,在该分布中采样多个合理的PEFT方案,并选择在下游数据上似然最高的一个。关键在于,这一过程将模型假设空间限制为有限个样本,不同于传统神经网络连续权重空间,从而支持更紧的泛化风险证书。我们实例化该边界,并在少样本条件下证明其优于现有方法——后者在此情形下会产生平凡(vacuous)的泛化界限。
原文摘要 · Abstract (English)
In contemporary deep learning, a prevalent and effective workflow for solving low-data problems is adapting powerful pre-trained foundation models (FMs) to new tasks via parameter-efficient fine-tuning (PEFT). However, while empirically effective, the resulting solutions lack generalisation guarantees to certify their accuracy - which may be required for ethical or legal reasons prior to deployment in high-importance applications. In this paper we develop a novel transfer learning approach that is designed to facilitate non-vacuous learning theoretic generalisation guarantees for downstream tasks, even in the low-shot regime. Specifically, we first use upstream tasks to train a distribution over PEFT parameters. We then learn the downstream task by a sample-and-evaluate procedure -- sampling plausible PEFTs from the trained diffusion model and selecting the one with the highest likelihood on the downstream data. Crucially, this confines our model hypothesis to a finite set of PEFT samples. In contrast to the typical continuous hypothesis spaces of neural network weights, this facilitates tighter risk certificates. We instantiate our bound and show non-trivial generalization guarantees compared to existing learning approaches which lead to vacuous bounds in the low-shot regime.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。