Residential College | false |
Status | 已發表Published |
Denoising Noisy Neural Networks: A Bayesian Approach with Compensation | |
Shao,Yulin1,2; Liew,Soung Chang3; Gunduz,Deniz2 | |
2023-06 | |
Source Publication | IEEE Transactions on Signal Processing |
ISSN | 1053-587X |
Volume | 71Pages:2460 - 2474 |
Abstract | Deep neural networks (DNNs) with noisy weights, which we refer to as noisy neural networks (NoisyNNs), arise from the training and inference of DNNs in the presence of noise. NoisyNNs emerge in many new applications, including the wireless transmission of DNNs, the efficient deployment or storage of DNNs in analog devices, and the truncation or quantization of DNN weights. This paper studies a fundamental problem of NoisyNNs: how to reconstruct the DNN weights from their noisy manifestations. While prior works relied exclusively on the maximum likelihood (ML) estimation, this paper puts forth a denoising approach to reconstruct DNNs with the aim of maximizing the inference accuracy of the reconstructed models. The superiority of our denoiser is rigorously proven in two small-scale problems, wherein we consider a quadratic neural network function and a shallow feedforward neural network, respectively. When applied to advanced learning tasks with modern DNN architectures, our denoiser exhibits significantly better performance than the ML estimator. Consider the average test accuracy of the denoised DNN model versus the weight variance to noise power ratio (WNR) performance. When denoising a noisy ResNet34 model arising from noisy inference, our denoiser outperforms ML estimation by up to 4.1 dB to achieve a test accuracy of $60\%$. When denoising a noisy ResNet18 model arising from noisy training, our denoiser outperforms ML estimation by 13.4 dB and 8.3 dB to achieve test accuracies of $60\%$ and $80\%$, respectively. |
Keyword | Denoiser Estimation Federated Edge Learning Maximum Likelihood Estimation Neural Networks Noise Measurement Noise Reduction Noisy Neural Network Training Wireless Communication Wireless Transmission Of Neural Networks |
DOI | 10.1109/TSP.2023.3290327 |
URL | View the original |
Indexed By | SCIE |
Language | 英語English |
WOS Research Area | Engineering |
WOS Subject | Engineering, Electrical & Electronic |
WOS ID | WOS:001030607300003 |
Publisher | Institute of Electrical and Electronics Engineers Inc. |
Scopus ID | 2-s2.0-85163532232 |
Fulltext Access | |
Citation statistics | |
Document Type | Journal article |
Collection | THE STATE KEY LABORATORY OF INTERNET OF THINGS FOR SMART CITY (UNIVERSITY OF MACAU) |
Corresponding Author | Shao,Yulin |
Affiliation | 1.State Key Laboratory of Internet of Things for Smart City, University of Macau, S.A.R 2.Department of Information Engineering, The Chinese University of Hong Kong, Shatin, New Territories, Hong Kong 3.Department of Electrical and Electronic Engineering, Imperial College London, London, U.K |
First Author Affilication | University of Macau |
Corresponding Author Affilication | University of Macau |
Recommended Citation GB/T 7714 | Shao,Yulin,Liew,Soung Chang,Gunduz,Deniz. Denoising Noisy Neural Networks: A Bayesian Approach with Compensation[J]. IEEE Transactions on Signal Processing, 2023, 71, 2460 - 2474. |
APA | Shao,Yulin., Liew,Soung Chang., & Gunduz,Deniz (2023). Denoising Noisy Neural Networks: A Bayesian Approach with Compensation. IEEE Transactions on Signal Processing, 71, 2460 - 2474. |
MLA | Shao,Yulin,et al."Denoising Noisy Neural Networks: A Bayesian Approach with Compensation".IEEE Transactions on Signal Processing 71(2023):2460 - 2474. |
Files in This Item: | There are no files associated with this item. |
Items in the repository are protected by copyright, with all rights reserved, unless otherwise indicated.
Edit Comment