Benchmarking the robustness of the correct identification of flexible 3D objects using common machine learning models

TitleBenchmarking the robustness of the correct identification of flexible 3D objects using common machine learning models
Publication TypeJournal Article
Year of Publication2025
AuthorsZhang Y., Vitalis A.
JournalPatterns
Volume6
Start Page101147
Issue1
Pagination101147
Date PublishedJan 10, 2025
Type of ArticleResearch Article
Keywords3D-CNNs, benchmarking, flexible object recognition, machine learning, molecular dynamics, Proteins
Abstract

True three-dimensional (3D) data are prevalent in domains such as molecular science or computer vision. In these data, machine learning models are often asked to identify objects subject to intrinsic flexibility. Our study introduces two datasets from molecular science to assess the classification robustness of common model/feature combinations. Molecules are flexible, and shapes alone offer intra-class heterogeneities that yield a high risk for confusions. By blocking training and test sets to reduce overlap, we establish a baseline requiring the trained models to abstract from shape. As training data coverage grows, all tested architectures perform better on unseen data with reduced overfitting. Empirically, 2D embeddings of voxelized data produced the best-performing models. Evidently, both featurization and task-appropriate model design are of continued importance, the latter point reinforced by comparisons to recent, more specialized models. Finally, we show that the shape abstraction learned from database samples extends to samples that are evolving explicitly in time.

URLhttps://www.sciencedirect.com/science/article/pii/S2666389924003192
DOI10.1016/j.patter.2024.101147
pubindex

0307

Alternate JournalPatterns
Highlight Role: 
Algorithms and Data Analysis