Cargando…
Deep reinforcement learning with significant multiplications inference
We propose a sparse computation method for optimizing the inference of neural networks in reinforcement learning (RL) tasks. Motivated by the processing abilities of the brain, this method combines simple neural network pruning with a delta-network algorithm to account for the input data correlation...
Autores principales: | , , , |
---|---|
Formato: | Online Artículo Texto |
Lenguaje: | English |
Publicado: |
Nature Publishing Group UK
2023
|
Materias: | |
Acceso en línea: | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10682459/ https://www.ncbi.nlm.nih.gov/pubmed/38012259 http://dx.doi.org/10.1038/s41598-023-47245-y |
_version_ | 1785150980119068672 |
---|---|
author | Ivanov, Dmitry A. Larionov, Denis A. Kiselev, Mikhail V. Dylov, Dmitry V. |
author_facet | Ivanov, Dmitry A. Larionov, Denis A. Kiselev, Mikhail V. Dylov, Dmitry V. |
author_sort | Ivanov, Dmitry A. |
collection | PubMed |
description | We propose a sparse computation method for optimizing the inference of neural networks in reinforcement learning (RL) tasks. Motivated by the processing abilities of the brain, this method combines simple neural network pruning with a delta-network algorithm to account for the input data correlations. The former mimics neuroplasticity by eliminating inefficient connections; the latter makes it possible to update neuron states only when their changes exceed a certain threshold. This combination significantly reduces the number of multiplications during the neural network inference for fast neuromorphic computing. We tested the approach in popular deep RL tasks, yielding up to a 100-fold reduction in the number of required multiplications without substantial performance loss (sometimes, the performance even improved). |
format | Online Article Text |
id | pubmed-10682459 |
institution | National Center for Biotechnology Information |
language | English |
publishDate | 2023 |
publisher | Nature Publishing Group UK |
record_format | MEDLINE/PubMed |
spelling | pubmed-106824592023-11-30 Deep reinforcement learning with significant multiplications inference Ivanov, Dmitry A. Larionov, Denis A. Kiselev, Mikhail V. Dylov, Dmitry V. Sci Rep Article We propose a sparse computation method for optimizing the inference of neural networks in reinforcement learning (RL) tasks. Motivated by the processing abilities of the brain, this method combines simple neural network pruning with a delta-network algorithm to account for the input data correlations. The former mimics neuroplasticity by eliminating inefficient connections; the latter makes it possible to update neuron states only when their changes exceed a certain threshold. This combination significantly reduces the number of multiplications during the neural network inference for fast neuromorphic computing. We tested the approach in popular deep RL tasks, yielding up to a 100-fold reduction in the number of required multiplications without substantial performance loss (sometimes, the performance even improved). Nature Publishing Group UK 2023-11-27 /pmc/articles/PMC10682459/ /pubmed/38012259 http://dx.doi.org/10.1038/s41598-023-47245-y Text en © The Author(s) 2023 https://creativecommons.org/licenses/by/4.0/Open Access This article is licensed under a Creative Commons Attribution 4.0 International License, which permits use, sharing, adaptation, distribution and reproduction in any medium or format, as long as you give appropriate credit to the original author(s) and the source, provide a link to the Creative Commons licence, and indicate if changes were made. The images or other third party material in this article are included in the article’s Creative Commons licence, unless indicated otherwise in a credit line to the material. If material is not included in the article’s Creative Commons licence and your intended use is not permitted by statutory regulation or exceeds the permitted use, you will need to obtain permission directly from the copyright holder. To view a copy of this licence, visit http://creativecommons.org/licenses/by/4.0/ (https://creativecommons.org/licenses/by/4.0/) . |
spellingShingle | Article Ivanov, Dmitry A. Larionov, Denis A. Kiselev, Mikhail V. Dylov, Dmitry V. Deep reinforcement learning with significant multiplications inference |
title | Deep reinforcement learning with significant multiplications inference |
title_full | Deep reinforcement learning with significant multiplications inference |
title_fullStr | Deep reinforcement learning with significant multiplications inference |
title_full_unstemmed | Deep reinforcement learning with significant multiplications inference |
title_short | Deep reinforcement learning with significant multiplications inference |
title_sort | deep reinforcement learning with significant multiplications inference |
topic | Article |
url | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10682459/ https://www.ncbi.nlm.nih.gov/pubmed/38012259 http://dx.doi.org/10.1038/s41598-023-47245-y |
work_keys_str_mv | AT ivanovdmitrya deepreinforcementlearningwithsignificantmultiplicationsinference AT larionovdenisa deepreinforcementlearningwithsignificantmultiplicationsinference AT kiselevmikhailv deepreinforcementlearningwithsignificantmultiplicationsinference AT dylovdmitryv deepreinforcementlearningwithsignificantmultiplicationsinference |