论文标题

基于新型特征选择方法和变异自动编码器的自闭症谱系障碍的识别

Identification of Autism spectrum disorder based on a novel feature selection method and Variational Autoencoder

论文作者

Zhang, Fangyu, Wei, Yanjie, Liu, Jin, Wang, Yanlin, Xi, Wenhui, Pan, Yi

论文摘要

非侵入性脑成像的发展,例如静止状态功能磁共振成像(RS-FMRI)及其与AI算法的组合为早期诊断自闭症谱系障碍(ASD)提供了有希望的解决方案。但是,基于RS-FMRI的当前ASD分类的性能仍然需要改进。本文介绍了一个基于RS-FMRI的ASD诊断的分类框架。在框架中,我们根据步骤分布曲线(DSDC)之间的差异提出了一种新型的滤波器特征选择方法,以选择显着的功能连接性(FCS),并利用了由简化的变量自动码器(VAE)预测的多层perceptron(MLP)进行分类。我们还设计了一个由归一化过程和修改的双曲线切线(TANH)激活函数组成的管道,以替换原始的Tanh函数,从而进一步提高了模型的准确性。我们的模型通过10倍10倍的交叉验证进行了评估,并达到了78.12%的平均准确性,表现优于同一数据集上报告的最新方法。鉴于敏感性和特异性在疾病诊断中的重要性,在我们的模型中设计了两个限制,可以将模型的敏感性和特异性提高高达9.32%和10.21%。附加的约束允许我们的模型处理不同的应用程序方案,并且可以广泛使用。

The development of noninvasive brain imaging such as resting-state functional magnetic resonance imaging (rs-fMRI) and its combination with AI algorithm provides a promising solution for the early diagnosis of Autism spectrum disorder (ASD). However, the performance of the current ASD classification based on rs-fMRI still needs to be improved. This paper introduces a classification framework to aid ASD diagnosis based on rs-fMRI. In the framework, we proposed a novel filter feature selection method based on the difference between step distribution curves (DSDC) to select remarkable functional connectivities (FCs) and utilized a multilayer perceptron (MLP) which was pretrained by a simplified Variational Autoencoder (VAE) for classification. We also designed a pipeline consisting of a normalization procedure and a modified hyperbolic tangent (tanh) activation function to replace the original tanh function, further improving the model accuracy. Our model was evaluated by 10 times 10-fold cross-validation and achieved an average accuracy of 78.12%, outperforming the state-of-the-art methods reported on the same dataset. Given the importance of sensitivity and specificity in disease diagnosis, two constraints were designed in our model which can improve the model's sensitivity and specificity by up to 9.32% and 10.21%, respectively. The added constraints allow our model to handle different application scenarios and can be used broadly.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源