There are 3 kinds of machine learning approaches- Supervised, Unsupervised, and Reinforcement Learning techniques. Supervised learning as we know is where data and labels are present. Unsupervised Learning is where only data and no labels are present. Reinforcement learning is where the agents learn from the actions taken to generate rewards.
Imagine a situation where for training there is less number of labelled data and more unlabelled data. A new technique called Semi-Supervised Learning(SSL) which is a mixture of both supervised and unsupervised learning. As the name suggests, semi-supervised learning has a set of training data which is labelled and another set of training data, which is unlabelled. We can think of this situation as when Google photos or Facebook identify people in the picture by their faces(data) and generate a suggested name(label) based on the previously stored images of that person.
Source – Link
In this article, I’ll be discussing how to generate pseudo labels using the semi-supervised learning technique.
Pseudo labelling is the process of using the labelled data model to predict labels for unlabelled data. Here at first, a model has trained with the dataset containing labels and that model is used to generate pseudo labels for the unlabelled dataset. Finally, both the datasets and labels(original labels and pseudo labels) are combined for a final model training. It is called pseudo(which means unreal) as these may or may not be real labels and we are generating them based on a similar data model.
Source – Link
Implementation in Python
For this demonstration, I’ve taken up the sklearn dataset breast cancer. I know that it already contains labels but we are going to modify it by splitting the data into two parts one having labels and the other with no labels. We’ll generate our own labels for the unlabelled data from the labelled data model that has been trained and then finally use both to train a final model.
Breast cancer dataset is a classification problem to predict whether the cancer is benign(B) or malignant(M). First two columns being 1)id and 2)diagnosis(target)
Feature set contains :
a) radius_mean (mean of distances from the centre to points on the perimeter)
b) texture_mean (standard deviation of gray-scale values)
e) smoothness_mean (local variation in radius lengths)
f) compactness_mean (perimeter^2 / area – 1.0)
g) concavity_mean (severity of concave portions of the contour)
h) concave points_mean (number of concave portions of the contour)
import pandas as pd import numpy as np from sklearn.model_selection import train_test_split from sklearn.datasets import load_breast_cancer from sklearn.ensemble import RandomForestClassifier
X,y = load_breast_cancer(True) X.shape
Splitting dataset into data with labels and data with no labels in 40:60 ratio
x_train,x_test,y_train,_ = train_test_split(X,y,test_size=.6) x_train.shape,y_train.shape,x_test.shape
((227, 30), (227,), (342, 30)
Model Creation and fitting the data containing labels
model1 = RandomForestClassifier() history = model1.fit(x_train,y_train) history
RandomForestRegressor(bootstrap=True, ccp_alpha=0.0, criterion=’mse’,
max_depth=None, max_features=’auto’, max_leaf_nodes=None,
n_estimators=100, n_jobs=None, oob_score=False,
random_state=None, verbose=0, warm_start=False)
Accuracy score for data-label model training
Now we use this model to predict labels (called pseudo labels) for no label data
y_new = model1.predict(x_test) y_new.shape
We concatenate both these datasets now
final_X = np.concatenate((x_train,x_test)) final_X.shape
Similarly both labels(original and pseudo) are also concatenated.
final_Y = np.concatenate((y_train,y_test)) final_Y.shape
Final model containing entire dataset is fitted and accuracy score is generated
model2 = RandomForestRegressor() model2.fit(final_X,final_Y) model2.score(final_X,final_Y)
This was the implementation of pseudo labelling. I’ve modified an existing dataset with labels but it can similarly be used in real-world data science scenarios to achieve pseudo labels for unlabelled data from labelled data. Semi-supervised learning has now gained much attention in both classical machine learning problems as well as deep learning.
The complete code of the above implementation is available at the AIM’s GitHub repository. Please visit this link to find the notebook of this code.
Subscribe to our NewsletterGet the latest updates and relevant offers by sharing your email.
Machine learning and data science enthusiast. Eager to learn new technology advances. A self-taught techie who loves to do cool stuff using technology for fun and worthwhile.