Light Bulb 💡: Machine learning (ML) made easy.

Go directly to the Github project 

.

Let’s say you want to build a photo-sharing app called SnapCat, that only allows users to send pictures of cats, and nothing else.

How would you go about starting this? It’ll probably looks something like this:

• Collect a large set of cat and not cat photos.

• Manually label the posts as cat or not cat.

• Split the dataset into train, test, and validation sets.

• Train some model (lets say a

  ) on the dataset.

• Look at the accuracy on the test set, if it’s not good enough, then go back and rethink each step.

• Save the model weights, and load them into some web backend, to start classifying new posts.

Let’s see how Light Bulb can help you with all of this.

Light Bulb is a service that will integrate this end to end, from labeling, all the way through to production. First define this config:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
# config/cat_not_cat.ymltask:
  title: Is this a cat?
dataset:
  directory: dataset/cat_not_cat/
  data_type: images
  judgements_file: outputs/cat_not_cat/labels.csv
label:
  type: classification
  classes:
    - Cat
    - Not Cat
model:
  directory: outputs/cat_not_cat/models/
user: chris

Then from the root of the app run:

1
2
3
make
make dataset/cat_not_cat
.virt/bin/python code/server.py --config config/cat_not_cat.yml

Which will start a server on

http://localhost:5000

As you label a few entries, you’ll see the Training icon change from No to Yes. This means that a model is actively training on the newly labeled posts. As you label more posts, the model gets smarter.

After labeling more images (and giving the model some time to train), you’ll see an Accuracy statistic, that shows you how well the model is doing. In this case our model is getting about 87% accuracy, which is pretty amazing since we only labeled 78 images.

Now that our model is trained, it’ll start helping us label more data 😲. Light Bulb will:

• Go through the dataset, and labeling images that it feels fairly confident about (97% confident in fact).

• Store the labels the model assigns and present them to you in batches. All you have do is confirm the labels are correct.

Verifying the automatically labeled images

With this batch labeling feature, I managed to label all 2000 images in just 15 minutes.

Light Bulb also exposes an API for your model. You can easily issue API requests to the server, and score new images. Let’s see what our model thinks about

.

1
2
3
curl --header "Content-Type: application/json" \
     --request POST \
     --data '{"type": "images","urls": ["https://github.com/czhu12/light-bulb/raw/master/docs/images/cat-image-1.jpg"]}' \ http://localhost:5000/score

Which returns:

1
2
3
4
5
6
7
8
9
10
11
12
{
  "labels": [
    "Cat",
    "Not Cat"
  ],
  "scores": [
    [
      0.9971857666969299, # Our model thinks its 99% a cat!
      0.0028141846414655447
    ]
  ]
}

And now lets try something thats 

:

1
2
3
curl --header "Content-Type: application/json" \
     --request POST \
     --data '{"type": "images","urls": ["https://raw.githubusercontent.com/czhu12/light-bulb/master/docs/images/not-cat-image-1.jpg"]}' \ http://localhost:5000/score

Our model returns:

1
2
3
4
5
6
7
8
9
10
11
12
{
  "labels": [
    "Cat",
    "Not Cat"
  ],
  "scores": [
    [
      0.007293896283954382,
      0.9927061200141907 # Our model thinks this is 99.2% not a cat!
    ]
  ]
}

Most deep learning tasks can be framed as an encoder — decoder architecture.

For all image tasks we use a

encoder pre-trained on

.

Image Classification: Image classification is done with a CNN based encoder, that is fed into a multi-layer perceptron decoder.

Object Detection (work in progress): Object detection can be framed as a CNN based encoder, that is fed into a regression decoder. For object detection, the decoder will be a pre-trained 

head.

For all text tasks, we use a 3 layer LSTM encoder pre-trained as a language model on Wikitext-103.

Text Classification

: Text classification can be framed as an LSTM encoder that outputs into a logistic regression decoder.

Sequence Tagging (work in progress): Sequence tagging can be framed as an LSTM encoder, where at each time-step, the output is fed into a 

model.

Light Bulb uses a few tricks to train a model as fast as efficiently as possible.

When Light Bulb decides which post to show you to label next, it chooses based on a process known as maximum entropy sampling. Before Light Bulb shows you an image to label, it’ll first try to make a prediction. Lets say one image was scored as

95% cat, 5% not cat

, and another image was scored as 50

% cat, 50% not cat. 

Which one should Light Bulb show you next? Intuitively, the second image should be labelled next, since that seems to be the one the model isn’t quite sure about. This way, we don’t waste any of your labels!

Light Bulb will leverage state of the art semi supervised learning, and pre-training. One of the reasons why deep learning is so powerful, is because of its unique ability to transfer knowledge from one task to another (see:

,

). For instance, a model that is good at predicting the next word in a sentence, ie: a language model, will also be good at classifying the sentiment of the sentence. Likewise, a model that is good at the

, will also be good at predicting cat vs not cat, with small tweaks of course.

Light Bulb also leverages semi supervised learning techniques to learn as much as possible from the dataset you provide. In the cat vs not cat dataset above, there were around 2000 images, but we only labeled about 80 of them. But that doesn’t mean that we can’t learn from the other 1920 images! Light Bulb uses all the images in the dataset,

even if you haven’t labeled them

, to train the model, by fine-tuning an auto-encoder, which will essentially learn the general properties of all the 2000 images.

For full details check out the

.

Any questions, suggestions, bugs, or just want to reach out, PM me on 

!

Happy Labeling!