Saved in:
Bibliographic Details
Main Authors: Tajs, Patryk, Skarupski, Mateusz, Rydzewski, Jakub
Format: Preprint
Published: 2025
Subjects:
Online Access:https://arxiv.org/abs/2505.16476
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908374366420992
author Tajs, Patryk
Skarupski, Mateusz
Rydzewski, Jakub
author_facet Tajs, Patryk
Skarupski, Mateusz
Rydzewski, Jakub
contents Unsupervised machine learning has recently gained much attention in the field of molecular dynamics (MD). Particularly, dimensionality reduction techniques have been regularly employed to analyze large volumes of high-dimensional MD data to gain insight into hidden information encoded in MD trajectories. Among many such techniques, t-distributed stochastic neighbor embedding (t-SNE) is particularly popular. A parametric version of t-SNE that employs neural networks is less commonly known, yet it has demonstrated superior performance in dimensionality reduction compared to the standard implementation. Here, we present a Python package called NeuralTSNE with our implementation of parametric t-SNE. The implementation is done using the PyTorch library and the PyTorch Lightning framework and can be imported as a module or used from the command line. We show that NeuralTSNE offers an easy-to-use tool for the analysis of MD data.
format Preprint
id arxiv_https___arxiv_org_abs_2505_16476
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle NeuralTSNE: A Python Package for the Dimensionality Reduction of Molecular Dynamics Data Using Neural Networks
Tajs, Patryk
Skarupski, Mateusz
Rydzewski, Jakub
Chemical Physics
Unsupervised machine learning has recently gained much attention in the field of molecular dynamics (MD). Particularly, dimensionality reduction techniques have been regularly employed to analyze large volumes of high-dimensional MD data to gain insight into hidden information encoded in MD trajectories. Among many such techniques, t-distributed stochastic neighbor embedding (t-SNE) is particularly popular. A parametric version of t-SNE that employs neural networks is less commonly known, yet it has demonstrated superior performance in dimensionality reduction compared to the standard implementation. Here, we present a Python package called NeuralTSNE with our implementation of parametric t-SNE. The implementation is done using the PyTorch library and the PyTorch Lightning framework and can be imported as a module or used from the command line. We show that NeuralTSNE offers an easy-to-use tool for the analysis of MD data.
title NeuralTSNE: A Python Package for the Dimensionality Reduction of Molecular Dynamics Data Using Neural Networks
topic Chemical Physics
url https://arxiv.org/abs/2505.16476