LGAIMLJul 5, 2021

End-to-End Weak Supervision

arXiv:2107.02233v347 citations
Originality Incremental advance
AI Analysis

This addresses the data-labeling bottleneck in machine learning applications, offering a more integrated solution.

The paper tackles the problem of aggregating weak supervision sources for data labeling by proposing an end-to-end approach that directly learns the downstream model, improving performance and robustness over prior methods.

Aggregating multiple sources of weak supervision (WS) can ease the data-labeling bottleneck prevalent in many machine learning applications, by replacing the tedious manual collection of ground truth labels. Current state of the art approaches that do not use any labeled training data, however, require two separate modeling steps: Learning a probabilistic latent variable model based on the WS sources -- making assumptions that rarely hold in practice -- followed by downstream model training. Importantly, the first step of modeling does not consider the performance of the downstream model. To address these caveats we propose an end-to-end approach for directly learning the downstream model by maximizing its agreement with probabilistic labels generated by reparameterizing previous probabilistic posteriors with a neural network. Our results show improved performance over prior work in terms of end model performance on downstream test sets, as well as in terms of improved robustness to dependencies among weak supervision sources.

Code Implementations1 repo
Foundations

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

Your Notes