A Light-weight CNN Model for Efficient Parkinson's Disease Diagnostics

Xuechao Wang,Junqing Huang,Marianna Chatzakou,Kadri Medijainen,Pille Taba,Aaro Toomela,Sven Nomm,Michael Ruzhansky
DOI: https://doi.org/10.48550/arXiv.2302.00973
2023-02-02
Abstract:In recent years, deep learning methods have achieved great success in various fields due to their strong performance in practical applications. In this paper, we present a light-weight neural network for Parkinson's disease diagnostics, in which a series of hand-drawn data are collected to distinguish Parkinson's disease patients from healthy control subjects. The proposed model consists of a convolution neural network (CNN) cascading to long-short-term memory (LSTM) to adapt the characteristics of collected time-series signals. To make full use of their advantages, a multilayered LSTM model is firstly used to enrich features which are then concatenated with raw data and fed into a shallow one-dimensional (1D) CNN model for efficient classification. Experimental results show that the proposed model achieves a high-quality diagnostic result over multiple evaluation metrics with much fewer parameters and operations, outperforming conventional methods such as support vector machine (SVM), random forest (RF), lightgbm (LGB) and CNN-based methods.
Machine Learning
What problem does this paper attempt to address?
This paper attempts to solve several key problems in the diagnosis of Parkinson's Disease (PD): 1. **Improve diagnostic efficiency**: By proposing a lightweight neural network model, combining Convolutional Neural Network (CNN) and Long Short - Term Memory (LSTM), to efficiently diagnose Parkinson's disease. This model aims to reduce the number of parameters and computational operations while maintaining high diagnostic accuracy. 2. **Utilize hand - drawn test data**: A series of hand - drawn data are used in the paper to distinguish between Parkinson's disease patients and healthy control groups. These hand - drawn data are collected from digital devices (such as iPad Pro) and can capture subtle tremor signals, thus reflecting the tremor degree of patients during fine - movement tasks. 3. **Solve the problem of small datasets**: Since the hand - drawn datasets for Parkinson's disease are usually small, the paper adopts data splitting techniques and generates more data samples through the sliding - window method to alleviate the training problems caused by small datasets. 4. **Optimize feature extraction and selection**: The paper describes in detail the data pre - processing steps, including normalization, feature engineering and selection. In particular, by calculating velocity features (such as \( v_t^x=\frac{x_t - x_{t - 1}}{\delta t} \) and \( v_t^y=\frac{y_t - y_{t - 1}}{\delta t} \)), the non - stationary coordinate features are transformed into stationary features, thereby improving the classification effect. 5. **Verify model performance**: Through comparative experiments with traditional machine learning methods (such as Support Vector Machine (SVM), Random Forest, LightGBM, etc.) and other deep learning models (such as Multi - Layer Perceptron (MLP), AlexNet, etc.), the superior performance of the proposed model on multiple evaluation metrics is verified, including accuracy, precision, recall, F1 - score and Matthews Correlation Coefficient (MCC). In summary, the main goal of this paper is to develop an efficient and accurate diagnostic model for Parkinson's disease, using hand - drawn test data and achieving this goal through lightweight deep - learning methods.