Penalizing small errors using an Adaptive Logarithmic Loss

Kaul, C., Pears, N., Dai, H., Murray-Smith, R. and Manandhar, S. (2021) Penalizing small errors using an Adaptive Logarithmic Loss. In: Del Bimbo, A., Cucchiara, R., Sclaroff, S., Farinella, G. M., Mei, T., Bertini, M., Escalante, H. J. and Vezzani, R. (eds.) Pattern Recognition. ICPR International Workshops and Challenges. Series: Lecture Notes in Computer Science, 12661. Springer: Cham, pp. 368-375. ISBN 9783030687632 (doi: 10.1007/978-3-030-68763-2_28)

[img] Text
253900.pdf - Accepted Version

485kB

Abstract

Loss functions are error metrics that quantify the difference between a prediction and its corresponding ground truth. Fundamentally, they define a functional landscape for traversal by gradient descent. Although numerous loss functions have been proposed to date in order to handle various machine learning problems, little attention has been given to enhancing these functions to better traverse the loss landscape. In this paper, we simultaneously and significantly mitigate two prominent problems in medical image segmentation namely: i) class imbalance between foreground and background pixels and ii) poor loss function convergence. To this end, we propose an Adaptive Logarithmic Loss (ALL) function. We compare this loss function with the existing state-of-the-art on the ISIC 2018 dataset, the nuclei segmentation dataset as well as the DRIVE retinal vessel segmentation dataset. We measure the performance of our methodology on benchmark metrics and demonstrate state-of-the-art performance. More generally, we show that our system can be used as a framework for better training of deep neural networks.

Item Type:Book Sections
Status:Published
Refereed:Yes
Glasgow Author(s) Enlighten ID:Murray-Smith, Professor Roderick and Kaul, Dr Chaitanya
Authors: Kaul, C., Pears, N., Dai, H., Murray-Smith, R., and Manandhar, S.
College/School:College of Science and Engineering > School of Computing Science
Publisher:Springer
ISSN:0302-9743
ISBN:9783030687632
Published Online:21 February 2021
Copyright Holders:Copyright © Springer Nature Switzerland AG 2021
Publisher Policy:Reproduced in accordance with the publisher copyright policy
Related URLs:

University Staff: Request a correction | Enlighten Editors: Update this record