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 条
  • [1] A multi-metric approach for research evaluation
    Rousseau, Ronald
    Ye, Fred Y.
    CHINESE SCIENCE BULLETIN, 2013, 58 (26): : 3288 - 3290
  • [2] DEEP MULTI-METRIC LEARNING FOR PERSON RE-IDENTIFICATION
    Ge, Yongxin
    Gu, Xinqian
    Chen, Min
    Wang, Hongxing
    Yang, Dan
    2018 IEEE INTERNATIONAL CONFERENCE ON MULTIMEDIA AND EXPO (ICME), 2018,
  • [3] An efficient multi-metric learning method by partitioning the metric space
    Yuan, Chao
    Yang, Liming
    NEUROCOMPUTING, 2023, 529 : 56 - 79
  • [4] Indexing Multi-Metric Data
    Franzke, Maximilian
    Emrich, Tobias
    Zufle, Andreas
    Renz, Matthias
    2016 32ND IEEE INTERNATIONAL CONFERENCE ON DATA ENGINEERING (ICDE), 2016, : 1122 - 1133
  • [5] Fuzzy multi-metric spaces
    Efe, Hakan
    Alaca, Cihangir
    Yildiz, Cemil
    JOURNAL OF COMPUTATIONAL ANALYSIS AND APPLICATIONS, 2008, 10 (03) : 367 - 375
  • [6] Multi-metric learning by a pair of twin-metric learning framework
    Min Zhang
    Liming Yang
    Chao Yuan
    Qiangqiang Ren
    Applied Intelligence, 2022, 52 : 17490 - 17507
  • [7] Multi-metric learning by a pair of twin-metric learning framework
    Zhang, Min
    Yang, Liming
    Yuan, Chao
    Ren, Qiangqiang
    APPLIED INTELLIGENCE, 2022, 52 (15) : 17490 - 17507
  • [8] An efficient method for clustered multi-metric learning
    Bac Nguyen
    Ferri, Francesc J.
    Morell, Carlos
    De Baets, Bernard
    INFORMATION SCIENCES, 2019, 471 : 149 - 163
  • [9] Deep multi-metric learning for text-independent speaker verification
    Xu, Jiwei
    Wang, Xinggang
    Feng, Bin
    Liu, Wenyu
    NEUROCOMPUTING, 2020, 410 : 394 - 400
  • [10] Geometry of manifolds with area metric: Multi-metric backgrounds
    Schuller, Frederic P.
    Wohlfarth, Mattias N. R.
    NUCLEAR PHYSICS B, 2006, 747 (03) : 398 - 422