Quanvolutional Autoencoder for synthesising Exoplanet transits

This abstract has open access
Problem description and relevance

Identifying exoplanets from light curves has become a primary method for exoplanet detection, with many deep learning classification models having been applied to exoplanetary transit data. 


Due to the difficulty of detecting exoplanets, mission data is heavily biased towards non-exoplanet light curves. As most machine learning based classification tasks assume an equal distribution of classes, the lack of data could decrease the performance of vetting processes (Leevy et al. 2018; Pratyush & Gangrade 2021) . The ability to generate synthetic exoplanetary data could therefore be highly beneficial for models using exoplanetary data, such as exoplanet classification, and could help mitigate issues such as overfitting and poor representation of the data features. 


We consider the ability of a 1D hybrid quantum-classical autoencoder (QAE) and an equivalent classical autoencoder trained on synthetic exoplanet candidate light curves (Fuentes & Solar 2024) to generate exoplanetary transit light curves samples to reduce the data imbalance. We also investigate how the model can be applied to the task of exoplanet classification through reconstruction error. 

Submission ID :
42
Submission Topics
Methodology :

We look at quantum convolutional (quanvolutional) neural networks (QNNs), extending the quanvolutional layer to produce a hybrid QAE. Autoencoders are a type of unsupervised feed-forward ANN that are designed to deconstruct input data and encode it into a compressed representation of the original data. The quanvolutional neural network is an architecture by Henderson et al (Henderson et al. 2020) inspired by the convolutional neural network (CNN). Convolutional layers are made of stacks of convolutional filters which convolve over the input to produce feature maps, each map containing only the useful features of the data. The quanvolutional neural network seeks to leverage aspects of the convolutional layer to create a 'quanvolutional layer' run on a quantum circuit. As with convolutional layers, quanvolutional layers are made of filters which operate over the input. These filters extract information by transforming local data via quantum circuits, which can be random circuits, or designed with a specific architecture.


The original concept of a QNN is that the quanvolutional layer would act as an initial layer within the overall classical CNN architecture. Like a convolutional layer, the quanvolutional layer has parameters that can be defined including the number of filters, the number of layers stacked, the quantum encoding and decoding method, and the quantum circuit itself. The flexibility of the layer makes it generalisable to many tasks. However, the rotational parameters within the quantum circuit are kept constant, meaning the quantum contribution of the model is not a part of the training. We chose to explore both keeping the layer as a static pre-processing step, prior to the autoencoder architecture, and as a parameterised quantum circuit (PQC) with trainable parameters that are optimised with the classical autoencoder parameters. 


We used a simple model architecture. Input data is encoded through angle encoding and is passed through the quanvolutional layer, which halves the dimensions with an output of a collection of feature maps, which are flattened and passed through a fully-connected layer to reduce the dimensions further. The number of filters the quanvolutional layer produces is equivalent to the number of qubits in the circuit. For our study, we chose to use four qubits, producing four channels.


We investigated three different circuit architectures, a randomly generated circuit, as used by Henderson et al., a circuit which consists solely of CNOTs to introduce entanglement, and a hardware-efficient ansatz, with one model containing fixed rotational gates and the other with trainable parameters, which are optimised along with the classical parameters. We compared the loss and reconstruction error for each model, including the classical equivalent, where the quanvolutional layer is swapped for an equivalent convolutional layer. The training was repeated 50 times with a different random seed for each run and we plot the average train and validation loss per epoch, the reconstruction error, which is an indicator of how well the model is learning the important data features, and their respective standard deviation. We also visually inspect the generated data samples.

Practical demonstration :

We implemented the classical part of our model using the Pytorch (Version 2.6) library. The quantum side of the model was designed using Pennylane (Bergholm et al. 2018), a quantum machine learning library in which it is possible to implement a PQC which can be trained via gradient descent. For the quantum computer simulation we ran all our models on, we employed the use of Pennylane's default_qubit device. All models were trained on a high-performance computer and the training was repeated 50 times with a different random seed for each run to gauge how much random chance affects the results in order to produce a more robust conclusion.

Application potential :

We explore a hybrid quantum-classical autoencoder in which only local subsections of the input tensor are processed by the quantum circuit, so the required number of qubits can remain low. For example, a quanvolutional filter of size 4 would only require 4 qubits, making QNNs a promising candidate as an application in the Noisy Intermediate-Scale Quantum (NISQ) era, where the number of qubits and noise can be a limitation. We also use shallow circuits, which will be less affected by noise. The classical part of the model takes on a larger portion of the workload, including the optimisation and reconstruction, while the quanvolutional layer contributes to how the model learns the essential features.

Associated Sessions

PhD candidate
,
University of Hull
University of Hull
7 visits