This project explores hybrid quantum-classical models for enhanced classification of class-imbalanced datasets. It implements Hybrid Quantum Neural Networks (HQNNs) and quantum-enhanced algorithms (Random Forest, KNN, SVM) to improve minority class prediction.
Class imbalance is a common problem in real-world datasets where one class is significantly less represented than others. This project investigates the use of hybrid quantum-classical models to address this challenge, aiming to improve the prediction of minority classes.
The project is divided into two main parts:
-
Hybrid Quantum Neural Networks (HQNNs):
- Explores various HQNN architectures for classification.
- Utilizes Qiskit to implement quantum circuits within neural networks.
- Applies HQNNs to datasets like Credit Card Fraud, Melanoma Lesion Classification, Higgs Boson, and Breast Cancer Wisconsin.
- Focuses on techniques to integrate classical neural networks with quantum circuits for enhanced feature extraction and classification.
-
Quantum-Enhanced Classical Algorithms:
- Implements quantum versions of Random Forest, KNN, and SVM.
- Uses quantum circuits to compute kernel matrices or embed data.
- Evaluates these models on the Credit Card Fraud and Breast Cancer datasets.
- Compares the performance of these quantum-enhanced models with their classical counterparts.
The following datasets were used in this project:
- Credit Card Fraud Detection dataset
- Melanoma Lesion Classification dataset
- Higgs Boson dataset
- Breast Cancer Wisconsin (Diagnostic) dataset
- Qiskit
- PennyLane
- scikit-learn
- PyTorch
- B.S. Tejas (ED22B004) - HQNN part
- Kanishq Garg (ED22B051) - Quantum-Enhanced Classical Algorithms part
This project was completed for the course DA6300 - Quantum Computing And Machine Learning at IIT Madras.
HQNN.ipynb- Jupyter Notebook for the HQNN part of the project.qcml.ipynb- Jupyter Notebook for the Quantum-Enhanced Classical Algorithms part of the project.
Instructions on setting up the environment and running the code can be found within the individual notebook files.