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.

arXiv comment36 pages, 6 figures

References in corpus (18)

• XGBoost: A Scalable Tree Boosting System
• Anomalous transport in the crowded world of biological cells
• Lévy walks
• In vivo anomalous diffusion and weak ergodicity breaking of lipid granules
• Quantitative analysis of single particle trajectories: mean maximal excursion method
• Nonergodic Subdiffusion from Brownian Motion in an Inhomogeneous Medium
• Machine learning method for single trajectory characterization
• Consistent feature attribution for tree ensembles
• Classification of diffusion modes in single-particle tracking data: Feature-based versus deep-learning approach
• Anomalous, non-Gaussian tracer diffusion in heterogeneously crowded environments
• 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
• Evidence of hydrodynamic and subdiffusive motion of tracers in a viscoelastic medium
• Characterization of anomalous diffusion classical statistics powered by deep learning (CONDOR)
• Optimal parameters for anomalous diffusion exponent estimation from noisy data
• Moses, Noah and Joseph Effects in Coupled Lévy Processes
• AnDi: The Anomalous Diffusion Challenge
• Statistical test for fractional Brownian motion based on detrending moving average algorithm

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