Affiliation:
1. Singapore Eye Research Institute
2. SingHealth
Abstract
Abstract
Automated machine learning (AutoML) allows for the simplified application of machine learning to real-world problems, by the implicit handling of necessary steps such as data pre-processing, feature engineering, model selection and hyperparameter optimization. This has encouraged its use in medical applications such as imaging. However, the impact of common parameter choices such as the number of trials allowed, and the resolution of the input images, has not been comprehensively explored in existing literature. We therefore benchmark AutoKeras (AK), an open-source AutoML framework, against several bespoke deep learning architectures, on five public medical datasets representing a wide range of imaging modalities. It was found that AK could outperform the bespoke models in general, although at the cost of increased training time. Moreover, our experiments suggest that a large number of trials and higher resolutions may not be necessary for optimal performance to be achieved.
Publisher
Research Square Platform LLC
Reference28 articles.
1. Rajkomar, Alvin, Jeffrey Dean, and Isaac Kohane. "Machine learning in medicine." New England Journal of Medicine 380.14 (2019): 1347–1358.
2. "A machine learning model for detecting invasive ductal carcinoma with Google Cloud AutoML Vision;Zeng Yan;Computers in Biology and Medicine,2020
3. "Classification of pachychoroid disease on ultrawide-field indocyanine green angiography using auto-machine learning platform;Kim In;British Journal of Ophthalmology,2021
4. "Evaluation of the performance of traditional machine learning algorithms, convolutional neural network and AutoML Vision in ultrasound breast lesions classification: a comparative study." Quantitative imaging in medicine and surgery 11;Wan Ka,2021
5. "Automated deep learning design for medical image classification by health-care professionals with no coding experience: a feasibility study;Faes Livia;The Lancet Digital Health,2019