Addressing database variability in learning from medical data: An ensemble-based approach using convolutional neural networks and a case of study applied to automatic sleep scoring.

Comput Biol Med

Laboratory for Research and Development of Artificial Intelligence, Computer Science Department, University of A Coruña, Campus Elviña s/n, 15071, A Coruña, Spain.

Published: April 2020

AI Article Synopsis

  • This work addresses the challenges of developing machine learning models for robust generalization across multiple databases, specifically in the context of sleep staging in sleep medicine.
  • The study highlights the "database variability problem," illustrating difficulties in applying a model's local generalization to independent external databases.
  • It proposes a new approach using an ensemble of local models to improve inter-database generalization and examines various model configurations and data pre-processing techniques to enhance overall performance in training with multiple datasets.

Article Abstract

In this work we examine some of the problems associated with the development of machine learning models with the objective to achieve robust generalization capabilities on common-task multiple-database scenarios. Referred to as the "database variability problem", we focus on a specific medical domain (sleep staging in sleep medicine) to show the non-triviality of translating the estimated model's local generalization capabilities into independent external databases. We analyze some of the scalability problems when multiple-database data are used as inputs to train a single learning model. Then, we introduce a novel approach based on an ensemble of local models, and we show its advantages in terms of inter-database generalization performance and data scalability. In addition, we analyze different model configurations and data pre-processing techniques to determine their effects on the overall generalization performance. For this purpose, we carry out experimentation that involves several sleep databases and evaluates different machine learning models based on convolutional neural networks.

Download full-text PDF

Source
http://dx.doi.org/10.1016/j.compbiomed.2020.103697DOI Listing

Publication Analysis

Top Keywords

convolutional neural
8
neural networks
8
machine learning
8
learning models
8
generalization capabilities
8
generalization performance
8
addressing database
4
database variability
4
learning
4
variability learning
4

Similar Publications

Want AI Summaries of new PubMed Abstracts delivered to your In-box?

Enter search terms and have AI summaries delivered each week - change queries or unsubscribe any time!