Train and You'll Miss It: Interactive Model Iteration with Weak Supervision and Pre-Trained Embeddings

Abstract

Our goal is to enable machine learning systems to be trained interactively. This requires models that perform well and train quickly, without large amounts of hand-labeled data. We take a step forward in this direction by borrowing from weak supervision (WS), wherein models can be trained with noisy sources of signal instead of hand-labeled data. But WS relies on training downstream deep networks to extrapolate to unseen data points, which can take hours or days. Pre-trained embeddings can remove this requirement. We do not use the embeddings as features as in transfer learning (TL), which requires fine-tuning for high performance, but instead use them to define a distance function on the data and extend WS source votes to nearby points. Theoretically, we provide a series of results studying how performance scales with changes in source coverage, source accuracy, and the Lipschitzness of label distributions in the embedding space, and compare this rate to standard WS without extension and TL without fine-tuning. On six benchmark NLP and video tasks, our method outperforms WS without extension by 4.1 points, TL without fine-tuning by 12.8 points, and traditionally-supervised deep networks by 13.1 points, and comes within 0.7 points of state-of-the-art weakly-supervised deep networks-all while training in less than half a second.

References in corpus (14)

• Prototypical Networks for Few-shot Learning
• Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer
• HuggingFace's Transformers: State-of-the-art Natural Language Processing
• Language Models are Few-Shot Learners
• YFCC100M: The New Data in Multimedia Research
• Momentum Contrast for Unsupervised Visual Representation Learning
• ALBERT: A Lite BERT for Self-supervised Learning of Language Representations
• Snorkel: Rapid Training Data Creation with Weak Supervision
• Deep k-Nearest Neighbors: Towards Confident, Interpretable and Robust Deep Learning
• Learning to Learn from Weak Supervision by Full Supervision
• Overton: A Data System for Monitoring and Improving Machine-Learned Products
• Rekall: Specifying Video Events using Compositions of Spatiotemporal Labels
• Retrieval-Augmented Convolutional Neural Networks for Improved Robustness against Adversarial Examples
• Pairwise Feedback for Data Programming

Cited by in corpus (1)

• Interactive Weak Supervision: Learning Useful Heuristics for Data Labeling