Weights Augmentation: it has never ever ever ever let her model down
Not on the map. Why?
Abstract (arXiv)
Weight play an essential role in deep learning network models. Unlike network structure design, this article proposes the concept of weight augmentation, focusing on weight exploration. The core of Weight Augmentation Strategy (WAS) is to adopt random transformed weight coefficients training and transformed coefficients, named Shadow Weight(SW), for networks that can be used to calculate loss function to affect parameter updates. However, stochastic gradient descent is applied to Plain Weight(PW), which is referred to as the original weight of the network before the random transformation. During training, numerous SW collectively form high-dimensional space, while PW is directly learned from the distribution of SW instead of the data. The weight of the accuracy-oriented mode(AOM) relies on PW, which guarantees the network is highly robust and accurate. The desire-oriented mode(DOM) weight uses SW, which is determined by the network model's unique functions based on WAT's performance desires, such as lower computational complexity, lower sensitivity to particular data, etc. The dual mode be switched at anytime if needed. WAT extends the augmentation technique from data augmentation to weight, and it is easy to understand and implement, but it can improve almost all networks amazingly. Our experimental results show that convolutional neural networks, such as VGG-16, ResNet-18, ResNet-34, GoogleNet, MobilementV2, and Efficientment-Lite, can benefit much at little or no cost. The accuracy of models is on the CIFAR100 and CIFAR10 datasets, which can be evaluated to increase by 7.32% and 9.28%, respectively, with the highest values being 13.42% and 18.93%, respectively. In addition, DOM can reduce floating point operations (FLOPs) by up to 36.33%. The code is available at https://github.com/zlearh/Weight-Augmentation-Technology.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Builds on 29 of its 44 references that are in the map. First posted on arXiv 30 May 2024.
Builds on
29 of 44 references.
References · 29
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2014Very Deep Convolutional Networks for Large-Scale Image Recognitioncited by 15,267
- 2019EfficientNet: Rethinking Model Scaling for Convolutional Neural Networkscited by 6,415
- 2018MobileNetV2: Inverted Residuals and Linear Bottleneckscited by 4,773
- 2014Going Deeper with Convolutionscited by 4,231
- 2019RandAugment: Practical automated data augmentation with a reduced search spacecited by 2,261
- 2020Knowledge Distillation: A Surveycited by 1,578
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
- 2018Averaging Weights Leads to Wider Optima and Better Generalizationcited by 1,194
- 2019How Can We Know What Language Models Know?cited by 953
- 2021Robust fine-tuning of zero-shot modelscited by 785
17 more
- 2021RepVGG: Making VGG-style ConvNets Great Againcited by 520
- 2019Be Your Own Teacher: Improve the Performance of Convolutional Neural Networks via Self Distillationcited by 491
- 2018Bag of Tricks for Image Classification with Convolutional Neural Networkscited by 430
- 2020BatchEnsemble: An Alternative Approach to Efficient Ensemble and Lifelong Learningcited by 368
- 2023Effective Data Augmentation With Diffusion Modelscited by 282
- 2014Learning with Pseudo-Ensemblescited by 241
- 2019Model Fusion via Optimal Transportcited by 217
- 2021Adversarial Graph Augmentation to Improve Graph Contrastive Learningcited by 200
- 2023Text-to-Audio Generation using Instruction-Tuned LLM and Latent Diffusion Modelcited by 145
- 2019Learning Data Augmentation Strategies for Object Detectioncited by 139
- 2022Image Data Augmentation for Deep Learning: A Surveycited by 96
- 2022Stop Wasting My Time! Saving Days of ImageNet and BERT Training with Latest Weight Averagingcited by 54
- 2023Learning from models beyond fine-tuningcited by 47
- 2020What Makes a "Good" Data Augmentation in Knowledge Distillation -- A Statistical Perspectivecited by 35
- 2020Improving Auto-Augment via Augmentation-Wise Weight Sharingcited by 28
- 2015Dropout as data augmentationcited by 27
- 2023Toward Understanding Generative Data Augmentationcited by 25
Built on by
Citing papers per year: 2025: 1.
1 paper.