CLDec 19, 2022

Aggregating Soft Labels from Crowd Annotations Improves Uncertainty Estimation Under Distribution Shift

arXiv:2212.09409v35 citationsh-index: 43Has Code
AI Analysis

This addresses the challenge of selecting reliable training signals from noisy crowd annotations for practitioners, though it is incremental as it builds on existing soft-labeling methods.

This paper tackles the problem of improving uncertainty estimation for machine learning models under distribution shift by aggregating soft labels from crowd annotations, demonstrating that simple averaging yields consistent performance across tasks and improves uncertainty estimation in most settings while maintaining raw performance.

Selecting an effective training signal for machine learning tasks is difficult: expert annotations are expensive, and crowd-sourced annotations may not be reliable. Recent work has demonstrated that learning from a distribution over labels acquired from crowd annotations can be effective both for performance and uncertainty estimation. However, this has mainly been studied using a limited set of soft-labeling methods in an in-domain setting. Additionally, no one method has been shown to consistently perform well across tasks, making it difficult to know a priori which to choose. To fill these gaps, this paper provides the first large-scale empirical study on learning from crowd labels in the out-of-domain setting, systematically analyzing 8 soft-labeling methods on 4 language and vision tasks. Additionally, we propose to aggregate soft-labels via a simple average in order to achieve consistent performance across tasks. We demonstrate that this yields classifiers with improved predictive uncertainty estimation in most settings while maintaining consistent raw performance compared to learning from individual soft-labeling methods or taking a majority vote of the annotations. We additionally highlight that in regimes with abundant or minimal training data, the selection of soft labeling method is less important, while for highly subjective labels and moderate amounts of training data, aggregation yields significant improvements in uncertainty estimation over individual methods. Code can be found at https://github.com/copenlu/aggregating-crowd-annotations-ood.

Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes