arXiv:2409.08474cs.LGcs.CV2024-09被引 3

提出新视角解决元学习中的过拟合与欠拟合问题

Rethinking Meta-Learning from a Learning Lens

  • 将元学习拆解为初始化参数与任务微调层两部分
  • 通过任务关联性正则化,提升不同任务间的相互促进效果
  • 无需修改模型结构,适合各类元学习框架快速集成

元学习旨在从训练任务中学习一个具有良好泛化能力的模型初始化,以应对未见任务。从“学习如何学习”的视角看,初始化质量通常通过内循环的一步梯度下降建模。然而,实证分析表明,这可能导致元学习出现欠拟合。为弥合理论与实践之间的差距,我们从“学习”视角重新审视元学习,提出其包含两个相互关联的组件:用于模型初始化的参数和用于任务特定微调的元层。这两个组件在不同任务下分别面临过拟合与欠拟合风险,其解决方案(减少参数或增加元层)常相互冲突。为此,我们致力于在不修改数据或模型结构的前提下调节模型接收的任务信息。理论分析表明,不同任务的适应过程可相互增强,从而凸显有效信息。基于此,我们提出 TRLearner——一种即插即用方法,利用任务关系来校准元学习。它首先提取任务关系矩阵,再施加关系感知的一致性正则化以指导优化。大量理论与实证评估验证了其有效性。

原文摘要 · Abstract (English)

Meta-learning seeks to learn a well-generalized model initialization from training tasks to solve unseen tasks. From the "learning to learn" perspective, the quality of the initialization is modeled with one-step gradient decent in the inner loop. However, contrary to theoretical expectations, our empirical analysis reveals that this may expose meta-learning to underfitting. To bridge the gap between theoretical understanding and practical implementation, we reconsider meta-learning from the "Learning" lens. We propose that the meta-learning model comprises two interrelated components: parameters for model initialization and a meta-layer for task-specific fine-tuning. These components will lead to the risks of overfitting and underfitting depending on tasks, and their solutions, fewer parameters vs. more meta-layer, are often in conflict. To address this, we aim to regulate the task information the model receives without modifying the data or model structure. Our theoretical analysis indicates that models adapted to different tasks can mutually reinforce each other, highlighting the effective information. Based on this insight, we propose TRLearner, a plug-and-play method that leverages task relation to calibrate meta-learning. It first extracts task relation matrices and then applies relation-aware consistency regularization to guide optimization. Extensive theoretical and empirical evaluations demonstrate its effectiveness.

元学习任务关系正则化

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