Enhancing ensemble classifiers utilizing gaze tracking data for autism spectrum disorder diagnosis

Comput Biol Med. 2024 Nov:182:109184. doi: 10.1016/j.compbiomed.2024.109184. Epub 2024 Sep 30.

Abstract

Problem: Diagnosing Autism Spectrum Disorder (ASD) remains a significant challenge, especially in regions where access to specialists is limited. Computer-based approaches offer a promising solution to make diagnosis more accessible. Eye tracking has emerged as a valuable technique in aiding the diagnosis of ASD. Typically, individuals' gaze patterns are monitored while they view videos designed according to established paradigms. In a previous study, we developed a method to classify individuals as having ASD or Typical Development (TD) by processing eye-tracking data using Random Forest ensembles, with a focus on a paradigm known as joint attention.

Aim: This article aims to enhance our previous work by evaluating alternative algorithms and ensemble strategies, with a particular emphasis on the role of anticipation features in diagnosis.

Methods: Utilizing stimuli based on joint attention and the concept of "floating regions of interest" from our earlier research, we identified features that indicate gaze anticipation or delay. We then tested seven class balancing strategies, applied seven dimensionality reduction algorithms, and combined them with five different classifier induction algorithms. Finally, we employed the stacking technique to construct an ensemble model.

Results: Our findings showed a significant improvement, achieving an F1-score of 95.5%, compared to the 82% F1-score from our previous work, through the use of a heterogeneous stacking meta-classifier composed of diverse induction algorithms.

Conclusion: While there remains an opportunity to explore new algorithms and features, the approach proposed in this article has the potential to be applied in clinical practice, contributing to increased accessibility to ASD diagnosis.

Keywords: Anticipation; Autism spectrum disorder; Ensemble classifier; Eye-tracking; Machine learning; Meta-classifier; Stacking.

MeSH terms

  • Algorithms*
  • Autism Spectrum Disorder* / diagnosis
  • Autism Spectrum Disorder* / physiopathology
  • Child
  • Diagnosis, Computer-Assisted / methods
  • Eye Movements / physiology
  • Eye-Tracking Technology*
  • Female
  • Fixation, Ocular / physiology
  • Humans
  • Male