Стохастическое вложение соседей с t-распределением

Стохастическое вложение соседей с t-распределением (англ. t-Distributed Stochastic Neighbor Embedding, t-SNE) — метод визуализации данных высокой размерности с помощью представления каждой точки данных в двух или трехмерном пространстве, являющийся модификацией метода стохастического вложения соседей.

Пример работы методов t-SNE, Isomap, Sammon mapping, LLE на наборе данных MNIST

Стохастическое вложение соседей

Пусть стоит задача вложить множество точек в пространстве высокой размерности [math]\{x_i \mid x_i \in X\}[/math] в пространство низкой размерности. Обозначим множество точек в пространстве низкой размерности, которые получаются после вложения через [math]\{y_i \mid y_i \in Y\}[/math]. Стохастическое вложение соседей (англ. Stochastic Neighbor Embedding, SNE) конвертирует расстояния в Евклидовом пространстве высокой размерности между точками в условные вероятности [math]p_{j|i}[/math]. [math]p_{j|i}[/math] — вероятность, что точка [math]x_i[/math] выберет в качестве своего соседа точку [math]x_j[/math] среди остальных точек данных. Будем считать, что вероятность для точки [math]x_i[/math] найти соседа падает с увеличением расстояния от точки [math]x_i[/math] в соответствии с распределением Гаусса с нулевым математическим ожиданием и стандартным отклонением [math]\sigma_i[/math]. В соответствии с этим [math]p_{j|i}[/math] выражается как

[math]p_{j|i} = \dfrac{\exp{(-\dfrac{{\left\Vert x_i - x_j \right\Vert}^2}{2\sigma_i^2})}}{\sum\limits_{k \neq i}\exp{(\dfrac{{-\left\Vert x_i - x_k \right\Vert}^2}{2\sigma_i^2})}}[/math].

Теперь определим похожие вероятности [math]q_{i|j}[/math] для пространства низкой размерности, куда вкладываются точки пространства высокой размерности.

[math]q_{j|i} = \dfrac{\exp{(-{\left\Vert y_i - y_j \right\Vert}^2)}}{\sum\limits_{k \neq i}\exp{({-\left\Vert y_i - y_k \right\Vert}^2)}}[/math].

Данные вероятности получаются из тех же самых предложений, что были сделаны для пространства высокой размерности, за исключением того, что все распределения Гаусса имеют стандартное отклонение [math]\dfrac{1}{\sqrt{2}}[/math] для всех точек.

Если удастся хорошо вложить одно пространство в другое, [math]p_{i|j}[/math] должны стать похожими на [math]q_{i|j}[/math]. В связи с этим SNE пытается уменьшить разницу в распределении вероятностей. Стандартной мерой для измерения различия вероятностей служит дивергенция Кульбака-Лейблера:

[math]KL(P \Vert Q) = \sum\limits_j p_j \log_2 \dfrac{p_j}{q_j}[/math].

В данном случае имеем [math]|X|[/math] распределений. Тогда целевую функцию, которую будем оптимизировать, определим как сумму соответствующих дивергенций Кульбака-Лейблера. То есть:

[math]C = \sum\limits_i KL(p_i \Vert q_i) = \sum\limits_i \sum\limits_j p_{j|i} \log_2 \dfrac{p_{j|i}}{q_{j|i}}[/math].

Дивергенция Кульбака-Лейблера не является симметричной мерой, поэтому, например, вложение близких точек в удаленные даёт гораздо большее значение ошибки, чем вложение далеких точек в близкие. Другими словами, целевая функция нацелена на сохранение локальной структуры вокруг точек.

Параметры [math]\sigma_i[/math] подбираются следующим образом. Каждое значение параметра порождает свое распределение вероятностей [math]P_i[/math]. Это распределение имеет энтропию [math]H(P_i) = \sum\limits_j p_{j|i}\log_2 p_{j|i} [/math], которая возрастает с ростом [math]\sigma_i[/math]. В самом алгоритме [math]\sigma_i[/math] вычисляются с помощью вещественного двоичного поиска по заранее заданной пользователем величине, называемой перплексией: [math]Perp(P_i) = 2 ^ {H(P_i)}[/math].

Изначально точки [math]y_i[/math] сэмплируют в пространстве низкой размерности в соответствии с распределением Гаусса с маленьким стандартным отклонением с математическим ожиданием в нуле, далее идет оптимизация целевой функции. Она проводится методом градиентного спуска. Градиент равен:

[math]\dfrac {\delta C} {\delta y_i} = 2 \sum\limits_j (p_{j|i} - q_{j|i} + p_{i|j} - q_{i|j})(y_i - y_j)[/math]

Физическая интерпретация

Есть следующая физическая интерпретация модели. В пространстве низкой размерности натянуты пружины между каждой парой точек [math]y_i[/math] и [math]y_j[/math], действующие в направлении [math]y_i - y_j[/math]. Пружины могут притягивать или отталкивать точки в зависимости от расстояния между ними. Сила, прикладываемая пружиной, пропорциональна её длине [math]\left\Vert y_i - y_j \right\Vert[/math] и жесткости [math]p_{j|i} - q_{j|i} + p_{i|j} - q_{i|j}[/math]. Оптимизация функционала в данной интерпретации эквивалентна поиску положения точек, в котором будет наблюдаться равновесие сил.

Симметричное стохастическое вложение соседей

Следующая модификация SNE носит название симметричное стохастическое вложение соседей (англ. Symmetric Stochastic Neighbor Embedding, Symmetric SNE), которая будет использоваться дальше в t-SNE. Симметричный SNE в качестве альтернативы использует совместные вероятности вместо условных. Теперь:

[math]C = KL(P \Vert Q) = \sum\limits_i \sum\limits_j p_{i j} \log_2 \dfrac {p_{i j}} {q_{i j}}[/math].

Очевидным образом можно определить [math]q_{i j}[/math]:

[math]q_{i j} = \dfrac {\exp ({ -{\left\Vert y_i - y_j \right\Vert}^2 }) } {\sum\limits_{k \neq l} \exp ({ -{\left\Vert y_k - y_l \right\Vert}^2) } }[/math],

но то же решение для [math]p_{i j}[/math] привело бы к проблеме, что для выброса [math]x_i[/math] [math]p_{i j}[/math] будет очень маленькой для любого [math]x_j[/math]. Таким образом, будет почти нулевой соответствующая дивергенция Кульбака-Лейблера для любого распределения [math]q_{i j}[/math]. Это означало бы, что положение точки [math]y_i[/math] определялось бы очень неточно относительно положения других точек и не было бы особой разницы в том, где она расположена.

Таким образом, в симметричном SNE в качестве [math]p_{i j}[/math] рассматривается следующая величина:

[math]p_{i j} = \dfrac {p_{i|j} + p_{j|i} } {2|X|}[/math].

Очевидный плюс в том, что [math]\sum\limits_j p_{i j} \gt \dfrac 1 {2|X|}[/math] для всех точек, что хорошо скажется на выбросах. А также теперь [math]p_{i j} = p_{j i}[/math], [math]q_{i j} = q_{j i}[/math].

Авторы утверждают, что симметричный SNE вкладывает данные в пространство низкой размерности почти так же как и ассиметричный, а иногда даже лучше.

Градиент при таком подходе принимает вид: [math]\dfrac {\delta C} {\delta y_i} = 2 \sum\limits_j (p_{i j} - q_{i j})(y_i - y_j)[/math].

Проблема скученности

Необходимо понимать, что невозможно абсолютно точно моделировать расстояния между точками пространства высокой размерности в низком. Например, в десятимерном пространстве существует [math]11[/math] равноудаленных друг от друга точек, в то время как на плоскости может быть максимум [math]3[/math] равноудаленные точки.

При использовании обычного SNE возникает проблема, которая вытекает из разного распределения вероятностей в пространствах высокой и низкой размерностей. Пусть есть некоторое пространство высокой размерности. Пусть точки [math]x_i[/math] равномерно распределены в нем вокруг точки [math]x_0[/math] в некотором шаре с радиусом [math]R[/math]. Заметим, что, чем больше размерность пространства, тем больше точек попадет рядом с границей шара, поэтому количество близких к [math]x_0[/math] точек с ростом размерности будет убывать. Теперь попытаемся вложить данное пространство в плоскость. Пусть точки [math]x_i[/math] перешли в точки [math]y_i[/math] на плоскости. Заметим, что если попытаться вложить точки [math]x_i[/math] в круг радиуса [math]R[/math] с центром в точке [math]y_0[/math] образуется большое количество маленьких расстояний между точками [math]y_i[/math], т.к. объем сферы в высокомерном пространстве несопоставим с площадью круга на плоскости. Таким образом, если мы хотим правильно моделировать маленькие расстояния на плоскости, следовало бы поместить удаленные от [math]x_0[/math] точки [math]x_i[/math] ещё дальше, чем в исходном пространстве. Но в таком случае, вспоминая физическую интерпретацию, на соответствующие им точки [math]y_i[/math] будет действовать небольшая сила притяжения к точке [math]y_0[/math]. Принимая во внимание, что точек наподобие [math]x_0[/math] в реальной выборке данных будет достаточно много, их пружины вместе образуют силу, что сожмет все точки в нуле и будет мешать образованию кластеров.

Стохастическое вложение соседей с t-распределением

Чтобы избежать проблемы скученности, было решено использовать в пространстве низкой размерности t-распределение Стьюдента с одной степенью свободы вместо распределения Гаусса. Данное распределение очень похоже на распределение Гаусса, но имеет большую вероятностную массу на участках, отдаленных от нуля (Рис. 2.), что решает описанную выше проблему, т.к. теперь удаленные точки лучше отталкиваются.

Рис. 2. Сравнение плотностей нормального распределения (синий цвет) и t-распределения с одной степенью свободы (красный цвет)

В связи с заменой распределения [math]q_{i j}[/math] определяется следующим образом:

[math]q_{i j} = \dfrac {(1 + {\left\Vert y_i - y_j \right\Vert}^2)^{-1}} {\sum\limits_{k \neq l} (1 + {\left\Vert y_k - y_l \right\Vert}^2)^{-1}}[/math].

Еще одно свойство данного распределения состоит в том, что [math](1 + {\left\Vert y_i - y_j \right\Vert}^2)^{-1}[/math] описывает закон обратных квадратов для далеких точек в пространстве низкой размерности, что позволяет думать не об отдельных точках, а о кластерах, которые будут взаимодействовать между собой как отдельные точки.

После замены распределения изменился градиент целевой функции, теперь он равен:

[math]\dfrac {\delta C} {\delta y_i} = 4 \sum\limits_j (p_{i j} - q_{i j})(y_i - y_j)(1 + {\left\Vert y_i - y_j \right\Vert}^2)^{-1}[/math].

Оптимизации t-SNE

В t-SNE используется 2 основные оптимизации:

1. Первая оптимизация называется "раннее сжатие". В данной оптимизации на ранних итерациях оптимизации к целевой функции добавляется [math]L_2[/math]-штраф на расстояния в пространстве низкой размерности, что влечет за собой сжатие всех точек в нуле. В связи с этим кластерам будет легче переходить друг через друга, чтобы правильно расположиться в пространстве.
2. Вторая оптимизация называется "раннее преувеличение". В данной оптимизации на ранних итерациях [math]p_{i j}[/math] умножаются на некоторое положительное число, например на [math]4[/math]. Так как [math]q_{i j}[/math] остаются теми же самыми, они слишком маленькие, чтобы моделировать соответствующие [math]p_{i j}[/math]. Как следствие, образуются очень плотные кластера, которые широко раскиданы в пространстве низкой размерности. Это создает много пустого пространства, которое используется кластерами, чтобы легко менять и находить наилучшее взаимное расположение.

Рис. 3. Визуализация работы t-SNE

На Рис. 3 представлена визуализация работы t-SNE, на которой видны эффекты от применения приведенных выше оптимизаций.

См. также

• Уменьшение размерности
• Метод главных компонент (PCA)
• Стохастический градиентный спуск

Источники информации

1. Laurens van der Maaten and Geoffrey Hinton — Visualizing Data using t-SNE
2. datareview.info — Алгоритм t-SNE. Иллюстрированный вводный курс
3. Wikipedia — Multivariate t-distribution