Transformer-Based Brainwave Image Reconstruction (2022)

Nikita Lokhmachev

ML Engineer
Software Engineer
AI Developer
Python
PyTorch
Reconstructing images from brainwave data is a major challenge with the potential to reveal deeper insights into brain function. Advances in transformers, diffusion models, and variational autoencoders have transformed art and image generation, as seen with tools like DALL-E, Parti, and Stable Diffusion. Since text is treated as a time series, this idea can be extended to other types, such as audio or brainwave data. In this project, I implemented a minimal transformer-based Parti architecture to convert time series data into images (ts2img). I built autoencoders (AE), variational autoencoders (VAE), and vector-quantized VAEs from scratch, explored multivariate time-series transformers, and tested the pix2pix architecture by converting time series into spectrogram images.
Partner With Nikita
View Services

More Projects by Nikita