Establishment and analysis of osteoarthritis diagnosis model based on artificial neural networks

Fan Yidong; Qin Gang; Su Guowei; Xiao Shifu; Liu Junliang; Li Weicai; Wu Guangtao

doi:10.12307/2024.294

Abstract：BACKGROUND:Rapid developments in the field of bioinformatics have provided new methods for the diagnosis of osteoarthritis.Artificial neural networks have powerful data computing and classification capabilities,which have shown better performance in disease diagnosis. OBJECTIVE:To establish a new diagnostic predictive model of osteoarthritis based on artificial neural network and to verify the diagnostic value of the model in osteoarthritis with an external dataset. METHODS:The eligible osteoarthritis-related data sets were downloaded through GEO database search and divided into Train group and Test group.The gene expression matrix of the Train group was analyzed to screen the differentially expressed genes.GO and KEGG enrichment analyses were performed on the differentially expressed genes.Through Lasso regression model,support vector machine model and random forest tree model,the key genes of osteoarthritis were further identified from the differentially expressed genes.The R software"Neuralnet"package was then used to construct the osteoarthritis diagnosis model based on artificial neural network,and the model performance was evaluated by the five-fold cross-validation.Two independent data sets in the Test group were used to verify their diagnostic results. RESULTS AND CONCLUSION:A total of 90 differentially expressed genes related to osteoarthritis were obtained by differential analysis,of which 33 were down-regulated and 57 were up-regulated.GO enrichment analysis showed that the differentially expressed genes were mainly involved in the following biological processes,including leukocyte-mediated immunity,leukocyte migration in bone marrow and chemokine production.KEGG enrichment analysis showed that these genes were mainly enriched in rheumatoid arthritis,interleukin-17 signaling pathway and osteoclast differentiation pathway.Five key genes for the diagnosis of osteoarthritis,HMGB2,GADD45A,SLC19A2,TPPP3 and FOLR2,were identified by three machine learning methods.The artificial neural network model of five key genes in the Train group showed that the accuracy was 96.36%and the area under the curve was 0.997.The five-fold cross validation of the neural network model showed that the average area under the curve was greater than 0.9 and the model was of robustness.Two independent data sets in the Test group showed its area under the curve was 0.814 and 0.788 respectively.Therefore,the establishment of an artificial neural network model for the diagnosis of osteoarthritis has a certain diagnostic value.