Developing Sustainable Classification of Diseases via Deep Learning and Semi-Supervised Learning

Abstract

Disease classification based on machine learning has become a crucial research topic in the fields of genetics and molecular biology. Generally, disease classification involves a supervised learning style; i.e., it requires a large number of labelled samples to achieve good classification performance. However, in the majority of the cases, labelled samples are hard to obtain, so the amount of training data are limited. However, many unclassified (unlabelled) sequences have been deposited in public databases, which may help the training procedure. This method is called semi-supervised learning and is very useful in many applications. Self-training can be implemented using high- to low-confidence samples to prevent noisy samples from affecting the robustness of semi-supervised learning in the training process. The deep forest method with the hyperparameter settings used in this paper can achieve excellent performance. Therefore, in this work, we propose a novel combined deep learning model and semi-supervised learning with self-training approach to improve the performance in disease classification, which utilizes unlabelled samples to update a mechanism designed to increase the number of high-confidence pseudo-labelled samples. The experimental results show that our proposed model can achieve good performance in disease classification and disease-causing gene identification.

A Lightweight Convolutional Neural Network Architecture Applied for Bone Metastasis Classification in Nuclear Medicine: A Case Study on Prostate Cancer Patients

“…Over the last decade, artificial intelligence (AI) has been applied in diagnostic disciplines to support medical decision-making and to interpret medical data, such as medical images [ 6 , 7 , 8 ]. Computer-aided detection of BS was proposed for bone scintigraphy images based on a parallelepiped classification method to map the radionuclide distribution in [ 9 ].…”

Machine Learning Methods for Small Data Challenges in Molecular Science

“…Ji et al 57 treated MicroRNAs disease association prediction as a semi-supervised learning problem and proposed a novel method to predict potential MicroRNA-disease associations, a new method for predicting potential MicroRNAs-disease associations. In healthcare, Yin et al 58 introduced deep forest and semisupervised self-training to address disease classification and gene selection for different types of diseases. Experimental results demonstrated that the proposed model could achieve good results in both disease classification and causative gene identification.…”