Focused Stochastic Neighbor Embedding for Better Preserving Points of Interest
Focused Stochastic Neighbor Embedding for Better Preserving Points of Interest
复制标题
DOI:
10.1109/bdcat56447.2022.00043
复制
发表时间:
2022-12
期刊:
影响因子:
--
通讯作者:
Rafael Baez Ramirez;Sanuj Kumar;Tuan M. V. Le;H. Cao
中科院分区:
文献类型:
--
作者:
Rafael Baez Ramirez;Sanuj Kumar;Tuan M. V. Le;H. Cao
Dimensionality reduction aims to find low-dimensional embeddings of high-dimensional data such that the low-dimensional representation preserves some meaningful properties of structures in the original data. When low-dimensional space is 2- or 3-dimensional, the low-dimensional embeddings can be visualized using a scatterplot map. Most of the existing methods try to preserve the local neighborhoods of all data points. However, in general, it is impossible to retain all such information for all data points in the low-dimensional space. As a result, there could be some data points whose neighborhoods are not faithfully displayed in the visualization due to information loss. If the information loss happens around a specific set of points of interest (e.g., specific patients, or proteins under observed), this may be problematic because the withdrawn insights may not be accurate for these observed data points. Therefore, in this paper, we introduce a problem called focused dimensionality reduction where given an original high-dimensional dataset and a set of points of interest, we want to find 2- or 3-dimensional embeddings of the original data such that the information loss in the local neighborhoods surrounding the points of interest is minimized as much as possible. In other words, if the information loss is inevitable, it should not happen around the points of interest. To solve the problem, we extend the stochastic neighbor embedding method and introduce a focused objective function where we put more weight on losses that involve points of interest. Experiments on real-world datasets show that our proposed method is better in preserving the local neighborhood structure of points of interest while the generated visualizations are as good as those generated by the stochastic neighbor embedding method.