Efficient image dataset classification difficulty estimation for predicting deep-learning accuracy

被引:20
作者
Scheidegger, Florian [1 ,2 ]
Istrate, Roxana [2 ,3 ]
Mariani, Giovanni [2 ]
Benini, Luca [1 ,4 ]
Bekas, Costas [2 ]
Malossi, Cristiano [2 ]
机构
[1] Swiss Fed Inst Technol, Ramistr 101, CH-8092 Zurich, Switzerland
[2] IBM Res Zurich, Saumerstr 4, CH-8803 Ruschlikon, Switzerland
[3] Queens Univ Belfast, Univ Rd, Belfast BT7 1NN, Antrim, North Ireland
[4] Univ Bologna, Via Zamboni 33, I-40126 Bologna, Italy
关键词
Dataset characterization; Classification difficulty; Deep learning; Image classification;
D O I
10.1007/s00371-020-01922-5
中图分类号
TP31 [计算机软件];
学科分类号
081202 ; 0835 ;
摘要
In the deep-learning community, new algorithms are published at a very fast pace. Therefore, solving an image classification problem for new datasets becomes a challenging task, as it requires to re-evaluate published algorithms and their different configurations in order to find a close to optimal classifier. To facilitate this process, before biasing our decision toward a class of neural networks or running an expensive search over the network space, we propose to estimate the classification difficulty of the dataset. Our method computes a single number that characterizes the dataset difficulty 97x faster than training state-of-the-art networks. The proposed method can be used in combination with network topology and hyper-parameter search optimizers to efficiently drive the search toward promising neural network configurations.
引用
收藏
页码:1593 / 1610
页数:18
相关论文
共 50 条
[41]   Aurora Image Classification with Deep Metric Learning [J].
Endo, Takeru ;
Matsumoto, Mitsuharu .
SENSORS, 2022, 22 (17)
[42]   Fruit Image Classification Using Deep Learning [J].
Gill, Harmandeep Singh ;
Khalaf, Osamah Ibrahim ;
Alotaibi, Youseef ;
Alghamdi, Saleh ;
Alassery, Fawaz .
CMC-COMPUTERS MATERIALS & CONTINUA, 2022, 71 (03) :5135-5150
[43]   Deep learning with Shallow architecture for Image Classification [J].
ElAdel, Asma ;
Ejbali, Ridha ;
Zaied, Mourad ;
Ben Amar, Chokri .
PROCEEDINGS OF THE 2015 INTERNATIONAL CONFERENCE ON HIGH PERFORMANCE COMPUTING & SIMULATION (HPCS 2015), 2015, :408-412
[44]   Deep Learning Approaches for Image Classification Techniques [J].
Guan, Youyou ;
Han, Yuxuan ;
Liu, Siqi .
2022 IEEE INTERNATIONAL CONFERENCE ON ELECTRICAL ENGINEERING, BIG DATA AND ALGORITHMS (EEBDA), 2022, :1132-1136
[45]   Research on Image Classification Based on Deep Learning [J].
Li, Jiao ;
Nanchang, Cheng ;
Song, Kang .
2021 IEEE/ACIS 20TH INTERNATIONAL CONFERENCE ON COMPUTER AND INFORMATION SCIENCE (ICIS 2021-SUMMER), 2021, :132-136
[46]   Image classification of sugar crystal with deep learning [J].
Chayatummagoon, Suriya ;
Chongstitvatana, Prabhas .
2021 13TH INTERNATIONAL CONFERENCE ON KNOWLEDGE AND SMART TECHNOLOGY (KST-2021), 2021, :118-122
[47]   Extended Deep-Learning Network for Histopathological Image-Based Multiclass Breast Cancer Classification Using Residual Features [J].
Mewada, Hiren .
SYMMETRY-BASEL, 2024, 16 (05)
[48]   An integrated deep-learning model for smart waste classification [J].
Shivendu Mishra ;
Ritika Yaduvanshi ;
Prince Rajpoot ;
Sharad Verma ;
Amit Kumar Pandey ;
Digvijay Pandey .
Environmental Monitoring and Assessment, 2024, 196
[49]   A Deep-learning based Method for the Classification of the Cellular Images [J].
Vununu, Caleb ;
Lee, Suk-Hwan ;
Kwon, Ki-Ryong .
PROCEEDINGS OF THE 13TH INTERNATIONAL JOINT CONFERENCE ON BIOMEDICAL ENGINEERING SYSTEMS AND TECHNOLOGIES, VOL 3: BIOINFORMATICS, 2020, :242-245
[50]   Hybrid deep learning model for density and growth rate estimation on weed image dataset [J].
Mishra, Anand Muni ;
Singh, Mukund Pratap ;
Singh, Prabhishek ;
Diwakar, Manoj ;
Gupta, Indrajeet ;
Bijalwan, Anchit .
SCIENTIFIC REPORTS, 2025, 15 (01)