arXiv:2502.08146cs.LGstat.ME2025-02ICML被引 4

用外部知识指导分布鲁棒优化,提升小样本迁移学习效果

Knowledge-Guided Wasserstein Distributionally Robust Optimization

论文配图:Knowledge-Guided Wasserstein Distributionally Robust Optimization
图 1 · 摘自论文原文
  • 基于源知识约束运输方向,缩小瓦瑟斯坦不确定集
  • 在小样本下显著优于传统方法,避免过度保守性
  • 适用于多种正则化,可处理源目标模型尺度差异

迁移学习通过利用外部知识提升小样本下的统计效率。本文提出一种新型知识引导的瓦瑟斯坦分布鲁棒优化(KG-WDRO)框架,通过自适应融合多源外部知识,缓解经典WDRO因过于保守而导致的预测值收缩至零的问题。该方法基于源知识指引的方向控制运输路径,构建更小的瓦瑟斯坦模糊集,有效减少协变量预测投影的扰动并防止信息损失。理论上建立了本方法与基于共线相似性的知识引导收缩估计之间的等价关系,保证可计算性并几何化可行集。这一结果从分布鲁棒性视角揭示了近期收缩类迁移学习方法的新通用解释。此外,该框架可调整源与目标模型间的尺度差异,并支持Lasso、Ridge等通用正则化形式。大量模拟实验表明,KG-WDRO在增强小样本迁移学习方面表现更优且具备良好适应性。

原文摘要 · Abstract (English)

Transfer learning is a popular strategy to leverage external knowledge and improve statistical efficiency, particularly with a limited target sample. We propose a novel knowledge-guided Wasserstein Distributionally Robust Optimization (KG-WDRO) framework that adaptively incorporates multiple sources of external knowledge to overcome the conservativeness of vanilla WDRO, which often results in overly pessimistic shrinkage toward zero. Our method constructs smaller Wasserstein ambiguity sets by controlling the transportation along directions informed by the source knowledge. This strategy can alleviate perturbations on the predictive projection of the covariates and protect against information loss. Theoretically, we establish the equivalence between our WDRO formulation and the knowledge-guided shrinkage estimation based on collinear similarity, ensuring tractability and geometrizing the feasible set. This also reveals a novel and general interpretation for recent shrinkage-based transfer learning approaches from the perspective of distributional robustness. In addition, our framework can adjust for scaling differences in the regression models between the source and target and accommodates general types of regularization such as lasso and ridge. Extensive simulations demonstrate the superior performance and adaptivity of KG-WDRO in enhancing small-sample transfer learning.

迁移学习分布鲁棒小样本知识引导

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