Dokumentenart: | Artikel | ||||
---|---|---|---|---|---|
Titel eines Journals oder einer Zeitschrift: | European Journal of Cancer | ||||
Verlag: | Elsevier | ||||
Ort der Veröffentlichung: | OXFORD | ||||
Band: | 173 | ||||
Seitenbereich: | S. 307-316 | ||||
Datum: | 2022 | ||||
Institutionen: | Medizin > Lehrstuhl für Dermatologie und Venerologie | ||||
Identifikationsnummer: |
| ||||
Stichwörter / Keywords: | IMAGE CLASSIFICATION; ARTIFICIAL-INTELLIGENCE; DERMATOLOGISTS; RECOGNITION; ROBUSTNESS; Dermatology; Melanoma; Nevus; Artificial intelligence; Deep learning; Ensembles; Model soups; Robustness; Generalisation; Calibration | ||||
Dewey-Dezimal-Klassifikation: | 600 Technik, Medizin, angewandte Wissenschaften > 610 Medizin | ||||
Status: | Veröffentlicht | ||||
Begutachtet: | Ja, diese Version wurde begutachtet | ||||
An der Universität Regensburg entstanden: | Ja | ||||
Dokumenten-ID: | 57740 |
Zusammenfassung
Background: Image-based cancer classifiers suffer from a variety of problems which negatively affect their performance. For example, variation in image brightness or different cameras can already suffice to diminish performance. Ensemble solutions, where multiple model predictions are combined into one, can improve these problems. However, ensembles are computationally intensive and less ...
Zusammenfassung
Background: Image-based cancer classifiers suffer from a variety of problems which negatively affect their performance. For example, variation in image brightness or different cameras can already suffice to diminish performance. Ensemble solutions, where multiple model predictions are combined into one, can improve these problems. However, ensembles are computationally intensive and less transparent to practitioners than single model solutions. Constructing model soups, by averaging the weights of multiple models into a single model, could circumvent these limitations while still improving performance. Objective: To investigate the performance of model soups for a dermoscopic melanoma-nevus skin cancer classification task with respect to (1) generalisation to images from other clinics, (2) robustness against small image changes and (3) calibration such that the confidences correspond closely to the actual predictive uncertainties. Methods: We construct model soups by fine-tuning pre-trained models on seven different image resolutions and subsequently averaging their weights. Performance is evaluated on a multi-source dataset including holdout and external components. Results: We find that model soups improve generalisation and calibration on the external component while maintaining performance on the holdout component. For robustness, we observe performance improvements for pertubated test images, while the performance on corrupted test images remains on par. Conclusions: Overall, souping for skin cancer classifiers has a positive effect on generalisation, robustness and calibration. It is easy for practitioners to implement and by combining multiple models into a single model, complexity is reduced. This could be an important factor in achieving clinical applicability, as less complexity generally means more transparency. (c) 2022 The Authors. Published by Elsevier Ltd.
Metadaten zuletzt geändert: 29 Feb 2024 13:02