Automated Machine Learning Model Development for Intracranial Aneurysm Treatment Outcome Prediction: A Feasibility Study

Background: The prediction of aneurysm treatment outcomes can help to optimize the treatment strategies. Machine learning (ML) has shown positive results in many clinical areas. However, the development of such models requires expertise in ML, which is not an easy task for surgeons.Objectives: The r...

Full description

Bibliographic Details
Main Authors: Chubin Ou, Jiahui Liu, Yi Qian, Winston Chong, Dangqi Liu, Xuying He, Xin Zhang, Chuan-Zhi Duan
Format: Article
Language:English
Published: Frontiers Media S.A. 2021-11-01
Series:Frontiers in Neurology
Subjects:
Online Access:https://www.frontiersin.org/articles/10.3389/fneur.2021.735142/full
_version_ 1818742232238260224
author Chubin Ou
Chubin Ou
Jiahui Liu
Yi Qian
Winston Chong
Dangqi Liu
Xuying He
Xin Zhang
Chuan-Zhi Duan
author_facet Chubin Ou
Chubin Ou
Jiahui Liu
Yi Qian
Winston Chong
Dangqi Liu
Xuying He
Xin Zhang
Chuan-Zhi Duan
author_sort Chubin Ou
collection DOAJ
description Background: The prediction of aneurysm treatment outcomes can help to optimize the treatment strategies. Machine learning (ML) has shown positive results in many clinical areas. However, the development of such models requires expertise in ML, which is not an easy task for surgeons.Objectives: The recently emerged automated machine learning (AutoML) has shown promise in making ML more accessible to non-computer experts. We aimed to evaluate the feasibility of applying AutoML to develop the ML models for treatment outcome prediction.Methods: The patients with aneurysms treated by endovascular treatment were prospectively recruited from 2016 to 2020. Treatment was considered successful if angiographic complete occlusion was achieved at follow-up. A statistical prediction model was developed using multivariate logistic regression. In addition, two ML models were developed. One was developed manually and the other was developed by AutoML. Three models were compared based on their area under the precision-recall curve (AUPRC) and area under the receiver operating characteristic curve (AUROC).Results: The aneurysm size, stent-assisted coiling (SAC), and posterior circulation were the three significant and independent variables associated with treatment outcome. The statistical model showed an AUPRC of 0.432 and AUROC of 0.745. The conventional manually trained ML model showed an improved AUPRC of 0.545 and AUROC of 0.781. The AutoML derived ML model showed the best performance with AUPRC of 0.632 and AUROC of 0.832, significantly better than the other two models.Conclusions: This study demonstrated the feasibility of using AutoML to develop a high-quality ML model, which may outperform the statistical model and manually derived ML models. AutoML could be a useful tool that makes ML more accessible to the clinical researchers.
first_indexed 2024-12-18T02:09:15Z
format Article
id doaj.art-72a0a387dcfb46f6aad06cd082ba06b4
institution Directory Open Access Journal
issn 1664-2295
language English
last_indexed 2024-12-18T02:09:15Z
publishDate 2021-11-01
publisher Frontiers Media S.A.
record_format Article
series Frontiers in Neurology
spelling doaj.art-72a0a387dcfb46f6aad06cd082ba06b42022-12-21T21:24:31ZengFrontiers Media S.A.Frontiers in Neurology1664-22952021-11-011210.3389/fneur.2021.735142735142Automated Machine Learning Model Development for Intracranial Aneurysm Treatment Outcome Prediction: A Feasibility StudyChubin Ou0Chubin Ou1Jiahui Liu2Yi Qian3Winston Chong4Dangqi Liu5Xuying He6Xin Zhang7Chuan-Zhi Duan8Neurosurgery Center, Department of Cerebrovascular Surgery, The National Key Clinical Specialty, The Engineering Technology Research Center of Education Ministry of China on Diagnosis and Treatment of Cerebrovascular Disease, Guangdong Provincial Key Laboratory on Brain Function Repair and Regeneration, The Neurosurgery Institute of Guangdong Province, Zhujiang Hospital, Southern Medical University, Guangzhou, ChinaDepartment of Biomedical Sciences, Faculty of Medicine and Health Sciences, Macquarie University, Sydney, NSW, AustraliaNeurosurgery Center, Department of Cerebrovascular Surgery, The National Key Clinical Specialty, The Engineering Technology Research Center of Education Ministry of China on Diagnosis and Treatment of Cerebrovascular Disease, Guangdong Provincial Key Laboratory on Brain Function Repair and Regeneration, The Neurosurgery Institute of Guangdong Province, Zhujiang Hospital, Southern Medical University, Guangzhou, ChinaNeurosurgery Center, Department of Cerebrovascular Surgery, The National Key Clinical Specialty, The Engineering Technology Research Center of Education Ministry of China on Diagnosis and Treatment of Cerebrovascular Disease, Guangdong Provincial Key Laboratory on Brain Function Repair and Regeneration, The Neurosurgery Institute of Guangdong Province, Zhujiang Hospital, Southern Medical University, Guangzhou, ChinaMonash Medical Centre, Monash University, Clayton, VIC, AustraliaDepartment of Neurosurgery, The First People's Hospital of Foshan, Foshan, ChinaNeurosurgery Center, Department of Cerebrovascular Surgery, The National Key Clinical Specialty, The Engineering Technology Research Center of Education Ministry of China on Diagnosis and Treatment of Cerebrovascular Disease, Guangdong Provincial Key Laboratory on Brain Function Repair and Regeneration, The Neurosurgery Institute of Guangdong Province, Zhujiang Hospital, Southern Medical University, Guangzhou, ChinaNeurosurgery Center, Department of Cerebrovascular Surgery, The National Key Clinical Specialty, The Engineering Technology Research Center of Education Ministry of China on Diagnosis and Treatment of Cerebrovascular Disease, Guangdong Provincial Key Laboratory on Brain Function Repair and Regeneration, The Neurosurgery Institute of Guangdong Province, Zhujiang Hospital, Southern Medical University, Guangzhou, ChinaNeurosurgery Center, Department of Cerebrovascular Surgery, The National Key Clinical Specialty, The Engineering Technology Research Center of Education Ministry of China on Diagnosis and Treatment of Cerebrovascular Disease, Guangdong Provincial Key Laboratory on Brain Function Repair and Regeneration, The Neurosurgery Institute of Guangdong Province, Zhujiang Hospital, Southern Medical University, Guangzhou, ChinaBackground: The prediction of aneurysm treatment outcomes can help to optimize the treatment strategies. Machine learning (ML) has shown positive results in many clinical areas. However, the development of such models requires expertise in ML, which is not an easy task for surgeons.Objectives: The recently emerged automated machine learning (AutoML) has shown promise in making ML more accessible to non-computer experts. We aimed to evaluate the feasibility of applying AutoML to develop the ML models for treatment outcome prediction.Methods: The patients with aneurysms treated by endovascular treatment were prospectively recruited from 2016 to 2020. Treatment was considered successful if angiographic complete occlusion was achieved at follow-up. A statistical prediction model was developed using multivariate logistic regression. In addition, two ML models were developed. One was developed manually and the other was developed by AutoML. Three models were compared based on their area under the precision-recall curve (AUPRC) and area under the receiver operating characteristic curve (AUROC).Results: The aneurysm size, stent-assisted coiling (SAC), and posterior circulation were the three significant and independent variables associated with treatment outcome. The statistical model showed an AUPRC of 0.432 and AUROC of 0.745. The conventional manually trained ML model showed an improved AUPRC of 0.545 and AUROC of 0.781. The AutoML derived ML model showed the best performance with AUPRC of 0.632 and AUROC of 0.832, significantly better than the other two models.Conclusions: This study demonstrated the feasibility of using AutoML to develop a high-quality ML model, which may outperform the statistical model and manually derived ML models. AutoML could be a useful tool that makes ML more accessible to the clinical researchers.https://www.frontiersin.org/articles/10.3389/fneur.2021.735142/fullintracranial aneurysmAutoMLendovascular treatmentmachine learningstroke
spellingShingle Chubin Ou
Chubin Ou
Jiahui Liu
Yi Qian
Winston Chong
Dangqi Liu
Xuying He
Xin Zhang
Chuan-Zhi Duan
Automated Machine Learning Model Development for Intracranial Aneurysm Treatment Outcome Prediction: A Feasibility Study
Frontiers in Neurology
intracranial aneurysm
AutoML
endovascular treatment
machine learning
stroke
title Automated Machine Learning Model Development for Intracranial Aneurysm Treatment Outcome Prediction: A Feasibility Study
title_full Automated Machine Learning Model Development for Intracranial Aneurysm Treatment Outcome Prediction: A Feasibility Study
title_fullStr Automated Machine Learning Model Development for Intracranial Aneurysm Treatment Outcome Prediction: A Feasibility Study
title_full_unstemmed Automated Machine Learning Model Development for Intracranial Aneurysm Treatment Outcome Prediction: A Feasibility Study
title_short Automated Machine Learning Model Development for Intracranial Aneurysm Treatment Outcome Prediction: A Feasibility Study
title_sort automated machine learning model development for intracranial aneurysm treatment outcome prediction a feasibility study
topic intracranial aneurysm
AutoML
endovascular treatment
machine learning
stroke
url https://www.frontiersin.org/articles/10.3389/fneur.2021.735142/full
work_keys_str_mv AT chubinou automatedmachinelearningmodeldevelopmentforintracranialaneurysmtreatmentoutcomepredictionafeasibilitystudy
AT chubinou automatedmachinelearningmodeldevelopmentforintracranialaneurysmtreatmentoutcomepredictionafeasibilitystudy
AT jiahuiliu automatedmachinelearningmodeldevelopmentforintracranialaneurysmtreatmentoutcomepredictionafeasibilitystudy
AT yiqian automatedmachinelearningmodeldevelopmentforintracranialaneurysmtreatmentoutcomepredictionafeasibilitystudy
AT winstonchong automatedmachinelearningmodeldevelopmentforintracranialaneurysmtreatmentoutcomepredictionafeasibilitystudy
AT dangqiliu automatedmachinelearningmodeldevelopmentforintracranialaneurysmtreatmentoutcomepredictionafeasibilitystudy
AT xuyinghe automatedmachinelearningmodeldevelopmentforintracranialaneurysmtreatmentoutcomepredictionafeasibilitystudy
AT xinzhang automatedmachinelearningmodeldevelopmentforintracranialaneurysmtreatmentoutcomepredictionafeasibilitystudy
AT chuanzhiduan automatedmachinelearningmodeldevelopmentforintracranialaneurysmtreatmentoutcomepredictionafeasibilitystudy