arXiv:2502.06076cs.LG2025-02

用规划方法动态调整标注策略,显著降低标签成本。

A Planning Framework for Adaptive Labeling

  • 将标注过程建模为马尔可夫决策过程,按批次智能分配标注资源。
  • 单步前瞻策略性能超越常见启发式方法,证明规划有效性。
  • 提出低方差的Smoothed-Autodiff方法,适合深度学习标注场景。

真实标签对推动科学与工程应用至关重要,如评估干预效果或模型性能。由于随机采样标注代价高昂,本文提出一种自适应标注框架,支持分批重新分配测量资源。将问题建模为马尔可夫决策过程,随着每批标签的收集,后验信念更新(状态转移),而选择批次动作以最小化最终不确定性。设计的计算框架对多种不确定性量化方法(包括基于深度学习的方法)均兼容,并通过连续策略参数化支持多样化的策略梯度算法。在真实和合成数据集上,即使采用单步前瞻策略,也显著优于常见自适应标注启发式方法,凸显规划优势。方法层面,指出标准REINFORCE类策略梯度估计器因仅依赖零阶信息而存在高方差问题。提出基于平滑版非可微MDP的直接反向传播方法Smoothed-Autodiff,虽引入偏差但方差极低,理论与实证均表明该权衡有利。

原文摘要 · Abstract (English)

Ground truth labels/outcomes are critical for advancing scientific and engineering applications, e.g., evaluating the treatment effect of an intervention or performance of a predictive model. Since randomly sampling inputs for labeling can be prohibitively expensive, we introduce an adaptive labeling framework where measurement effort can be reallocated in batches. We formulate this problem as a Markov decision process where posterior beliefs evolve over time as batches of labels are collected (state transition), and batches (actions) are chosen to minimize uncertainty at the end of data collection. We design a computational framework that is agnostic to different uncertainty quantification approaches including those based on deep learning, and allows a diverse array of policy gradient approaches by relying on continuous policy parameterizations. On real and synthetic datasets, we demonstrate even a one-step lookahead policy can substantially outperform common adaptive labeling heuristics, highlighting the virtue of planning. On the methodological side, we note that standard REINFORCE-style policy gradient estimators can suffer high variance since they rely only on zeroth order information. We propose a direct backpropagation-based approach, Smoothed-Autodiff, based on a carefully smoothed version of the original non-differentiable MDP. Our method enjoys low variance at the price of introducing bias, and we theoretically and empirically show that this trade-off can be favorable.

自适应标注强化学习不确定性量化策略梯度

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