Boosting the performance of anomalous diffusion classifiers with the proper choice of features

Abstract

Understanding and identifying different types of single molecules' diffusion that occur in a broad range of systems (including living matter) is extremely important, as it can provide information on the physical and chemical characteristics of particles' surroundings. In recent years, an ever-growing number of methods have been proposed to overcome some of the limitations of the mean-squared displacements approach to tracer diffusion. In March 2020, the Anomalous Diffusion (AnDi) Challenge was launched by a community of international scientists to provide a framework for an objective comparison of the available methods for anomalous diffusion. In this paper, we introduce a feature-based machine learning method developed in response to Task 2 of the challenge, i.e. the classification of different types of diffusion. We discuss two sets of attributes that may be used for the classification of single-particle tracking data. The first one was proposed as our contribution to the AnDi Challenge. The latter is the result of our attempt to improve the performance of the classifier after the deadline of the competition. Extreme gradient boosting was used as the classification model. Although the deep-learning approach constitutes the state-of-the-art technology for data classification in many domains, we deliberately decided to pick this traditional machine learning algorithm due to its superior interpretability. After the extension of the feature set our classifier achieved the accuracy of 0.83, which is comparable with the top methods based on neural networks.

36 pages, 6 figures

References in corpus (6)

• Anomalous transport in the crowded world of biological cells
• Lévy walks
• Measurement of Anomalous Diffusion Using Recurrent Neural Networks
• Classification of particle trajectories in living cells: machine learning versus statistical testing hypothesis for fractional anomalous diffusion
• Characterization of anomalous diffusion classical statistics powered by deep learning (CONDOR)
• Moses, Noah and Joseph Effects in Coupled Lévy Processes

Cited by in corpus (11)

• Bayesian deep learning for error estimation in the analysis of anomalous diffusion
• Machine-Learning Solutions for the Analysis of Single-Particle Diffusion Trajectories
• Characterization of anomalous diffusion through convolutional transformers
• Gramian Angular Fields for leveraging pretrained computer vision models with anomalous diffusion trajectories
• Anomalous diffusion, non-Gaussianity, and nonergodicity for subordinated fractional Brownian motion with a drift
• Machine Learning Analysis of Anomalous Diffusion
• Preface: Characterisation of Physical Processes from Anomalous Diffusion Data
• Change-point detection in anomalous-diffusion trajectories utilising machine-learning-based uncertainty estimates
• Characterization of pedestrian contact interaction trajectories
• Characterizing pedestrian contact interaction trajectories to understand spreading risk in human crowds
• Recurrent neural network analysis of single trajectories switching between anomalous diffusion states