Timezone: »

 
Accelerating Batch Active Learning Using Continual Learning Techniques
Gantavya Bhatt · Arnav M Das · · Rui Yang · Vianne Gao · Jeff Bilmes

@
in DMLR Workshop: Data-centric Machine Learning Research »

A major problem with Active Learning (AL) is high training costs since models are typically retrained from scratch after every query round. We start by demonstrating that standard AL on neural networks with warm starting fails, both to accelerate training and to avoid catastrophic forgetting when using fine-tuning over AL query rounds. We then develop a new class of techniques, circumventing this problem, by biasing further training towards previously labeled sets, thereby complementing existing work on AL acceleration. We accomplish this by employing existing, and developing novel, replay-based Continual Learning (CL) algorithms that are effective at quickly learning the new without forgetting the old, especially when data comes from an evolving distribution. We call this paradigm "Continual Active Learning" (CAL). We show CAL achieves significant speedups using a plethora of replay schemes that use model distillation and that select diverse/uncertain points from the history.We conduct experiments across many diverse data domains, including natural language, vision, medical imaging, and computational biology, each with very different neural architectures (transformers/CNNs/MLPs) and dataset sizes. CAL consistently provides a 3x reduction in training time, while retaining performance and out-of-distribution robustness, showing its wide applicability.

Author Information

Gantavya Bhatt (University of Washington)
Arnav M Das (University of Washington)
Rui Yang (MSKCC)
Vianne Gao (Memorial Sloan Kettering)
Jeff Bilmes (UW)

More from the Same Authors