Active Learning#
Label Sleuth guides the users through the labeling process by suggesting a ranked list of elements that they should label next. This list is generated through the use of an active learning component. Label Sleuth automatically invokes the active learning component every time a new model is trained. The system employs a customizable and extensible architecture, which allows different active learning approaches to be used.
Active learning strategies#
Label Sleuth currently supports the following active learning strategies:
Active learning strategy |
Description |
---|---|
|
Randomly sample from unlabeled data (active learning baseline) |
|
Uncertainty sampling/least confidence (Lewis and Gale 1994) |
|
Select top scored instances by the model (Ein-Dor et al. 2019) |
The employed active learning strategy can be configured by setting the value of the active_learning_strategy
parameter in the system’s configuration file.
Implement a new active learning strategy#
In addition to the preloaded active learning strategies, Label Sleuth can be extended to support additional active learning techniques. To integrate a new active learning approach, follow the steps below:
Implement a new
ActiveLearner
.Active learning modules inherit from the ActiveLearner API. The function to implement is:
def get_per_element_score(self, candidate_text_elements: Sequence[TextElement], candidate_text_element_predictions: Sequence[Prediction], workspace_id: str, dataset_name: str, category_name: str) -> Sequence[float]:
Given sequences of text elements and the model predictions for these elements, this function returns an active learning score for each element. The elements with the highest scores will be recommended for the user to label next.
Add the newly implemented ActiveLearner to the
ActiveLearningCatalog
.