Voyages in sentence space

Imagine a sentence. “I went looking for adventure.”

Imagine another one. “I never returned.”

Now imagine a sentence gradient between them—not a story, but a smooth interpolation of meaning. This is a weird thing to ask for! I’d never even bothered to imagine an interpolation between sentences before encountering the idea in a recent academic paper. But as soon as I did, I found it captivating, both for the thing itself—a sentence… gradient?—and for the larger artifact it suggested: a dense cloud of sentences, all related; a space you might navigate and explore.

Here’s what a neural network instructed to produce such a cloud of sentences (specifically, sentences from science fiction) delivers when you ask it to draw a gradient between “I went looking for adventure.” and “I never returned.”

I won't acknowledge.
I stared after midnight
I stared agitated.
I get definitely.
I never returned.

You can ask it to draw a gradient of your own! Just replace the first and last sentences and use this button:

So, does that sentence gradient make sense? I honestly don’t know. Is it useful? Probably not! But I do know it’s interesting, and the larger artifact—the dense cloud, the sentence space—feels very much like something worth exploring.

A comfortable embedding

I’ve been exploring neural networks—in particular, neural networks that generate text—for a while now. (You can find a previous experiment here.)

When you’re tinkering with these tools, trying to produce something interesting (maybe even artful) from a dataset, whether it’s composed of text or images or something else, you often find yourself embedding that data into numeric space.

At a super simple level, imagine a dataset consisting of color swatches: rusty orange, dusty magenta, deep purple. You can see why it might make sense to embed these standalone swatches into a one-dimensional number line, a smooth sweep of color—

—so each has its own coordinate and there are also, as a significant added bonus, coordinates for all the intermediate colors between them.

Imagine a more complex dataset consisting of more colors. You can see how two dimensions might be useful:

Just like that, this dataset becomes something a neural network can chomp on, because it’s no longer color swatches described with metaphors, but a set of numbers. You know what computers love? SETS OF NUMBERS.

In practice, because datasets are often very rich—images of faces, sentences from science fiction stories—the numeric space into which you do this embedding will have not just two but dozens or hundreds of dimensions. I definitely can’t visualize a space like that—but so what? It turns out imaginary spaces are useful even if you can’t, in fact, imagine them.

Up above, embedding our color swatches into one or two dimensions was straightforward; the mapping was obvious. But how do we embed a face or a sentence into a numeric space with a hundred dimensions? How do we learn to map from “I went looking for adventure” to (-0.0036, -0.063, 0.014, …) and back?

One tool we can use is called a variational autoencoder. It’s a kind of neural network that learns to embed rich data into numeric space, and not only embed it, but “pack” it densely. A variational autoencoder, even more than nature, abhors a vacuum.

In academic papers about autoencoders (like this one) you’ll often see a diagram demonstrating how a dataset of celebrity faces has been embedded into numeric space. The paper will show smooth (and perhaps slightly unsettling) gradients between points in that space, each of which represents a unique face:

Here’s where things get interesting. In 2016, a paper called “Generating Sentences from a Continuous Space” published by Samuel R. Bowman, Luke Vilnis, et. al., showed that you can use a variational autoencoder to embed sentences into numeric space, and pioneered a few techniques to make it possible.

The paper also introduced, along the same lines as the unsettling celebrity gradient, the concept of a smooth homotopy, or linear interpolation, between sentences. I understood these immediately as sentence gradients and as soon as I read the paper… I had to have them.

Programming is hard

I tried to implement the paper myself. I failed. Even after corresponding with the authors, I just couldn’t get the basic autoencoding engine to work.

Lucky me: not even a year later, another paper appeared, extending the work of Bowman, et. al. “A Hybrid Convolutional Variational Autoencoder for Text Generation” by Stanislau Semeniuta, Aliaksei Severyn, and Erhardt Barth offered substantial additions to the idea and, even better: it offered THE CODE!!

(Let me just take a moment to praise researchers who publish their code. Without this project from Semeniuta, et. al., as a starting point, I would never have been able to explore these techniques. What a gift.)

Code in hand, I was well on my way to generating sentence gradients myself. I figured out the math to move through sentence space, implemented a few features to help organize experiments, added a simple server.

But there was a persistent problem: it ran too slow. I would write two sentences, ask the neural network to generate a gradient between them, and… wait. And wait and wait. Minutes passed. The process was too drawn out for experimentation, for exploration, for play.

Again, I tried to fix it myself. But I didn’t (and still don’t!) understand the innermost engine enough to see how I could speed up that process of moving sentences in and out of numeric space.

That’s when I asked for help.

The programmer Richard Assar’s implementation of a paper called SampleRNN, shared on GitHub, had impressed me with its usability and its speed. Sound generated by his code made its way into the audiobook of my latest novel. So, I reached out to him, asking, could I commission you to take a look at this sentence space project?

Richard said yes, and overnight—not literally overnight, but… basically overnight—he made it go faster by an order of magnitude. Now, you only wait a beat. Sentence one, sentence two, beat… gradient.

Working with this code shared by Semeniuta, et. al., streamlined by Richard Assar, what did I end up with?

Welcome to sentence space

My project sentence-space, now public on GitHub, provides an API that serves up two things:

  1. Sentence gradients: smooth interpolations between two input sentences.
  2. Sentence neighborhoods: clouds of alternative sentences closely related to an input sentence.

Sentence neighborhoods are simpler than gradients. Given an input sentence, what if we imagine ourselves standing at its location in sentence space, peering around, jotting down some of the other sentences we see nearby?

From the input

we get

The ship rose from the planet's farm surroundings.
The ship rose from the planet's great surface.
The ship rose from the planet's green corridor.
The ship rose from the rocks of a great mountain.
The ship rose from the planet's farm surface.

You can replace the sentence above with your own and

You can also increase or decrease the distance you peer into sentence space from your initial location; as you increase it, the results get more diverse. Adjust this slider, then use the button again:

Closer
to home

Further
afield

If you drag the slider fully to the left and look around, the results will all be identical, showing you the autoencoder’s best attempt at capturing your original sentence. Its reproduction is sometimes perfect; for example, try “The ship landed on the runway.” Don’t forget the period—it matters!

More often, the autoencoder returns something that seems… a bit… blurred? The effect gets stronger as your style and subject matter diverge from the autoencoder’s original dataset of sentences from science fiction. What you’re seeing is the transition from the richness of arbitrary text to the regularity of this particular sentence space. It’s very expressive—there are a lot of sentences in here to explore—but not infinitely so.

Anyway.

After I’d gotten this up and running, I felt something similar to what I remembered from an earlier machine learning project: a sense of, well, I did it… now what?

That feeling is an important waystation. Sentence gradients are weird; maybe nothing more than linguistic baubles. But I believe there’s something undeniably deep and provocative about this space packed full of language. Drawing gradients and exploring neighborhoods are just two ways of moving through it. How else might you travel?

I’ve published the code, which is mostly the work of Semeniuta, et. al., with important improvements by Richard Assar and a few embroideries by me.

Maybe you can imagine something different to do inside this science fiction sentence space, or maybe you’d rather establish a space all your own, built on sentences of your choosing. You could implement new operations; maybe you want to add sentences together or find the average of many sentences. These spaces are dense with meaning and difficult to wrap your head around, and to me, that’s a very attractive combination.

As before, this is all about making tools (that make language) for humans to use—never automatic, always interactive. Playful, surprising, destabilizing. The other day, I generated a gradient, and a sentence appeared that I haven’t stopped thinking about since:

“The information buzzed, emptying his lips.”

What a sequence of words! I’d never have written that on my own, and now I want to use it somewhere—a treasure smuggled out of sentence space.

Go explore. Send back reports of your progress.

Or stay here on this page and play a little.


Thanks to Dan Bouk for his feedback on a draft of this post. Dan wrote a book about how not just sentences but whole lives got plotted and gridded, smoothed and statisticized. How Our Days Became Numbered is essential reading.


February 2018, Berkeley


Really, the main thing to do here is sign up for my email newsletter. I try to make it feel like a note from a friend, and it’s very easy to unsubscribe. Plus, there are occasionally… SECRETS

What do you seek in these shelves?