PuSH - Publication Server of Helmholtz Zentrum München

Thammasorn, P.* ; Chaovalitwongse, W.A.* ; Hippe, D.S.* ; Wootton, L.S.* ; Ford, E.C.* ; Spraker, M.B.* ; Combs, S.E. ; Peeken, J.C. ; Nyflot, M.J.*

Nearest neighbor-based strategy to optimize multi-view triplet network for classification of small-sample medical imaging data.

IEEE Trans. Neural. Netw. Learn. Syst., DOI: 10.1109/TNNLS.2021.3059635 (2021)
DOI PMC
IEEE Multi-view classification with limited sample size and data augmentation is a very common machine learning (ML) problem in medicine. With limited data, a triplet network approach for two-stage representation learning has been proposed. However, effective training and verifying the features from the representation network for their suitability in subsequent classifiers are still unsolved problems. Although typical distance-based metrics for the training capture the overall class separability of the features, the performance according to these metrics does not always lead to an optimal classification. Consequently, an exhaustive tuning with all feature-classifier combinations is required to search for the best end result. To overcome this challenge, we developed a novel nearest-neighbor (NN) validation strategy based on the triplet metric. This strategy is supported by a theoretical foundation to provide the best selection of the features with a lower bound of the highest end performance. The proposed strategy is a transparent approach to identify whether to improve the features or the classifier. This avoids the need for repeated tuning. Our evaluations on real-world medical imaging tasks (i.e., radiation therapy delivery error prediction and sarcoma survival prediction) show that our strategy is superior to other common deep representation learning baselines [i.e., autoencoder (AE) and softmax]. The strategy addresses the issue of feature's interpretability which enables more holistic feature creation such that the medical experts can focus on specifying relevant data as opposed to tedious feature engineering.
Altmetric
Additional Metrics?
Edit extra informations Login
Publication type Article: Journal article
Document type Scientific Article
Corresponding Author
Keywords Medical Data Classification ; Multi-view Learning ; Representation Learning ; Transfer Learning Metric Learning.; Radiomic Analysis; Images
ISSN (print) / ISBN 2162-237X
e-ISSN 2162-2388
Publisher IEEE
Publishing Place 445 Hoes Lane, Piscataway, Nj 08855-4141 Usa
Non-patent literature Publications
Reviewing status Peer reviewed
Grants National Science Foundation (NSF)