As shown in Figure 1, the area is 0.84 under the
ROC curve of CatBoost, the accuracy of this model is
high. Compared with other models, CatBoost model
solves the problems of gradient deviation and
prediction offset, so the occurrence of over fitting is
reduced and the accuracy and generalization ability of
the algorithm are improved. Randomforestentr and
RandomForestGini belong to random forest models,
which can be over fitted by reducing randomness, so
their prediction accuracy is high. However, due to the
high randomness of their parameters, their accuracy
is lower than CatBoost. Xgboost model uses
approximate algorithm to improve operation speed.
Kneigborsdist and Kneigborsunif are too simple in
structure, resulting in insufficient fitting of data.
Extratreesgini model adopts randomization for model
structure and model parameters, resulting in too
strong randomization of model. Therefore, the
prediction accuracy of Xgboost, Kneigborsdist,
Kneigborsunif and Extratreesgini are lower.
4 DISCUSSION
The results of machine learning prediction model
based on the functional connection characteristics of
brain network show that the effectiveness of
neurofeedback in insomnia patients is significantly
higher than the random probability. The clinical
practical significance of the model can be explained
as whether an individual with insomnia can be
predicted to be suitable for neural feedback training
after resting state MRI scanning and brain network
data analysis, which provided a reliable basis for the
selection of treatment schemes for insomnia patients.
However, our study has some limitations. On the
one hand, the degree of each subject is different in
insomnia, and there are individual differences in the
treatment effect after neurofeedback training. For the
ineffective treatment of patients after neurofeedback,
whether it is caused by internal factors or external
factors such as errors in the experimental process,
these problems need to be further discussed. For
example, we can divide the subjects' disease degree
more carefully, explore the impact of different
severity of insomnia on the effect of neurofeedback
training, or adjust the neurofeedback experimental
design by changing the neurofeedback target area. On
the other hand, due to the limited experimental
conditions and the limited number of samples
included in the experiment, 24 subjects participated
in neurofeedback training. With the development of
more clinical experiments, this method is expected to
achieve more accurate prediction.
5 CONCLUSIONS
In the paper, we propose a neurofeedback
effectiveness prediction method based on the
functional connectivity of resting state brain
networks. The functional connections of brain
networks such as DMN, ECN, SAN and SM et.al.
before neural feedback in insomnia are extracted as
features, and a prediction model based on automatic
machine learning is constructed to predict the neural
feedback training effect of insomnia patients, so as to
realize the prediction of the effectiveness of neural
feedback training of insomnia patients based on
machine learning. The experimental results show that
the highest prediction accuracy of all machine
learning models reaches 75%, which provides an
important support for insomnia patients to make
decisions in the treatment plan.
ACKNOWLEDGEMENTS
This work was supported by the National Natural
Science Foundation of China [grant number
82071884]; and the Key Project of Medical Science
and Technology of Henan Province [grant number
LHGJ20200060].
REFERENCES
Alkoby, O., Abu-Rmileh, A., Shriki, O., & Todder, D..
(2017). Can we predict who will respond to
neurofeedback? a review of the inefficacy problem and
existing predictors for successful eeg neurofeedback
learning. Neuroscience, S0306452216307576.
Albert, K. M. , GG Potter, Boyd, B. D. , Kang, H. , &
Taylor, W. D. . (2018). Brain network functional
connectivity and cognitive performance in major
depressive disorder. Journal of Psychiatric Research.
American, P. . (1994). Diagnostic and statistical manual of
mental disorders. Encyclopedia of the Neurological
Sciences, 25(2), 4-8.
Calhoun, V. D. , Adali, T. , Pearlson, G. D. , & Pekar, J. J.
. (2010). A method for making group inferences from
functional mri data using independent component
analysis. Human Brain Mapping, 14(2), 140.
Fatt, C. R. C. , Jha, M. K. , Cooper, C. M. , Fonzo, G. , &
Trivedi, M. H. . (2019). Effect of intrinsic patterns of
functional brain connectivity in moderating
antidepressant treatment response in major depression.
American Journal of Psychiatry, 177(2),
appi.ajp.2019.1.
Haugg, A., Sladky, R., Skouras, S., Mcdonald, A., &
Scharnowski, F.. (2020). Can we predict real‐time fmri