CDF-Based Importance Sampling and Visualization for Neural Network Training
2023 (English)In: Eurographics Workshop on Visual Computing for Biology and Medicine / [ed] Thomas Höllt and Daniel Jönsson, 2023Conference paper, Published paper (Refereed)
Abstract [en]
Training a deep neural network is computationally expensive, but achieving the same network performance with less computation is possible if the training data is carefully chosen. However, selecting input samples during training is challenging as their true importance for the optimization is unknown. Furthermore, evaluation of the importance of individual samples must be computationally efficient and unbiased. In this paper, we present a new input data importance sampling strategy for reducing the training time of deep neural networks. We investigate different importance metrics that can be efficiently retrieved as they are available during training, i.e., the training loss and gradient norm. We found that choosing only samples with large loss or gradient norm, which are hard for the network to learn, is not optimal for the network performance. Instead, we introduce an importance sampling strategy that selects samples based on the cumulative distribution function of the loss and gradient norm, thereby making it more likely to choose hard samples while still including easy ones. The behavior of the proposed strategy is first analyzed on a synthetic dataset, and then evaluated in the application of classification of malignant cancer in digital pathology image patches. As pathology images contain many repetitive patterns, there could be significant gains in focusing on features that contribute stronger to the optimization. Finally, we show how the importance sampling process can be used to gain insights about the input data through visualization of samples that are found most or least useful for the training.
Place, publisher, year, edition, pages
2023.
Series
Eurographics Workshop on Visual Computing for Biomedicine, ISSN 2070-5778, E-ISSN 2070-5786
Keywords [en]
Computing methodologies, Neural networks, Human-centered computing, Visualization techniques;
National Category
Computer Sciences
Identifiers
URN: urn:nbn:se:liu:diva-199166DOI: 10.2312/vcbm.20231212ISBN: 978-3-03868-216-5 (print)OAI: oai:DiVA.org:liu-199166DiVA, id: diva2:1811538
Conference
VCBM 2023: Eurographics Workshop on Visual Computing for Biology and Medicine, Norrköping, Sweden, September 20 – 22, 2023
Funder
Wallenberg AI, Autonomous Systems and Software Program (WASP)
Note
The fulltext is published under Creative Common license Attribution 4.0https://creativecommons.org/licenses/by/4.0/
No changes have been made to the publication.
2023-11-132023-11-132023-11-21