arXiv:2604.27723cs.LGstat.ML2026-04被引 7

解决专家不均衡下的精准任务转接问题,提升模型效率与准确率。

Optimized Deferral for Imbalanced Settings

  • 将转接损失建模为输入-专家域的代价敏感学习问题。
  • 在图像分类和LLM路由任务中显著优于现有基线方法。
  • 适合医疗诊断、语言生成等需高效资源调度的高风险场景。

学习算法可通过将复杂或不确定的输入路由至专用专家来显著提升性能,实现准确率与计算成本的平衡。这种称为‘学习转接’的方法在自然语言生成、医学诊断和计算机视觉等领域至关重要,有效转接可在低额外资源消耗下降低错误。然而,两阶段学习转接设置常因专家不均衡问题而表现不佳,导致转接算法偏向多数专家。本文系统研究了专家不均衡环境下的两阶段学习转接。我们将转接损失优化建模为输入-专家域上的新型代价敏感学习问题,推导出针对该场景设计的边际型损失函数与理论保证,并提出新的代价敏感学习算法。基于此,我们设计了专为专家不均衡场景优化的原理性转接算法MILD(Margin-based Imbalanced Learning to Defer)。大量实验表明,该方法在图像分类和真实世界大型语言模型(LLM)路由任务上均显著优于现有基线。

原文摘要 · Abstract (English)

Learning algorithms can be significantly improved by routing complex or uncertain inputs to specialized experts, balancing accuracy with computational cost. This approach, known as learning to defer, is essential in domains like natural language generation, medical diagnosis, and computer vision, where an effective deferral can reduce errors at low extra resource consumption. However, the two-stage learning to defer setting, which leverages existing predictors such as a collection of LLMs or other classifiers, often faces challenges due to an expert imbalance problem. This imbalance can lead to suboptimal performance, with deferral algorithms favoring the majority expert. We present a comprehensive study of two-stage learning to defer in expert imbalance settings. We cast the deferral loss optimization as a novel cost-sensitive learning problem over the input-expert domain. We derive new margin-based loss functions and guarantees tailored to this setting, and develop novel algorithms for cost-sensitive learning. Leveraging these results, we design principled deferral algorithms, MILD (Margin-based Imbalanced Learning to Defer), specifically suited for expert imbalance settings. Extensive experiments demonstrate the effectiveness of our approach, showing clear improvements over existing baselines on both image classification and real-world Large Language Model (LLM) routing tasks.

学习转接专家不均衡代价敏感学习LLM路由

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