Deep multi-metric training: the need of multi-metric curve evaluation to avoid weak learning

被引:0
|
作者
Mamalakis, Michail [1 ,2 ,3 ]
Banerjee, Abhirup [8 ,9 ]
Ray, Surajit [4 ]
Wilkie, Craig [4 ]
Clayton, Richard H. [2 ,3 ]
Swift, Andrew J. [3 ,5 ]
Panoutsos, George [6 ]
Vorselaars, Bart [7 ]
机构
[1] Department of Psychiatry, University of Cambridge, Herchel Smith Building, Robinson Way, Cambridge,CB2 0SZ, United Kingdom
[2] Department of Computer Science, University of Sheffield, 211 Portobello, Sheffield,S1 4DP, United Kingdom
[3] Insigneo Institute for in silico Medicine, University of Sheffield, The Pam Liversidge Building, Sheffield,S1 3JD, United Kingdom
[4] School of Mathematics and Statistics, University of Glasgow, Glasgow,G12 8QQ, United Kingdom
[5] Department of Infection, Immunity and Cardiovascular Disease, University of Sheffield, Sheffield,S10 2HQ, United Kingdom
[6] Department of Automatic Control and Systems Engineering, University of Sheffield, Sheffield,S10 2TN, United Kingdom
[7] School of Mathematics and Physics, University of Lincoln, Brayford Pool, Lincoln,LN6 7TS, United Kingdom
[8] Department of Engineering Science, Institute of Biomedical Engineering, University of Oxford, Oxford,OX3 7DQ, United Kingdom
[9] Division of Cardiovascular Medicine, Radcliffe Department of Medicine, University of Oxford, Oxford,OX3 9DU, United Kingdom
关键词
COVID-19;
D O I
10.1007/s00521-024-10182-6
中图分类号
学科分类号
摘要
The development and application of artificial intelligence-based computer vision systems in medicine, environment, and industry are playing an increasingly prominent role. Hence, the need for optimal and efficient hyperparameter tuning strategies is more than crucial to deliver the highest performance of the deep learning networks in large and demanding datasets. In our study, we have developed and evaluated a new training methodology named deep multi-metric training (DMMT) for enhanced training performance. The DMMT delivers a state of robust learning for deep networks using a new important criterion of multi-metric performance evaluation. We have tested the DMMT methodology in multi-class (three, four, and ten), multi-vendors (different X-ray imaging devices), and multi-size (large, medium, and small) datasets. The validity of the DMMT methodology has been tested in three different classification problems: (i) medical disease classification, (ii) environmental classification, and (iii) ecological classification. For disease classification, we have used two large COVID-19 chest X-rays datasets, namely the BIMCV COVID-19+ and Sheffield hospital datasets. The environmental application is related to the classification of weather images in cloudy, rainy, shine or sunrise conditions. The ecological classification task involves a classification of three animal species (cat, dog, wild) and a classification of ten animals and transportation vehicles categories (CIFAR-10). We have used state-of-the-art networks of DenseNet-121, ResNet-50, VGG-16, VGG-19, and DenResCov-19 (DenRes-131) to verify that our novel methodology is applicable in a variety of different deep learning networks. To the best of our knowledge, this is the first work that proposes a training methodology to deliver robust learning, over a variety of deep learning networks and multi-field classification problems.
引用
收藏
页码:18841 / 18862
页数:21
相关论文
共 50 条
  • [31] A hierarchical multi-metric framework for item clustering
    Kotouza, Maria Th.
    Vavliakis, Konstantinos N.
    Psomopoulos, Fotis E.
    Mitkas, Pericles A.
    2018 IEEE/ACM 5TH INTERNATIONAL CONFERENCE ON BIG DATA COMPUTING APPLICATIONS AND TECHNOLOGIES (BDCAT), 2018, : 191 - 197
  • [32] Dynamic Multi-Metric Thresholds for Scaling Applications Using Reinforcement Learning
    Rossi, Fabiana
    Cardellini, Valeria
    Lo Presti, Francesco
    Nardelli, Matteo
    IEEE TRANSACTIONS ON CLOUD COMPUTING, 2023, 11 (02) : 1807 - 1821
  • [33] Collaborative discriminative multi-metric learning for facial expression recognition in video
    Yan, Haibin
    PATTERN RECOGNITION, 2018, 75 : 33 - 40
  • [34] A multi-metric similarity based analysis of microarray data
    Altiparmak, Fatih
    Erdal, Selnur
    Ozturk, Ozgur
    Ferhatosmanoglu, Hakan
    2007 IEEE INTERNATIONAL CONFERENCE ON BIOINFORMATICS AND BIOMEDICINE, PROCEEDINGS, 2007, : 317 - +
  • [35] A macroinvertebrate multi-metric index for Ethiopian highland streams
    Alemneh, Temesgen
    Ambelu, Argaw
    Zaitchik, Benjamin F.
    Bahrndorff, Simon
    Mereta, Seid Tiku
    Pertoldi, Cino
    HYDROBIOLOGIA, 2019, 843 (01) : 125 - 141
  • [36] Large Margin Multi-metric Learning for Face and Kinship Verification in the Wild
    Hu, Junlin
    Lu, Jiwen
    Yuan, Junsong
    Tan, Yap-Peng
    COMPUTER VISION - ACCV 2014, PT III, 2015, 9005 : 252 - 267
  • [37] What Makes Objects Similar: A Unified Multi-Metric Learning Approach
    Ye, Han-Jia
    Zhan, De-Chuan
    Si, Xue-Min
    Jiang, Yuan
    Zhou, Zhi-Hua
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 29 (NIPS 2016), 2016, 29
  • [38] Large margin projection-based multi-metric learning for classification
    Yuan, Chao
    Yang, Liming
    KNOWLEDGE-BASED SYSTEMS, 2022, 243
  • [39] KERNEL MULTI-METRIC LEARNING FOR MULTI-CHANNEL TRANSIENT ACOUSTIC SIGNAL CLASSIFICATION
    Zhang, Haichao
    Zhang, Yanning
    Nasrabadi, Nasser M.
    Huang, Thomas S.
    2012 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), 2012, : 1989 - 1992
  • [40] What Makes Objects Similar: A Unified Multi-Metric Learning Approach
    Ye, Han-Jia
    Zhan, De-Chuan
    Jiang, Yuan
    Zhou, Zhi-Hua
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2019, 41 (05) : 1257 - 1270