In the previous session, we explored active learning query scenarios. We learned that the query scenario varies depending on the data type and environment of the problem to be solved. In this session, we will look into the criteria for finding a good active learning strategy.
What makes a good active learning strategy? Active learning is a technique that selects data samples with the greatest impact on model learning from unlabeled samples. Therefore, a good active learning strategy can be seen as one that effectively selects data that greatly contributes to model learning. Let’s think more about what it means to significantly help model learning.
To significantly help model learning means that by learning certain data, the model’s performance (accuracy) improves substantially. In other words, if the model learns data it hasn’t learned before, its accuracy would increase dramatically.
Let’s consider an example. Imagine there is an image classification model that distinguishes between apples, pears, and grapes. What would happen if we input a banana into this model? It will likely produce an incorrect result because the model has never been trained on banana data and doesn’t know what a banana is. Now, what if we added banana data to the model and retrained it? After retraining, if we input a banana, the model would correctly classify it as a banana.
In this way, data that the model has not yet learned and has a large error is considered data that significantly helps model learning. The criteria used to evaluate this data is called informativeness. Informativeness is measured by the model’s uncertainty. A representative metric for uncertainty is entropy, a concept derived from information theory. In the previous example, the banana image is the most informative image. To summarize, the higher the uncertainty, the more informative the data is, and thus, it should be prioritized.
The second criterion is representativeness. As the name suggests, the data selected by active learning should represent the overall characteristics of the entire dataset. For example, suppose you have 50 images of apples, pears, and grapes from the front, bottom, and top views. As shown in [Figure 2], if you train the model only with front-facing images, its performance is likely to be poor. A model trained with images from all views — front, bottom, and top — will perform better.
The last criterion is diversity. To reuse the previous example, in order to train a model that classifies apples, pears, and grapes, you need to train the model with a balanced mix of images of all three fruits. As shown in [Figure 3], if the model is only trained on images of pears, for example, it will only be able to classify pears and will fail to classify apples or grapes. In other words, the model needs to explore new areas that it hasn’t yet encountered.
So far, we have explored the three criteria for evaluating active learning. A good active learning strategy should be able to build a dataset that is highly informativeness, representativeness, and diversity. In the next session, we will dive into active learning strategies in more detail.
[Reference]
[1] D. Wu, “Pool-Based Sequential Active Learning for Regression,” in IEEE Transactions on Neural Networks and Learning Systems, vol. 30, no. 5, pp. 1348–1359, May 2019, doi: 10.1109/TNNLS.2018.2868649.
[2] Wang, K., Zhang, D., Li, Y., Zhang, R., & Lin, L. (2016). Cost-effective active learning for deep image classification. IEEE Transactions on Circuits and Systems for Video Technology, 27(12), 2591–2600.Wang, K., Zhang, D., Li, Y., Zhang, R., & Lin, L. (2016).